From c535d24a1c4401baea88025bcbff8b7582f96389 Mon Sep 17 00:00:00 2001 From: MuslemRahimi Date: Mon, 10 Feb 2025 01:44:57 +0100 Subject: [PATCH] bugfixing --- app/cron_options_gex_dex.py | 4 ++-- app/cron_options_hottest_contracts.py | 2 +- app/cron_options_oi.py | 2 ++ 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/app/cron_options_gex_dex.py b/app/cron_options_gex_dex.py index f7e1ca4..41bff35 100644 --- a/app/cron_options_gex_dex.py +++ b/app/cron_options_gex_dex.py @@ -35,7 +35,7 @@ index_symbols =["^SPX","^VIX"] con.close() etf_con.close() -total_symbols = stocks_symbols+etf_symbols+index_symbols +total_symbols = index_symbols #stocks_symbols+etf_symbols+index_symbols def save_json(data, symbol, directory_path): os.makedirs(directory_path, exist_ok=True) # Ensure the directory exists @@ -66,7 +66,7 @@ def get_contracts_from_directory(directory: str): def get_expiration_date(option_symbol): # Define regex pattern to match the symbol structure - match = re.match(r"([A-Z]+)(\d{6})([CP])(\d+)", option_symbol) + match = re.match(r"(\^?[A-Z]+)(\d{6})([CP])(\d+)", option_symbol) if not match: raise ValueError(f"Invalid option_symbol format: {option_symbol}") diff --git a/app/cron_options_hottest_contracts.py b/app/cron_options_hottest_contracts.py index a5144bc..e8e7570 100644 --- a/app/cron_options_hottest_contracts.py +++ b/app/cron_options_hottest_contracts.py @@ -46,7 +46,7 @@ def save_json(data, symbol, directory="json/hottest-contracts/companies"): def parse_option_symbol(option_symbol): - match = re.match(r"([A-Z]+)(\d{6})([CP])(\d+)", option_symbol) + match = re.match(r"(\^?[A-Z]+)(\d{6})([CP])(\d+)", option_symbol) if not match: raise ValueError(f"Invalid option_symbol format: {option_symbol}") ticker, expiration, option_type, strike_price = match.groups() diff --git a/app/cron_options_oi.py b/app/cron_options_oi.py index 0de8ccc..d846bdb 100644 --- a/app/cron_options_oi.py +++ b/app/cron_options_oi.py @@ -84,6 +84,8 @@ async def get_single_contract_data(symbol, expiration, semaphore): return None async def process_batch(symbol, batch, semaphore, pbar): + symbol = symbol.replace("^","") #for index symbols + tasks = [get_single_contract_data(symbol, contract, semaphore) for contract in batch] results = []