146 lines
2.1 KiB
Python
146 lines
2.1 KiB
Python
#!/usr/bin/env python
|
|
# coding: utf-8
|
|
|
|
# In[1]:
|
|
|
|
|
|
import pandas as pd
|
|
import datetime
|
|
import numpy as np
|
|
import uuid
|
|
|
|
|
|
# In[2]:
|
|
|
|
|
|
class DealManager():
|
|
|
|
def __init__(self):
|
|
self.commission=0.04
|
|
self.columns=['uuid','figi','amount','startPrice','profit']
|
|
self.deals = pd.DataFrame(columns=self.columns)
|
|
self.deals = self.deals.set_index('uuid')
|
|
|
|
def findDealByPriceAndFig(self,price,figi):
|
|
ans=None
|
|
for i in range(self.deals.shape[0]):
|
|
if self.deals.iloc[i].startPrice == price and self.deals.iloc[i].figi == figi:
|
|
ans = self.deals.iloc[i].name
|
|
break
|
|
return ans
|
|
|
|
def openDeal(self,figi,startPrice,amount=1):
|
|
desiredDeal=self.findDealByPriceAndFig(startPrice,figi)
|
|
if desiredDeal == None:
|
|
newDealDict={
|
|
'uuid':[str(uuid.uuid4())],
|
|
'figi':[figi],
|
|
'startPrice':[startPrice],
|
|
'amount':[amount]
|
|
}
|
|
|
|
#newDealDict['profit']=[startPrice*pow(1+self.commission,2)]
|
|
|
|
|
|
|
|
newDeal=pd.DataFrame.from_dict(newDealDict).set_index('uuid')
|
|
self.deals=pd.concat([self.deals, newDeal])
|
|
else:
|
|
self.deals.at[desiredDeal,'amount'] += amount
|
|
|
|
def closeDeal(self,uuid,amount):
|
|
|
|
desiredDeal=self.deals.loc[uuid]
|
|
if desiredDeal.amount - amount == 0:
|
|
self.deals = self.deals.drop(labels = [uuid],axis = 0)
|
|
else:
|
|
self.deals.at[uuid,'amount'] -= amount
|
|
#self.deals.loc[uuid].amount = desiredDeal.amount - amount
|
|
|
|
|
|
|
|
# In[3]:
|
|
|
|
|
|
t=DealManager()
|
|
t.__dict__
|
|
|
|
|
|
# In[ ]:
|
|
|
|
|
|
|
|
|
|
|
|
# In[4]:
|
|
|
|
|
|
t.deals.shape[0]
|
|
|
|
|
|
# In[5]:
|
|
|
|
|
|
t.openDeal('huigi',100,1)
|
|
t.openDeal('huigi',100,3)
|
|
t.openDeal('huigi1',100,3)
|
|
t.openDeal('huigi1',200,3)
|
|
|
|
|
|
# In[6]:
|
|
|
|
|
|
t.deals
|
|
|
|
|
|
# In[7]:
|
|
|
|
|
|
t.deals[t.deals.figi == 'huigi1']
|
|
|
|
|
|
# In[ ]:
|
|
|
|
|
|
|
|
|
|
|
|
# In[8]:
|
|
|
|
|
|
for i in range(t.deals.shape[0]):
|
|
print(t.deals.iloc[i])
|
|
|
|
|
|
# In[9]:
|
|
|
|
|
|
t.findDealByPriceAndFig
|
|
|
|
|
|
# In[10]:
|
|
|
|
|
|
t.closeDeal('78228979-3daf-470a-9c2a-8db180c8c3b0',1)
|
|
t.deals
|
|
|
|
|
|
# In[11]:
|
|
|
|
|
|
t.deals.iloc[0].name
|
|
|
|
|
|
# In[12]:
|
|
|
|
|
|
a=2
|
|
a==None
|
|
|
|
|
|
# In[ ]:
|
|
|
|
|
|
|
|
|