mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-03-30 18:36:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

adding test to web_programming/current_stock_price

Browse Source
This commit is contained in:
Julian Perez Ramirez 2024-11-20 15:46:21 +01:00
parent 3e9ca92ca9
commit 8eab6b355e
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
web_programming/current_stock_price.py
View File

@ -1,14 +1,22 @@
import requests from doctest import testmod
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from requests import get
def stock_price(symbol: str = "AAPL") -> str: def stock_price(symbol: str = "AAPL") -> str:
"""
>>> stock_price("EEEE")
'-'
>>> isinstance(float(stock_price("GOOG")),float)
True
"""
url = f"https://finance.yahoo.com/quote/{symbol}?p={symbol}" url = f"https://finance.yahoo.com/quote/{symbol}?p={symbol}"
yahoo_finance_source = requests.get( yahoo_finance_source = get(
url, headers={"USER-AGENT": "Mozilla/5.0"}, timeout=10 url, headers={"USER-AGENT": "Mozilla/5.0"}, timeout=10
).text ).text
soup = BeautifulSoup(yahoo_finance_source, "html.parser") soup = BeautifulSoup(yahoo_finance_source, "html.parser")
specific_fin_streamer_tag = soup.find("fin-streamer", {"data-test": "qsp-price"}) specific_fin_streamer_tag = soup.find("fin-streamer", {"data-testid": "qsp-price"})
if specific_fin_streamer_tag: if specific_fin_streamer_tag:
text = specific_fin_streamer_tag.get_text() text = specific_fin_streamer_tag.get_text()
@ -18,5 +26,6 @@ def stock_price(symbol: str = "AAPL") -> str:
# Search for the symbol at https://finance.yahoo.com/lookup # Search for the symbol at https://finance.yahoo.com/lookup
if __name__ == "__main__": if __name__ == "__main__":
testmod()
for symbol in "AAPL AMZN IBM GOOG MSFT ORCL".split(): for symbol in "AAPL AMZN IBM GOOG MSFT ORCL".split():
print(f"Current {symbol:<4} stock price is {stock_price(symbol):>8}") print(f"Current {symbol:<4} stock price is {stock_price(symbol):>8}")