from oneclass.onedb import Onedb import pandas as pd from datetime import datetime import time import pyautogui as auto import threading durtime=0.001 auto.FAILSAFE = True auto.PAUSE=1.0 class autotrade: def __init__(self): self.init_parameter() self.read_config() def init_parameter(self): self.stockcode="" self.location={} self.process={} self.screenpic=0 self.autotradelist=pd.DataFrame() self.applylist=pd.DataFrame() self.onedb=Onedb() def read_config(self): with open("./config/autotrade.config", 'r',encoding='utf-8') as file_to_read: i=0 while True: lines = file_to_read.readline() lines=lines.strip() if not lines: break #如果是文件结尾,则退出 else: if lines=="[坐标]": while True: lines = file_to_read.readline() lines=lines.strip() i+=1 if lines=="[流程]": #流程的处理 while True: lines = file_to_read.readline() lines=lines.strip() i+=1 if not lines: print("self.location",self.location) print("self.process",self.process) return 1 #EOF!!! else: npos=lines.index(",") tmpwhat=lines[:npos].strip() tmpprocess=lines[npos+2:-1].split(",") self.process[tmpwhat]=tmpprocess else: #坐标的处理 npos=lines.index(",") #到第一个逗号前的中文名称 tmpwhere=lines[:npos].strip() tmpxy=lines[npos+2:-1].split(",") #第一个逗号后的元组 self.location[tmpwhere]=(int(tmpxy[0]),int(tmpxy[1])) def thread_it(self, func, *args): # '''将函数放入线程中执行''' #有两个参数 1，函数名称 func 2,函数变量 **args task = threading.Thread(target=func, args=args) # 创建线程 task.setDaemon(True) # 守护线程 task.start() def moveandclick(self,location): x,y=location[0],location[1] auto.moveTo(x, y, duration=durtime) auto.click(button='left') #解析autotrade.config的[流程]中定义的每一步操作该如何处理 def deal_process(self, task, stockcode, price,qty): if len(task)==1: time.sleep(int(task)) elif task =='stockcode': auto.typewrite(stockcode, durtime) elif task =='price': auto.typewrite(str(price),durtime) elif task =='qty': auto.typewrite(str(qty),durtime) elif task[:2] =='d:': auto.typewrite(task,durtime) elif task in ['enter','tab','space']: auto.press(task) else: self.moveandclick(self.location[task]) #股票交易操作(异常状况的处理方法 - 例如出了一个特殊的子窗口,且如果不关闭这个子窗口则所有操作没有办法继续) def sell_stock(self,stockcode, sellingprice, sellingqty): print("sell stock is start") for task in self.process['卖出股票']: self.deal_process(task,stockcode,sellingprice,sellingqty) print("sell stock is end") def buy_stock(self,stockcode, buyprice, buyqty): print("buy stock is start") for task in self.process['买入股票']: self.deal_process(task,stockcode,buyprice,buyqty) print("buy stock end") def sell_131(self,stockcode,price=0,qty=0): print(f"sell {self.stockcode} is start") for task in self.process['国债逆回购']: self.deal_process(task,stockcode,0,0) print(f"sell {self.stockcode} end") #如果没有"买入"单, 则没有需要撤销的"买入单"是会出错还是可以正常运行??? def withdraw(self,stockcode=0,price=0,qty=0): print(f"withdraw is start") for task in self.process['撤全买']: self.deal_process(task,0,0,0) print(f"withdraw end") def apply_newstock(self,stockcode=0,price=0,qty=0): print(f"apply new stock is start") for task in self.process['申购新股']: self.deal_process(task,0,0,0) print(f"apply new stock end") #当天还没有一笔"申请"时, "输出"按钮"是灰色的, 没有办法输出Xls文件, 但是程序不会出错 def export_applylist (self): print(f"export application txt file start......") for task in self.process['输出当日委托']: self.deal_process(task,0,0,0) print(f"export application txt file done!!!!!!") #读取交易系统输出到/tmpout/applylist.xls里面的记录 def read_applylist(self): file_data = "" i=rawdata=0 with open(r"./tmpout/applylist.xls", "r",encoding="GBK") as f: for line in f: line = line.replace("=", "") file_data += line i+=1 with open(r"./tmpout/applylist1.xls", "w",encoding="GBK") as f: f.write(file_data) df = pd.read_table(r"./tmpout/applylist1.xls", sep='\t', encoding="GBK") df=df.reset_index(drop=True) df = df.iloc[:, :df.shape[1] - 1] #删除最后列 self.applylist=df print(self.applylist) #从db 读取 autotradelist表 (需要执行的交易清单) def read_autotradelist(self): #onedb=Onedb() sql="SELECT * FROM autotradelist" self.autotradelist=self.onedb.From_table(sql) print(self.autotradelist) def timeisok(self, now, plantime): curhour=int(now.strftime("%H")) curmin=int(now.strftime("%M")) planhour=int(plantime[:2]) planmin=int(plantime[-2:]) if (curhour>=planhour and curmin>=planmin) or curhour>planhour: return 1 else: return 0 def mark_applyed_yes(self, stockcode, time, ntype=0): #onedb=Onedb() self.onedb.update_table(stockcode, time, ntype) def trade_time(self): now=datetime.now() tmphour=int(now.strftime("%H")) #9点开始到15:59都会执行 #不需要执行的时段 # #11:31-12:59也会执行 #15:30-15:59 if tmphour >=9 and tmphour<=15: return 1 else: return 0 def get_havelist(self): file_data = "" i=0 with open(r"./tmpout/havelist.xls", "r") as f: for line in f: if i>=2: line=line.replace("=","") file_data+=line i+=1 #print(file_data) with open(r"./tmpout/havelist1.xls", "w") as f: #f.write("file_table",file_data) f.write(file_data) # 重读 持仓 df = pd.read_table(r"./havelist1.xls", sep='\t', encoding="ansi") #df=df[['证券代码', '证券名称', '证券数量', '可卖数量', '摊簿成本价', '当前价', '摊簿浮动盈亏', '实现盈亏', '参考盈亏比例(%)']]#, '冻结数量']] print(df) df = df[['证券代码', '证券名称', '证券数量','摊簿成本价', '当前价', '摊簿浮动盈亏', '参考盈亏比例(%)']] # , '冻结数量']] df['证券代码'] = df['证券代码'].map(lambda x: str(x).rjust(6,'0')) df.columns = ['代码', '名称', '数量','成本', '当前价', '盈亏', '盈亏%'] self.onedb.del_table("havelist") self.onedb.into_table(df, "havelist", "代码", 9) return df def isdone(self): tradetype={"B":"买入","S":"卖出","Z":"卖出","":""} now=datetime.now() strdate=now.strftime('%m')+"-"+now.strftime('%d') #print("strdate of today is:",strdate) for keyauto,autotradelist in self.autotradelist.iterrows(): BSZname=tradetype[autotradelist.BSZ] autotradetime=autotradelist.time[-8:] autotradedate=autotradelist.time[5:10] if autotradedate!=strdate: continue for keyapply, applylist in self.applylist.iterrows(): #print("BSZname",BSZname) #print("autotradetime:",autotradetime) #print("tradedate",autotradedate) if autotradelist.stockcode==str(applylist['证券代码']) and \ BSZname==applylist['委托类别'] and \ autotradelist.qty==str(applylist['委托数量']) and \ autotradetime