epogrebnyak · March 23, 2024 16:58 · Mar 23, 2024
diff --git a/unstir.py b/unstir.py
@@ -0,0 +1,68 @@
+"""A minimal version of https://github.com/PatzEdi/Stockstir/."""
+
+import re
+from dataclasses import dataclass
+
+import httpx
+
+
+@dataclass
+class Provider:
+    base_url: str
+    regex: str # this a nice idea 
+
+    def ticker_url(self, ticker: str) -> str:
+        return self.base_url + ticker
+
+    def get_quote(self, ticker: str) -> float:
+        source = get_page(self.ticker_url(ticker))
+        return gather_price(source, self.regex)
+
+
+def gather_price(source: str, pattern: str) -> float:
+    # this can be refoctored in due course
+    # important that you treat gather_price() as a function that just needs
+    # source text and a pattern to run, you do not pass any other information inside
+    prices = re.findall(pattern, source)
+    counter = 0
+    for i in range(len(prices)):  # why not just enumerate()?
+        try:
+            # do you want the last or first valid counter?
+            # if you want the first - you can just return here
+            float(prices[i])
+            break
+        except:  # E722 Do not use bare `except`
+            counter += 1
+    return float(prices[counter].replace(",", ""))
+
+
+def get_page(url):
+    # will need to randomize headers
+    headers = {
+        "User-Agent": "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:49.0) Gecko/20100101 Firefox/49.0"
+    }
+    return httpx.get(url, headers=headers).text
+
+
+def get_quote(ticker: str, provider="cnbc"):
+    return providers[provider].get_quote(ticker)
+
+
+providers = {
+    "cnbc": Provider(
+        "https://www.cnbc.com/quotes/", r'(?<="price":")(.*)(?=","priceChange":")'
+    ),
+    "bi": Provider(
+        "https://markets.businessinsider.com/stocks/",
+        r'"currentValue":(.*?),"previousClose":',
+    ),
+    "zacks": Provider(
+        "https://www.zacks.com/stock/quote/", r'last_price">\$(.*?)<span>'
+    ),
+}
+
+# this goes to unit test
+assert providers["cnbc"].ticker_url("MSFT") == "https://www.cnbc.com/quotes/MSFT"
+
+# this is a sample run for the program
+print(get_quote("MSFT")) # 428.74
No results found