current_stock_price test added (#12390)

* adding test to web_programming/current_stock_price

* adding test to web_programming/current_stock_price

* Update current_stock_price.py

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update current_stock_price.py

---------

Co-authored-by: Maxim Smolskiy <mithridatus@mail.ru>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
jperezr 2024-12-30 11:37:21 +01:00 committed by GitHub
parent 68b4c6b479
commit 2ca96b7c8e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,14 +1,30 @@
import requests import requests
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
"""
Get the HTML code of finance yahoo and select the current qsp-price
Current AAPL stock price is 228.43
Current AMZN stock price is 201.85
Current IBM stock price is 210.30
Current GOOG stock price is 177.86
Current MSFT stock price is 414.82
Current ORCL stock price is 188.87
"""
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 = requests.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 +34,9 @@ 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__":
from doctest import testmod
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}")