Rename pycatchs/st.py to pycatchs/stdb.py

pycatchs/stdb.py

@@ -0,0 +1,87 @@
+"""
+This module provides functionality for performing Automatic data updating tasks.
+"""
+import threading
+import time
+import pandas as pd
+from stocklist import Stocks
+from utils import eqt
+from datetime import datetime
+from pymongo import MongoClient
+from datetime import datetime, timedelta
+import pytz
+import os
+
+tz = pytz.timezone('Asia/Kolkata')
+df_logo = pd.read_csv('https://raw.githubusercontent.com/jarvisx17/nifty500/main/Nifty500.csv')[['Symbol', 'logo']]
+mongo_url = os.environ['MongoURL']
+
+def UpdatedCollectionName():
+  current_time = datetime.now(tz)
+  collection_name = current_time.strftime('%Y-%m-%d')
+  if current_time.time() >= datetime.strptime('15:30', '%H:%M').time():
+      collection_name = (current_time + timedelta(days=1)).strftime('%Y-%m-%d')
+      return collection_name
+  else:
+    return collection_name
+  
+import concurrent.futures
+import yfinance as yf 
+
+def get_live_price(symbol):
+    return yf.Ticker(symbol).history(period="1d").iloc[-1][['High','Close']].round(2)
+
+def status(row):
+    if row['LTP'] > row['High'] or row['High_T'] > row['High']:
+        return "Active"
+    else:
+        return "Pending"
+
+def get_live_prices(df):
+    print("it's live")
+    symbols = df['Symbol'].tolist()
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        prices = list(executor.map(get_live_price, symbols))
+        df[['High_T', 'LTP']] = prices
+        df['Status'] = df.apply(status, axis=1)
+    return df
+
+class DataManager:
+  """
+  This is a DataManager class that demonstrates its functionality.
+  """
+  def __init__(self):
+      self.stocks = None
+      self.equity = None
+      self.data_thread = threading.Thread(target=self.update_data)
+      self.data_thread.daemon = True
+      self.data_thread.start()
+
+  def update_data(self):
+    while True:
+      client = MongoClient(mongo_url)
+      db = client['mydatabase']
+      collection_name = UpdatedCollectionName()
+      if collection_name in db.list_collection_names():
+        collection = db[collection_name]
+        cursor = collection.find({})
+        stocks = pd.DataFrame(list(cursor))
+        stocks.drop('_id', axis=1, inplace=True)
+        self.stocks = stocks           
+      else: 
+        stocks = Stocks()
+        collection = db[collection_name]
+        cursor = collection.find({})
+        stocks = pd.DataFrame(list(cursor))
+        stocks.drop('_id', axis=1, inplace=True) 
+        self.stocks = stocks 
+      time.sleep(120)
+      
+  def get_stocks_data(self):
+    stocks = get_live_prices(self.stocks)
+    self.results = stocks.to_dict(orient="records") 
+    return self.results
+  
+  def get_equity_data(self, ticker, startdate, share_qty):
+    self.equity = eqt(ticker, startdate, share_qty)
+    return self.equity