database/upsert_trade_example.py

43 lines
1.2 KiB
Python

import pandas as pd
from database.trades import upsert
trade_data = {
'Date': ['2024-01-22'],
'Symbol': ['SPX'],
'Strategy': ['Iron Condor @ Market Open'],
'Entry Time': [pd.to_datetime('2024-01-22 09:30:00')],
'Exit Time': [None],
'Spreads': [
[
{
'Legs': [
{'Action': 'BUY', 'Strike': 150, 'Type': 'CALL'},
{'Action': 'SELL', 'Strike': 155, 'Type': 'CALL'}
],
'Open': 1.5
},
{
'Legs': [
{'Action': 'SELL', 'Strike': 160, 'Type': 'PUT'},
{'Action': 'BUY', 'Strike': 155, 'Type': 'PUT'}
],
'Open': 2.5
}
]
],
'Profit': [None]
}
upsert(pd.DataFrame(trade_data))
trade_data['Exit Time'] = [pd.to_datetime('2024-01-22 16:00:00')]
trade_data['Spreads'][0][0]['High'] = 1.5
trade_data['Spreads'][0][0]['Low'] = 0.0
trade_data['Spreads'][0][0]['Close'] = 0.0
trade_data['Spreads'][0][1]['High'] = 2.5
trade_data['Spreads'][0][1]['Low'] = 0.0
trade_data['Spreads'][0][1]['Close'] = 0.0
trade_data['Profit'] = [400.0]
upsert(pd.DataFrame(trade_data))