if main_window.get_show_state() != 3: main_window.maximize() main_window.set_focus() def winmin(): #窗口最小化 if main_window.GetShowState() != 2: main_window.Minimize() def closepopup(): #关闭弹窗 popup_hwnd = main_window.PopupWindow() if popup_hwnd: popup_window = app.window_(handle=popup_hwnd) popup_window.SetFocus() popup_window.Button.Click() return True return False def pos(): #获取持仓并刷新复制到剪切板 dialog_window.CCustomTabCtrl.ClickInput(coords=(30, 8)) #点击持仓 dialog_window.Button5.click() time.sleep(0.5) dialog_window.Button5.click() # time.sleep(0.2) # dialog_window.CVirtualGridCtrl.RightClick(coords=(100, 70)) # 右击持仓 # main_window.TypeKeys('C') #如果能复制了 就把这些打开 def copypos(): #获取剪切板信息 wincb.OpenClipboard() t = wincb.GetClipboardData(win32con.CF_TEXT) wincb.CloseClipboard() return t def copyposition(): #导出持仓并读取 dialog_window.CVirtualGridCtrl.RightClick(coords=(100, 70)) # 右击持仓 main_window.TypeKeys('S') time.sleep(0.1) closepopup() closepopup() with open('C:/Users/Administrator/Desktop/table.xls','r') as f: return [[i.split('\t')[1],i.split('\t')[3],i.split('\t')[4]] for i in f.readlines()[1:]] def order(x): #B是买 S是卖 开始下单 dialog_window.TypeKeys("{F6}") if x == 'B': for i in Blis: # dialog_window.window(title_re='重填').click() time.sleep(0.1) dialog_window.Edit1.set_focus() dialog_window.Edit1.set_edit_text(i[0]) dialog_window.Edit3.set_edit_text(i[1]) time.sleep(0.2) dialog_window.Button1.click() if x == 'S': for i in Slis: time.sleep(0.1) dialog_window.Edit4.set_focus() dialog_window.Edit4.set_edit_text(i[0]) dialog_window.Edit6.set_edit_text(i[1]) time.sleep(0.2) dialog_window.Button2.click() def cancel(x): #撤单 B:撤买 S:撤卖 all:全撤 dialog_window.CCustomTabCtrl.ClickInput(coords=(140, 8)) #点击委托 try: dialog_window.Button6.Click() time.sleep(0.1) dialog_window.Button6.Click() except Exception as e: pass if x == 'B': dialog_window.Button8.Click() if x == 'S': dialog_window.Button9.Click() if x == 'all': dialog_window.Button7.Click() time.sleep(0.1) closepopup() def BSlist(x): #返回买卖剩余量 B是买 S是卖 global Blis global Slis pos() #可以复制了就打开old # old = [[i.split('\t')[1],i.split('\t')[3],i.split('\t')[4]] for i in copypos().decode("gb2312").split('\r\n')[1:]] old = copyposition() new = [[i[0],'0'] for i in Slis if int(i[1]) > 0 ]+Blis if x == 'B': B1 = [[v[0],str(int(i[1])-int(v[1]))] for i in [i for i in new if i[1] != '0'] for v in old if i[0] == v[0]] B2 = [[k,v] for k,v in dict([i for i in new if i[1] != '0']).items() if k not in dict(B1).keys()] Blis = [i for i in B1 if i[1] != '0']+B2 return Blis if x == 'S': Slis = [[i[0],i[2]] for i in old for v in [i for i in new if i[1] == '0'] if i[0] == v[0] and i[2] != '0'] return Slis if __name__ == '__main__': files = [i for i in os.listdir('D:/abdata/csv/') if 'cod' in i] Blis = [] Slis = [] with open('D:/abdata/csv/'+sorted(files)[-1],'r',encoding='utf-8') as f: for i in f: i = i.strip().split(',') if i[4] == '0' and int(i[2]) >0:Blis.append([i[0],i[2]]) if i[4] == '1' and int(i[2]) >0:Slis.append([i[0],i[2]]) ''' order(x): # 需要传参 B是买 S是卖 cancel(x): # 撤单 B:撤买 S:撤卖 all:全撤 BSlist(x): # 返回买卖剩余量 B是买 S是卖 winmax(): # 窗口最大化 　　winmin(): # 窗口最小化 pos(): # 获取持仓并刷新复制到剪切板 copypos(): # 获取剪切板信息 closepopup(): #关闭弹窗 copyposition(): #导出持仓并读取 ''' title = '网上股票交易系统5.0' app = pywinauto.application.Application() app.connect(title=title) top_hwnd = pywinauto.findwindows.find_window(title=title) dialog_hwnd = pywinauto.findwindows.find_windows(top_level_only=False, class_name=u'#32770', parent=top_hwnd)[0] wanted_hwnds = pywinauto.findwindows.find_windows(top_level_only=False, parent=dialog_hwnd) main_window = app.window(handle=top_hwnd) dialog_window = app.window(handle=dialog_hwnd) winmax() #窗口最大 # pos() #获取复制持仓 # old = [[i.split('\t')[1],i.split('\t')[3]] for i in copypos().decode("gb2312").split('\r\n')[1:]] # new = [[i[0],'0'] for i in Slis if int(i[1]) > 0 ]+Blis B = 1 S = 1 while S > 0 : closepopup() time.sleep(0.5) # pos() #获取复制持仓 Slis = BSlist('S') S = len(Slis) if S > 0: closepopup() order('S') closepopup() time.sleep(2) cancel('all') time.sleep(2) while B > 0 : time.sleep(0.5) closepopup() # pos() #获取复制持仓 Blis = BSlist('B') B = len(Blis) if B > 0: closepopup() order('B') closepopup() time.sleep(2) cancel('all') time.sleep(2)