48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
|
import os
|
||
|
import pandas as pd
|
||
|
import paramiko
|
||
|
import schedule
|
||
|
import time
|
||
|
import zipfile
|
||
|
|
||
|
quotes_directory = os.environ['DOWNLOAD_DIRECTORY']
|
||
|
download_directory = quotes_directory + 'Zip Archive/'
|
||
|
|
||
|
def process_file(file: str):
|
||
|
quote_data = pd.read_csv(file)
|
||
|
current_date = quote_data.iloc[0]['quote_datetime'][:10]
|
||
|
zero_dte_quotes = quote_data[(quote_data['expiration'] == current_date) & (quote_data['root'] == 'SPXW')]
|
||
|
if not zero_dte_quotes.empty:
|
||
|
zero_dte_quotes.to_csv(os.path.join(quotes_directory, current_date[:4], current_date + '.csv'), index=False)
|
||
|
|
||
|
def download_quotes():
|
||
|
client = paramiko.SSHClient()
|
||
|
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||
|
client.connect(
|
||
|
hostname='sftp.datashop.livevol.com',
|
||
|
username=os.environ['USER'],
|
||
|
password=os.environ['PASSWORD']
|
||
|
)
|
||
|
sftp = client.open_sftp()
|
||
|
sftp.chdir('./subscriptions/' + os.environ['ORDER'] + '/' + os.environ['ITEM'])
|
||
|
files = sftp.listdir()
|
||
|
for file in files:
|
||
|
local_file = download_directory + file
|
||
|
if not os.path.exists(local_file):
|
||
|
print('Downloading', file)
|
||
|
sftp.get(file, local_file)
|
||
|
with zipfile.ZipFile(local_file, 'r') as zip_file:
|
||
|
print('Extracting', file)
|
||
|
zip_file.extractall(download_directory)
|
||
|
extracted_file = local_file.replace('.zip', '.csv')
|
||
|
print('Processing', extracted_file)
|
||
|
process_file(extracted_file)
|
||
|
print('Deleting', extracted_file)
|
||
|
os.remove(extracted_file)
|
||
|
client.close()
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
schedule.every().day.at('04:00', 'America/Los_Angeles').do(download_quotes)
|
||
|
while True:
|
||
|
schedule.run_pending()
|
||
|
time.sleep(1)
|