Add gurufocus

Author: doruirimescu

Trading/stock/gurufocus/cli.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+
+from Trading.stock.gurufocus.gurufocus import get_piotroski_f_score
+from Trading.utils.google_search import get_first_google_result
+from Trading.utils.cli import Named
+from Trading.utils.custom_logging import get_logger
+import fire
+from selenium import webdriver
+import time
+
+LOGGER = get_logger(__file__)
+
+
+# DEBUG=true cli.py analyze --names '["pdco", "paypal", "johnson&johnson", "mcdonalds", "pepsi", "uniper", "palantir"]'
+class GuruFocusCLI(Named):
+    def analyze(self):
+        if not self.name and not self.names:
+            LOGGER.error("Name is required")
+        urls = []
+        if self.name:
+            urls.append(get_first_google_result("gurufocus summary " + self.name))
+        if self.names:
+            for name in self.names:
+                LOGGER.debug(f"Searching for: {name}")
+                urls.append(get_first_google_result("gurufocus summary " + name))
+
+        LOGGER.debug(f"URLs: {urls}")
+
+        driver = webdriver.Chrome()
+        for url in urls:
+            LOGGER.info(f"Scraping {url}")
+            name, f_score, g_score = get_piotroski_f_score(driver, url)
+            LOGGER.info(f"{name} " f"Piotroski F-score: {f_score}. GF score: {g_score}")
+
+            time.sleep(0.5)
+        driver.quit()
+
+if __name__ == "__main__":
+    fire.Fire(GuruFocusCLI)

Trading/stock/gurufocus/gurufocus.py

@@ -0,0 +1,68 @@
+import re
+from enum import Enum
+from typing import Tuple, Optional
+
+import requests
+from bs4 import BeautifulSoup
+from pydantic import BaseModel
+
+
+import time
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.action_chains import ActionChains
+from selenium.webdriver.common.keys import Keys
+
+def scroll_down(driver):
+    """Scroll down the webpage using ActionChains."""
+    actions = ActionChains(driver)
+    body = driver.find_element(By.TAG_NAME, 'body')
+    body.click()
+    actions.send_keys(Keys.PAGE_DOWN).perform()
+    actions.send_keys(Keys.PAGE_DOWN).perform()
+
+    time.sleep(0.5)
+    actions.send_keys(Keys.PAGE_DOWN).perform()
+    actions.send_keys(Keys.PAGE_DOWN).perform()
+    time.sleep(0.5)
+
+
+def get_piotroski_f_score(driver, url):
+    driver.get(url)
+    try:
+        # Wait for the table containing the Piotroski F-Score to load
+        scroll_down(driver)
+
+        WebDriverWait(driver, 15).until(
+            EC.presence_of_element_located((By.CSS_SELECTOR, "div.relative-box.children-card"))
+        )
+
+        last_breadcrumb = WebDriverWait(driver, 15).until(EC.presence_of_element_located((By.XPATH, "//div[@aria-label='Breadcrumb']//span[@class='el-breadcrumb__item'][last()]//span[@class='el-breadcrumb__inner']")))
+        name = last_breadcrumb.text.strip()
+        # Locate the table row containing the Piotroski F-Score and get the corresponding score
+        score_element = driver.find_element(By.XPATH, "//div[@class='relative-box children-card']//table//tr[1]//td[@class='t-right t-primary']")
+        f_score = score_element.text.strip()
+
+        # Wait for the GF Score element and retrieve it
+        gfscore_element = WebDriverWait(driver, 15).until(
+            EC.presence_of_element_located((By.XPATH, "//div[contains(@id, 'gf-score-section-')]//span[@class='t-primary']"))
+        )
+        gf_score = gfscore_element.text.strip()
+
+        return name, f_score, gf_score
+
+    except Exception as e:
+        print(f"Error: {e}")
+    return None
+
+# # Initialize the WebDriver
+# driver = webdriver.Chrome()
+
+# # URL to scrape
+# url = "https://www.gurufocus.com/stock/CHIX:VITBS/summary"
+
+# # Get the Piotroski F-Score
+# f_score, g_score = get_piotroski_f_score(driver, url)
+# if f_score:
+#     print(f"Piotroski F-score: {f_score} found. GF score: {g_score}")