GoekenDataScience 0.1.14__tar.gz → 0.1.18__tar.gz

{goekendatascience-0.1.14 → goekendatascience-0.1.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GoekenDataScience
-Version: 0.1.14
+Version: 0.1.18
 Summary: Python package for Goeken Data Science
 Author-email: Ryan Goeken <your@email.com>
 License: MIT

{goekendatascience-0.1.14 → goekendatascience-0.1.18}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "GoekenDataScience"                # Package name on PyPI
-version = "0.1.14"                   # Increment this for new releases
+version = "0.1.18"                   # Increment this for new releases
 description = "Python package for Goeken Data Science"
 readme = "README.md"
 requires-python = ">=3.7"           # minimum Python version

{goekendatascience-0.1.14 → goekendatascience-0.1.18}/src/GoekenDataScience.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GoekenDataScience
-Version: 0.1.14
+Version: 0.1.18
 Summary: Python package for Goeken Data Science
 Author-email: Ryan Goeken <your@email.com>
 License: MIT

{goekendatascience-0.1.14 → goekendatascience-0.1.18}/src/goekendatascience/goekendatascience.py

@@ -3,6 +3,7 @@ import pandas as pd
 import requests
 import datetime
 from datetime import date
+import yfinance as yf
 
 class TickerDataFetch:
 
@@ -20,34 +21,16 @@ class TickerDataFetch:
             ticker_date = item.split('.')[0].split()[1]
             self.namelist[ticker_name] = ticker_date
 
-    def fetch_historical_data(self, ticker, AV_keys):
-        key_success = 0
-        for i in AV_keys:
-            if key_success == 0:
-                url = f'https://www.alphavantage.co/query?function=TIME_SERIES_DAILY&symbol={ticker}&outputsize=full&apikey={i}'
-                r = requests.get(url)
-                data = r.json()     
-                if "Information" in data and "rate limit" in data["Information"].lower():
-                    print('Alpha Vangtage key',i,'failed, trying next key')
-                else:
-                    try:
-                        historical_data = pd.DataFrame(data[list(data.keys())[1]]).transpose()
-                        key_success = 1
-                    except:
-                        print(str(data))
-                        key_success = 0
-                
-        if key_success == 1:
-            print('Successfully fetched data for',ticker)
-            historical_data.columns = ['O', 'H', 'L', 'C', 'V']
-            os.makedirs(self.output_folder, exist_ok=True)
-            csv_path = os.path.join(self.output_folder, f"{ticker} {self.todays_date}.csv")
-            historical_data.to_csv(csv_path, index=True)
-            print(f"Saved CSV to: {csv_path}")
-        else:
-            print("Failed to fetch data")
+    def fetch_historical_data(self, ticker):
+        ticker_data = yf.download(ticker, period='max')
+        os.makedirs(self.output_folder, exist_ok=True)
+        csv_path = os.path.join(self.output_folder, f"{ticker} {self.todays_date}.csv")
+        DataFrame = pd.DataFrame(ticker_data)
+        DataFrame.columns = ['Close','High','Low','Open','Volume']
+        DataFrame.to_csv(csv_path,index=True,header=True)
+        print(f"Saved CSV to: {csv_path}")
 
-    def update_tickers (self,tickers,AV_key): 
+    def update_tickers (self,tickers): 
         self.folder_scan()
         for ticker in tickers:
             if ticker in self.namelist:
@@ -55,10 +38,10 @@ class TickerDataFetch:
                     print(ticker,"data found for",self.todays_date)
                 else:
                     os.remove(os.path.join(self.output_folder, f"{ticker} {self.namelist[ticker]}.csv"))
-                    self.fetch_historical_data(ticker,AV_key)
+                    self.fetch_historical_data(ticker)
                     print(ticker,"data found for",self.namelist[ticker],". Updating data for",self.todays_date)
             else:
-                self.fetch_historical_data(ticker,AV_key)
+                self.fetch_historical_data(ticker)
                 print(ticker,"data not found",{ticker},". Updating data for",self.todays_date)
         for ticker in self.namelist:
             if ticker in tickers:
@@ -73,7 +56,7 @@ class TickerDataFetch:
         self.folder_scan()
         for ticker in self.namelist:
             data = pd.read_csv(os.path.join(self.output_folder, f"{ticker} {self.namelist[ticker]}.csv"))
-            print(data.columns[0])
+            data.set_index('Date',inplace=True)
             ticker_data[ticker] = data
         return ticker_data