| import sqlite3 | |
| import pandas as pd | |
| import os | |
| 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 bars_1d WHERE symbol='{sym}' 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 | |
| new_univ = ",".join(selected) | |
| print("New Universe:", new_univ) | |
| with open('.env', 'r', encoding='utf-8') as f: | |
| lines = f.readlines() | |
| with open('.env', 'w', encoding='utf-8') as f: | |
| for line in lines: | |
| if line.startswith('UNIVERSE="'): | |
| f.write(f'UNIVERSE="{new_univ}"\n') | |
| else: | |
| f.write(line) | |