import matplotlib.pyplot as plt
import pandas as pd
odf = pd.read_csv(f"{order_path}/order_20221112.csv")
odf.columns = ["order_id","timestamp","symbol","token","side","direction","price","amount","close","u","signal_ts","create_time"]
kdf = pd.read_csv(f"{kline_path}/BTCUSDT_5.csv")
kdf['ts'] = (kdf['open_time'] / 1000).astype('int')
btc_order = odf[odf['symbol'] == f"BTCUSDT_{symbol}"]
btc_order = btc_order[0:5]
btc_order['ts'] = btc_order['create_time']
# print(btc_order)
btc_order = btc_order.set_index('ts')
print(btc_order)
kdf = kdf[(kdf['ts'] >= btc_order['create_time'].iloc[0] - 60) & (kdf['ts'] <= btc_order['timestamp'].iloc[-1] + 60)]
kdf = kdf.set_index('ts')
print(kdf)
figure,axes = plt.subplots(nrows=2,ncols=1,figsize=(30,30))
figure.suptitle(f"BTCUSDT-{symbol}",fontsize=40,va='bottom')
kdf['open'].plot(label='open',ax=axes[0])
kdf['high'].plot(label='high',ax=axes[0])
kdf['low'].plot(label='low',ax=axes[0])
ax = kdf['close'].plot(label='close',ax=axes[0])
# p = btc_order['price'].plot(label='order',style="*",ms=20,ax=axes[0])
for i,row in btc_order.iterrows():
if(row['price'] is not None):
axes[0].scatter(row['create_time'], row['price'], color='b')
axes[0].scatter(row['timestamp'], row['price'], color='b')
axes[0].plot([row['create_time'],row['timestamp']], [row['price'],row['price']], color='black')
axes[0].annotate(row['price'],xy=(i,row['price']),xytext=(i,row['price']))
# row[''].plot(label='order',style="*",ms=20,ax=axes[0])
# plt.show()
axes[0].legend()