Trading-Bot-M20 / filter.py
raghava4u's picture
Upload folder using huggingface_hub
d53dc44 verified
Raw History Blame Contribute Delete
670 Bytes
import sqlite3
import pandas as pd
universe = "AAPL,MSFT,NVDA,AMD,META,GOOG,TSLA,AMZN,JPM,GS,V,JNJ,PFE,UNH,XOM,CVX,WMT,KO,CAT,GE,DIS,NFLX,SPY,QQQ,IWM,INTC,BAC,SOFI,PLTR,UBER,SHOP,PYPL,MA,CSCO,PEP,MRK,BRK.B,LLY,AVGO,HD,PG,COST,ABBV,CRM,MCD,LIN,IONQ,NBIS,VRT".split(",")
conn = sqlite3.connect('data/market_data.db')
selected = []
for sym in universe:
try:
df = pd.read_sql_query(f"SELECT close FROM '{sym}_1Day' ORDER BY timestamp DESC LIMIT 1", conn)
if not df.empty:
price = df['close'].iloc[0]
if 50 <= price <= 140:
selected.append(sym)
except Exception as e:
pass
print(",".join(selected))