diff --git a/app/create_stock_db.py b/app/create_stock_db.py index fb9222f..d54eb37 100755 --- a/app/create_stock_db.py +++ b/app/create_stock_db.py @@ -181,6 +181,21 @@ class StockDatabase: print(e) pass + # --- New Code: Check marketCap for PNK stocks --- + # Query the exchange short name from the database for this symbol + self.cursor.execute("SELECT exchangeShortName FROM stocks WHERE symbol = ?", (symbol,)) + row = self.cursor.fetchone() + if row and row[0] == "PNK": + market_cap = fundamental_data.get('marketCap') + if market_cap is not None and market_cap < 10E9: + # Delete the stock record as market cap is below 1 billion + self.cursor.execute("DELETE FROM stocks WHERE symbol = ?", (symbol,)) + self.conn.commit() + # Optionally, drop the OHLC table for that symbol if it exists + self.cursor.execute(f"DROP TABLE IF EXISTS '{symbol}'") + self.conn.commit() + print(f"Deleted {symbol} from database due to marketCap ({market_cap}) below 1B.") + return # Exit early without proceeding further # Check if columns already exist in the table self.cursor.execute("PRAGMA table_info(stocks)")