"""
|
同花顺软件工具
|
"""
|
# 打开交易环境
|
import time
|
|
import win32api
|
import win32con
|
import win32gui
|
|
from db import redis_manager
|
from db.redis_manager import RedisUtils
|
from utils import tool
|
from trade import trade_gui
|
|
__redisManager = redis_manager.RedisManager(2)
|
|
|
def __click(hwnd):
|
win32gui.SendMessage(hwnd, win32con.WM_LBUTTONDOWN, 0, 0)
|
win32gui.SendMessage(hwnd, win32con.WM_LBUTTONUP, 0, 0)
|
|
|
def __input_number(hwnd, num_str):
|
# delete
|
for c in num_str:
|
code = -1
|
lp = 0
|
if c == '.':
|
code = 110
|
win32gui.SendMessage(hwnd, win32con.WM_KEYDOWN, code, 0)
|
win32gui.PostMessage(hwnd, win32con.WM_KEYUP, code, 0)
|
continue
|
elif c == '0':
|
code = 48
|
elif c == '1':
|
code = 49
|
elif c == '2':
|
code = 50
|
elif c == '3':
|
code = 51
|
elif c == '4':
|
code = 52
|
elif c == '5':
|
code = 53
|
elif c == '6':
|
code = 54
|
elif c == '7':
|
code = 55
|
elif c == '8':
|
code = 56
|
elif c == '9':
|
code = 57
|
win32gui.SendMessage(hwnd, win32con.WM_KEYDOWN, code, 0)
|
win32gui.PostMessage(hwnd, win32con.WM_KEYUP, code, 0)
|
|
|
def show_window(hwnd):
|
win32gui.ShowWindow(hwnd, win32con.SW_SHOWNORMAL)
|
|
|
def open_trade_gui():
|
main_win = __open_main_win()
|
show_window(main_win)
|
win_delegate = __open_trade_delegate_win(main_win)
|
|
# 最大化委托
|
win_temp = win32gui.FindWindowEx(win_delegate, None, None, "HexinScrollWnd")
|
left, top, right, bottom = win32gui.GetWindowRect(win_temp)
|
height_1 = bottom - top
|
win_temp = win32gui.FindWindowEx(win_delegate, win_temp, None, "HexinScrollWnd")
|
left, top, right, bottom = win32gui.GetWindowRect(win_temp)
|
height_2 = bottom - top
|
if height_1 > 10 and height_2 > 10:
|
maxwin = win32gui.GetDlgItem(win_delegate, 0x00002EF1)
|
# 点击按钮
|
__click(maxwin)
|
|
# 打开闪电买入
|
__open_buy_win(main_win)
|
|
|
def __open_main_win():
|
# 打开同花顺主页
|
hWndList = []
|
win32gui.EnumWindows(lambda hWnd, param: param.append(hWnd), hWndList)
|
for hwnd in hWndList:
|
_title = win32gui.GetWindowText(hwnd)
|
if _title.__contains__("同花顺("):
|
if not win32gui.IsWindowVisible(hwnd):
|
show_window(hwnd)
|
win32gui.SetForegroundWindow(hwnd)
|
win32gui.SetFocus(hwnd)
|
return hwnd
|
raise Exception("尚未打开同花顺")
|
|
|
# 打开闪电买入窗口
|
def __open_buy_win(hwnd):
|
show_window(hwnd)
|
|
wins = trade_gui.THSGuiTrade.get_buy_wins()
|
if len(wins) < 1:
|
try:
|
win32gui.SetForegroundWindow(hwnd)
|
except:
|
pass
|
left, top, right, bottom = win32gui.GetWindowRect(hwnd)
|
x = int(round((left + right) / 2, 0))
|
y = int(round(top + 2, 0))
|
win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN | win32con.MOUSEEVENTF_LEFTUP, x, y, 0, 0)
|
|
# 输入21
|
time.sleep(0.1)
|
|
__input_number(hwnd, "21")
|
# 输入enter
|
win32api.keybd_event(win32con.VK_RETURN, 0, 0, 0)
|
time.sleep(1)
|
wins = trade_gui.THSGuiTrade.get_buy_wins()
|
if len(wins) < 1:
|
raise Exception("打开闪电买入失败")
|
|
if len(wins) < 3:
|
for i in range(0, 3):
|
btn = win32gui.GetDlgItem(wins[len(wins) - 1], 0x000005ED)
|
# 点击事件添加
|
__click(btn)
|
time.sleep(0.5)
|
wins = trade_gui.THSGuiTrade.get_buy_wins()
|
|
pass
|
|
|
# 获取专业版下单句柄
|
def __get__trade_delegate_win_profession():
|
hWndList = []
|
win32gui.EnumWindows(lambda hWnd, param: param.append(hWnd), hWndList)
|
for hwnd in hWndList:
|
_title = win32gui.GetWindowText(hwnd)
|
if _title.__contains__("专业版下单"):
|
return hwnd
|
return None
|
|
|
# 打开交易委托
|
def __open_trade_delegate_win(hwnd1):
|
win32gui.SetForegroundWindow(hwnd1)
|
# 按下F12
|
win32api.keybd_event(win32con.VK_F12, 0, 0, 0)
|
# 检测交易窗口是否打开
|
for i in range(0, 5):
|
hWndList = []
|
win32gui.EnumWindows(lambda hWnd, param: param.append(hWnd), hWndList)
|
for hwnd in hWndList:
|
_title = win32gui.GetWindowText(hwnd)
|
if _title.__contains__("专业版下单"):
|
if not win32gui.IsWindowVisible(hwnd):
|
win32gui.SetForegroundWindow(hwnd)
|
return hwnd
|
elif _title.__contains__("网上股票交易系统"):
|
show_window(hwnd)
|
time.sleep(0.2)
|
hwnd = win32gui.FindWindowEx(hwnd, None, "ToolbarWindow32", None)
|
hwnd = win32gui.FindWindowEx(hwnd, None, "#32770", None)
|
hwnd = win32gui.FindWindowEx(hwnd, None, "Button", "专业")
|
# 点击按钮
|
__click(hwnd)
|
win32gui.EnumWindows(lambda hWnd, param: param.append(hWnd), hWndList)
|
for j in range(0, 5):
|
win = __get__trade_delegate_win_profession()
|
if win is not None:
|
return win
|
time.sleep(1)
|
time.sleep(1)
|
raise Exception("专业版下单打开失败")
|
|
|
def set_ths_dead_state(client_id, dead):
|
redis = __redisManager.getRedis()
|
key = "ths_state_dead_count-{}".format(client_id)
|
if not dead:
|
RedisUtils.setex(redis, key, tool.get_expire(), 0)
|
else:
|
RedisUtils.incrby(redis, key, 1)
|
RedisUtils.expire(redis, key, tool.get_expire())
|
|
|
# 同花顺是否卡死
|
def is_ths_dead(client_id):
|
key = "ths_state_dead_count-{}".format(client_id)
|
redis = __redisManager.getRedis()
|
val = RedisUtils.get(redis, key)
|
if val is not None and int(val) >= 5:
|
return True
|
else:
|
return False
|
|
|
if __name__ == "__main__":
|
open_trade_gui()
|
|
pass
|
