Skip to content
Merged
Prev Previous commit
Next Next commit
current_stock_price test added (TheAlgorithms#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>
  • Loading branch information
3 people authored Dec 30, 2024
commit 2ca96b7c8ec2134a7282bed13f1cc93358c13c45
22 changes: 21 additions & 1 deletion web_programming/current_stock_price.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,30 @@
import requests
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:
"""
>>> stock_price("EEEE")
'-'
>>> isinstance(float(stock_price("GOOG")),float)
True
"""
url = f"https://finance.yahoo.com/quote/{symbol}?p={symbol}"
yahoo_finance_source = requests.get(
url, headers={"USER-AGENT": "Mozilla/5.0"}, timeout=10
).text
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:
text = specific_fin_streamer_tag.get_text()
Expand All @@ -18,5 +34,9 @@ def stock_price(symbol: str = "AAPL") -> str:

# Search for the symbol at https://finance.yahoo.com/lookup
if __name__ == "__main__":
from doctest import testmod

testmod()

for symbol in "AAPL AMZN IBM GOOG MSFT ORCL".split():
print(f"Current {symbol:<4} stock price is {stock_price(symbol):>8}")