【自动下单外挂】

# -*- coding: utf-8 -*-
'''
登录
买入
卖出 
撤单

'''
import pandas as pd  
import numpy as np
import sys   
import datetime
import time
import scipy.io
import os
import uuid

import signal

import win32api
import win32con
import win32com,win32com.client

import win32console
import win32process
import win32clipboard

import win32gui
import pywinauto # 需要pip安装 pip install pywinauto
import pywinauto.controls.hwndwrapper as hwndWrapper
import pywinauto.findwindows as findwindows
import pywinauto.controls.win32_controls as win32_controls

def getclip():
    win32clipboard.OpenClipboard()
    formats=[]
    lastformat=0
    while 1:
        nextfomat=win32clipboard.EnumClipboardFormats(lastformat)
        if 0==nextfomat:
            break
        else:
            formats.append(nextfomat)
            lastformat=nextfomat
    if len(formats)==0:
        win32clipboard.CloseClipboard()
        return ''
    clipmemory=win32clipboard.GetClipboardData(13)#13是str
    # print('
format:{}
clipboard:{}'.format(lastformat,clipmemory))
    win32clipboard.CloseClipboard()
    return clipmemory


class zyc_time:
    def __init__(self):
        pass
    def zyc_timeNow(self):
        t=time.localtime(time.time())
        tS=time.strftime("%Y-%m-%d %H:%M:%S",t)
        return tS
    def zyc_print(self,strG):
        tS=self.zyc_timeNow()
        print(tS+'  '+strG)
    def start(self):
        self.tt_start=time.time()
    def end(self):
        self.tt_end=time.time()
        self.tt=self.tt_end-self.tt_start
        self.zyc_print('耗时: '+str(round(self.tt,3))+' 秒')
        
class ths_api(zyc_time):
    __is_run=0
    def __init__(self):
        zyc_time.__init__(self)
    # 同花顺客户端登录
    def 同花顺客户端登录(self,Name='网上股票交易系统5.0'):
        # 获取主句柄 同花顺模拟账户 42763175  密码 789789
        i=0;self.hand=0
        self.hand = win32gui.FindWindowEx(0, self.hand, None,Name)
        while self.hand==0 and i<5:
            i=i+1
            self.hand = win32gui.FindWindowEx(0, self.hand, None,Name)
            time.sleep(0.1)
        if self.hand==0:
            self.zyc_print('请登录同花顺股票统一客户端')
            return
        else:
            pass
            self.zyc_print('获取主句柄成功,self.hand='+str(self.hand))


        # 获取买入句柄
        # 点击F1
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F1)  # 发送双向交易窗口
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F1)
        time.sleep(0.5)
        self.买入句柄={}
        i = 0;
        self.买入句柄['hand'] = 0
        self.买入句柄['hand'] = win32gui.GetDlgItem(self.hand, 0xE900)
        self.买入句柄['hand'] = win32gui.GetDlgItem(self.买入句柄['hand'], 0xE901)
        while self.买入句柄['hand'] == 0 and i < 5:
            i = i + 1
            self.买入句柄['hand'] = win32gui.GetDlgItem(self.hand, 0xE900)
            self.买入句柄['hand'] = win32gui.GetDlgItem(self.买入句柄['hand'], 0xE901)
            time.sleep(0.1)
        if self.买入句柄['hand'] == 0:
            self.zyc_print('请登录同花顺股票统一客户端')
            return
        else:
            pass
            self.zyc_print('获取买入句柄成功,买入句柄=' + str(self.买入句柄['hand']))

        i = 0
        self.买入句柄['hand_stock']=0
        self.买入句柄['hand_price'] = 0
        self.买入句柄['hand_num'] = 0
        self.买入句柄['hand_buy'] = 0
        while ( self.买入句柄['hand_stock']== 0 or self.买入句柄['hand_price'] ==0 or self.买入句柄['hand_num'] == 0 or self.买入句柄['hand_buy'] == 0 ) and i < 10:
            i = i + 1
            self.买入句柄['hand_stock'] = win32gui.GetDlgItem(self.买入句柄['hand'], 1032)
            self.买入句柄['hand_price'] = win32gui.GetDlgItem(self.买入句柄['hand'], 1033)
            self.买入句柄['hand_num'] = win32gui.GetDlgItem(self.买入句柄['hand'], 1034)
            self.买入句柄['hand_buy'] = win32gui.GetDlgItem(self.买入句柄['hand'], 1006)
            time.sleep(0.1)
        if i>5:
            self.zyc_print('异常')

        # 点击F6
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F6)  # 发送双向交易窗口
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F6)
        time.sleep(0.5)
        i = 0
        self.买入句柄['hand_F6'] = 0
        while self.买入句柄['hand_F6'] == 0 and i < 5:
            self.买入句柄['hand_F6'] = win32gui.GetDlgItem(self.买入句柄['hand'], 1047)
            time.sleep(0.5)
        # 点击F7
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F7)  # 发送双向交易窗口
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F7)
        time.sleep(0.5)
        i = 0
        self.买入句柄['hand_F7'] = 0
        while self.买入句柄['hand_F7'] == 0 and i < 5:
            self.买入句柄['hand_F7'] = win32gui.GetDlgItem(self.买入句柄['hand'], 1047)
            time.sleep(0.5)
        # 点击F8
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F8)  # 发送双向交易窗口
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F8)
        time.sleep(0.5)
        i = 0
        self.买入句柄['hand_F8'] = 0
        while self.买入句柄['hand_F8'] == 0 and i < 5:
            self.买入句柄['hand_F8'] = win32gui.GetDlgItem(self.买入句柄['hand'], 1047)
            time.sleep(0.5)

        # 获取卖出句柄
        # 点击F2
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F2)  # 发送双向交易窗口
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F2)
        time.sleep(0.5)
        self.卖出句柄 = {}
        i = 0;
        self.卖出句柄['hand'] = 0
        self.卖出句柄['hand'] = win32gui.GetDlgItem(self.hand, 0xE900)
        self.卖出句柄['hand'] = win32gui.GetDlgItem(self.卖出句柄['hand'], 0xE901)
        while self.卖出句柄['hand'] == 0 and i < 5:
            i = i + 1
            self.卖出句柄['hand'] = win32gui.GetDlgItem(self.hand, 0xE900)
            self.卖出句柄['hand'] = win32gui.GetDlgItem(self.卖出句柄['hand'], 0xE901)
            time.sleep(0.1)
        if self.卖出句柄['hand'] == 0:
            self.zyc_print('请登录同花顺股票统一客户端')
            return
        else:
            pass
            self.zyc_print('获取卖出句柄成功,卖出句柄=' + str(self.卖出句柄['hand']))
        i = 0
        self.卖出句柄['hand_stock'] = 0
        self.卖出句柄['hand_price'] = 0
        self.卖出句柄['hand_num'] = 0
        self.卖出句柄['hand_sell'] = 0
        while (self.卖出句柄['hand_stock'] == 0 or self.卖出句柄['hand_price'] == 0 or self.卖出句柄['hand_num'] == 0 or self.卖出句柄['hand_sell'] == 0) and i < 10:
            i = i + 1
            self.卖出句柄['hand_stock'] = win32gui.GetDlgItem(self.卖出句柄['hand'], 1032)
            self.卖出句柄['hand_price'] = win32gui.GetDlgItem(self.卖出句柄['hand'], 1033)
            self.卖出句柄['hand_num'] = win32gui.GetDlgItem(self.卖出句柄['hand'], 1034)
            self.卖出句柄['hand_sell'] = win32gui.GetDlgItem(self.卖出句柄['hand'], 1006)
            time.sleep(0.1)
        if i > 5:
            self.zyc_print('异常')

        # 获取撤单句柄
        # 点击F3
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F3)  # 发送双向交易窗口
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F3)
        time.sleep(0.5)
        self.撤单句柄 = {}
        i = 0;
        self.撤单句柄['hand'] = 0
        self.撤单句柄['hand'] = win32gui.GetDlgItem(self.hand, 0xE900)
        self.撤单句柄['hand'] = win32gui.GetDlgItem(self.撤单句柄['hand'], 0xE901)
        while self.撤单句柄['hand'] == 0 and i < 5:
            i = i + 1
            self.撤单句柄['hand'] = win32gui.GetDlgItem(self.hand, 0xE900)
            self.撤单句柄['hand'] = win32gui.GetDlgItem(self.撤单句柄['hand'], 0xE901)
            time.sleep(0.1)
        if self.撤单句柄['hand'] == 0:
            self.zyc_print('请登录同花顺股票统一客户端')
            return
        else:
            pass
            self.zyc_print('获取撤单句柄成功,撤单句柄=' + str(self.撤单句柄['hand']))
        
        i = 0
        self.撤单句柄['撤单'] = 0
        self.撤单句柄['全撤'] = 0
        self.卖出句柄['撤买'] = 0
        self.卖出句柄['撤卖'] = 0
        while (self.撤单句柄['撤单'] == 0 or self.撤单句柄['全撤'] == 0 or self.撤单句柄['撤买'] == 0 or self.撤单句柄['撤卖'] == 0) and i < 10:
            i = i + 1
            self.撤单句柄['撤单'] = win32gui.GetDlgItem(self.撤单句柄['hand'], 1099)
            self.撤单句柄['全撤'] = win32gui.GetDlgItem(self.撤单句柄['hand'], 30001)
            self.撤单句柄['撤买'] = win32gui.GetDlgItem(self.撤单句柄['hand'], 30002)
            self.撤单句柄['撤卖'] = win32gui.GetDlgItem(self.撤单句柄['hand'], 30003)
            time.sleep(0.1)
        if i > 5:
            self.zyc_print('异常')
            
            
        # 获取查询资金句柄
        # 点击F4
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F4)  # 发送双向交易窗口
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F4)
        time.sleep(0.5)
        self.查询句柄 = {}
        i = 0;
        self.查询句柄['hand'] = 0
        self.查询句柄['hand'] = win32gui.GetDlgItem(self.hand, 0xE900)
        self.查询句柄['hand'] = win32gui.GetDlgItem(self.查询句柄['hand'], 0xE901)
        while self.查询句柄['hand'] == 0 and i < 5:
            i = i + 1
            self.查询句柄['hand'] = win32gui.GetDlgItem(self.hand, 0xE900)
            self.查询句柄['hand'] = win32gui.GetDlgItem(self.查询句柄['hand'], 0xE901)
            time.sleep(0.1)
        if self.查询句柄['hand'] == 0:
            self.zyc_print('请登录同花顺股票统一客户端')
            return
        else:
            pass
            self.zyc_print('获取查询句柄成功,查询句柄=' + str(self.查询句柄['hand']))

    # 买入
    def 买入(self,stock,price,num):
        self.start()
        self.KnkDlg()
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F1)
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F1)
        time.sleep(0.1)
        i=0
        while i==0 :
            try:
                win32_controls.EditWrapper(self.买入句柄['hand_stock']).set_edit_text(stock)
                win32_controls.EditWrapper(self.买入句柄['hand_price']).set_edit_text(price)
                win32_controls.EditWrapper(self.买入句柄['hand_num']).set_edit_text(num)
            except:
                pass
            a = win32_controls.EditWrapper(self.买入句柄['hand_stock']).texts()[1] == stock
            b = win32_controls.EditWrapper(self.买入句柄['hand_price']).texts()[1] == price
            c = win32_controls.EditWrapper(self.买入句柄['hand_num']).texts()[1] == num
            if a and b and c:
                i=1
            if i==0:
                time.sleep(0.1)
        hand_buy = win32_controls.ButtonWrapper(self.买入句柄['hand_buy'])
        while hand_buy.IsVisible() == 0 or hand_buy.IsEnabled() == 0:
            time.sleep(0.1)
        hand_buy.Click()
        self.end()
        if self.KnkDlg()==0:
            self.zyc_print('买入失败,'+stock+','+price+','+num)
            return 1
        self.zyc_print('买入成功,'+stock+','+price+','+num)
        return 0
    # 卖出
    def 卖出(self,stock,price,num):
        self.start()
        self.KnkDlg()
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F2)
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F2)
        time.sleep(0.1)
        i=0
        while i==0 :
            try:
                win32_controls.EditWrapper(self.卖出句柄['hand_stock']).set_edit_text(stock)
                win32_controls.EditWrapper(self.卖出句柄['hand_price']).set_edit_text(price)
                win32_controls.EditWrapper(self.卖出句柄['hand_num']).set_edit_text(num)
            except:
                pass
            a = win32_controls.EditWrapper(self.卖出句柄['hand_stock']).texts()[1] == stock
            b = win32_controls.EditWrapper(self.卖出句柄['hand_price']).texts()[1] == price
            c = win32_controls.EditWrapper(self.卖出句柄['hand_num']).texts()[1] == num
            if a and b and c:
                i=1
            if i==0:
                time.sleep(0.1)
        hand_sell = win32_controls.ButtonWrapper(self.卖出句柄['hand_sell'])
        while hand_sell.IsVisible() == 0 or hand_sell.IsEnabled() == 0:
            time.sleep(0.1)
        hand_sell.Click()
        self.end()
        if self.KnkDlg()==0:
            self.zyc_print('卖出失败,'+stock+','+price+','+num)
            return 1
        self.zyc_print('卖出成功,'+stock+','+price+','+num)
        return 0
    
    # 撤单
    def 撤单(self,batch=None):
        # self.撤单(batch='all')#全撤
        # self.撤单(batch='buy')#全撤
        # self.撤单(batch='sell')#全撤
        # self.撤单(single='600784,7.00,证券买入')#按条件撤单
        #self.撤单(subnum='1234')#按委托单号撤单
        # 
        self.KnkDlg()
        win32gui.PostMessage(self.hand, win32con.WM_KEYDOWN, win32con.VK_F3)
        win32gui.PostMessage(self.hand, win32con.WM_KEYUP, win32con.VK_F3)
        time.sleep(0.5)
        
        if batch=='all':
          hwndWrapper.HwndWrapper(self.撤单句柄['全撤']).SetFocus()
          win32_controls.ButtonWrapper(self.撤单句柄['全撤']).Click()          
          
        elif batch=='buy':
            hwndWrapper.HwndWrapper(self.撤单句柄['撤买']).SetFocus()
            win32_controls.ButtonWrapper(self.撤单句柄['撤买']).Click()   
        elif batch=='sell':
            hwndWrapper.HwndWrapper(self.撤单句柄['撤卖']).SetFocus()
            win32_controls.ButtonWrapper(self.撤单句柄['撤卖']).Click()               
        else:
            pass
        trec=time.time()
        while True:
          if time.time()-trec>10:
            break 
          time.sleep(0.5)   
          if self.KnkDlg()==1:
            break      
        return 0
     
    def KnkDlg(self):
        # 关闭所有同进程同线程的对话框
        thread, processId = win32process.GetWindowThreadProcessId(self.hand)
        temp_hwnd = pywinauto.findwindows.find_windows(top_level_only=True, class_name='#32770', parent=None)
        for hdlg in temp_hwnd:
          dlgthread, dlgprocessId = win32process.GetWindowThreadProcessId(hdlg)
          if dlgprocessId==processId and win32gui.IsWindowVisible(hdlg) :
            hcmd1 = pywinauto.findwindows.find_windows(top_level_only=False, class_name='Button', title_re='是',parent=hdlg)
            hcmd2 = pywinauto.findwindows.find_windows(top_level_only=False, class_name='Button', title_re='确定',parent=hdlg)
            hcmd3 = pywinauto.findwindows.find_windows(top_level_only=False, class_name='Button', title_re='确认',parent=hdlg)
            time.sleep(0.1)
            if len(hcmd1)>0:
              hwndWrapper.HwndWrapper(hcmd1[0]).SetFocus()
              win32_controls.ButtonWrapper(hcmd1[0]).Click()
              return 0
            elif len(hcmd2)>0:
              hwndWrapper.HwndWrapper(hcmd2[0]).SetFocus()
              win32_controls.ButtonWrapper(hcmd2[0]).Click()
              return 0
            elif len(hcmd3)>0:
              hwndWrapper.HwndWrapper(hcmd3[0]).SetFocus()
              win32_controls.ButtonWrapper(hcmd3[0]).Click()
              return 0      
        return 1   
    def chkwnd(self):
      if win32gui.IsWindow(self.hand)==0 or self.hand==0:
        print ('未发现交易窗口！')
        return 1      
      trec=time.time()
      while win32gui.IsWindowVisible( self.hand)==0:
        if time.time()-trec>3:
          print('无法调出隐藏的窗口！')
          return 1
        thread, processId = win32process.GetWindowThreadProcessId(self.hand)
        p = psutil.Process(processId)
        win32process.CreateProcess(p.exe(),'',None,None,0,win32process.CREATE_NO_WINDOW,None,None,win32process.STARTUPINFO())
        time.sleep(0.1)
      return 0  
if __name__=='__main__':
    pass
    self=ths_api()
    self.同花顺客户端登录()
    for i in range(3):
        is_buy= self.买入('000001','9.00','100') # stock='000001';price='10.5';num='300'
        is_sell= self.卖出('000001','11.35','100') # stock='000001';price='10.5';num='300'
    self.撤单(batch='all')#全撤