python 自动交易股票_利用python3.5 +TK 开发股票自动交易伴侣

#-*- encoding: utf8 -*-#version 1.11

importtkinter.messagebox,osfrom tkinter import *

from tkinter.ttk import *

from tkinter importMenuimportdatetimeimportthreadingimportpickleimporttimeimporttushare as tsimportpywinautoimportpywinauto.clipboardimportpywinauto.application

NUM_OF_STOCKS= 5 #自定义股票数量

is_start =False

is_monitor=True

set_stocks_info=[]

actual_stocks_info=[]

consignation_info=[]

is_ordered= [1] * NUM_OF_STOCKS #1：未下单 0：已下单

is_dealt = [0] * NUM_OF_STOCKS #0: 未成交 负整数：卖出数量， 正整数：买入数量

stock_codes = [''] *NUM_OF_STOCKSclassOperationThs:def __init__(self):try:

self.__app =pywinauto.application.Application()

self.__app.connect(title='网上股票交易系统5.0')

top_hwnd= pywinauto.findwindows.find_window(title='网上股票交易系统5.0')

dialog_hwnd= pywinauto.findwindows.find_windows(top_level_only=False, class_name='#32770', parent=top_hwnd)[0]

wanted_hwnds= pywinauto.findwindows.find_windows(top_level_only=False, parent=dialog_hwnd)print('wanted_hwnds length', len(wanted_hwnds))if len(wanted_hwnds) not in (99,97,96,98,100,101):

tkinter.messagebox.showerror('错误', '无法获得“同花顺双向委托界面”的窗口句柄,请将同花顺交易系统切换到“双向委托界面”！')

exit()

self.__main_window = self.__app.window_(handle=top_hwnd)

self.__dialog_window = self.__app.window_(handle=dialog_hwnd)except:pass

def __buy(self, code, quantity):"""买函数

:param code: 代码， 字符串

:param quantity: 数量， 字符串"""self.__dialog_window.Edit1.SetFocus()

time.sleep(0.2)

self.__dialog_window.Edit1.SetEditText(code)

time.sleep(0.2)if quantity != '0':

self.__dialog_window.Edit3.SetEditText(quantity)

time.sleep(0.2)

self.__dialog_window.Button1.Click()

time.sleep(0.2)def __sell(self, code, quantity):"""卖函数

:param code: 股票代码， 字符串

:param quantity: 数量， 字符串"""self.__dialog_window.Edit4.SetFocus()

time.sleep(0.2)

self.__dialog_window.Edit4.SetEditText(code)

time.sleep(0.2)if quantity != '0':

self.__dialog_window.Edit6.SetEditText(quantity)

time.sleep(0.2)

self.__dialog_window.Button2.Click()

time.sleep(0.2)def __closePopupWindow(self):"""关闭一个弹窗。

:return: 如果有弹出式对话框，返回True，否则返回False"""popup_hwnd= self.__main_window.PopupWindow()ifpopup_hwnd:

popup_window= self.__app.window_(handle=popup_hwnd)

popup_window.SetFocus()

popup_window.Button.Click()returnTruereturnFalsedef __closePopupWindows(self):"""关闭多个弹出窗口

:return:"""

while self.__closePopupWindow():

time.sleep(0.5)deforder(self, code, direction, quantity):"""下单函数

:param code: 股票代码， 字符串

:param direction: 买卖方向， 字符串

:param quantity: 买卖数量， 字符串"""

if direction == 'B':

self.__buy(code, quantity)if direction == 'S':

self.__sell(code, quantity)

self.__closePopupWindows()defmaxWindow(self):"""最大化窗口"""

if self.__main_window.GetShowState() != 3:

self.__main_window.Maximize()

self.__main_window.SetFocus()defminWindow(self):"""最小化窗体"""

if self.__main_window.GetShowState() != 2:

self.__main_window.Minimize()def refresh(self, t=0.5):"""点击刷新按钮

:param t:刷新后的等待时间"""self.__dialog_window.Button5.Click()

time.sleep(t)defgetMoney(self):"""获取可用资金"""

return float(self.__dialog_window.Static19.WindowText())

@staticmethoddef __cleanClipboardData(data, cols=11):"""清洗剪贴板数据

:param data: 数据

:param cols: 列数

:return: 清洗后的数据，返回列表"""lst= data.strip().split()[:-1]

matrix=[]for i in range(0, len(lst) //cols):

matrix.append(lst[i* cols:(i &