Skip to content

fix: 修复国金证券下单客户端更新后无法使用的问题 #27

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
skyformat99 wants to merge 1 commit into
base: master
Choose a base branch
from
Open

fix: 修复国金证券下单客户端更新后无法使用的问题 #27

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
40 changes: 29 additions & 11 deletions thsauto.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
refresh_sleep_time = 0.5
retry_time = 10

window_title = u'网上股票交易系统5.0'
window_title = u'国金全能行远航版'

def get_clipboard_data():
win32clipboard.OpenClipboard()
Expand Down Expand Up @@ -135,14 +135,24 @@ def handler(hwnd, results):
return False
enum_children(hwnd, results)
return len(results) == 0

def callback1(hwnd, _):
if win32gui.GetClassName(hwnd) == "#32770":
windows.append(hwnd)
return True

def callback2(hwnd, _):
if win32gui.GetClassName(hwnd) == "Static":
windows.append(hwnd)
return True

windows = []
win32gui.EnumWindows(callback1, None)
for window in windows:
win32gui.EnumChildWindows(window, callback2, None)


popups = []
windows = []
win32gui.EnumThreadWindows(tid, lambda hwnd, l: l.append(hwnd), windows)
for hwnd in windows:
if not handler(hwnd, popups):
break
for ctrl in popups:
for ctrl in windows:
text = get_text(ctrl)
if u"检测到您正在拷贝数据" in text:
return ctypes.windll.user32.GetWindow(ctrl, win32con.GW_HWNDNEXT)
Expand All @@ -162,6 +172,13 @@ def get_balance(self):
'code': 0, 'status': 'succeed',
'data': data,
}

def empty_clipboard(self):
win32clipboard.OpenClipboard()
try:
win32clipboard.EmptyClipboard()
finally:
win32clipboard.CloseClipboard()

def get_position(self):
self.switch_to_normal()
Expand All @@ -172,9 +189,9 @@ def get_position(self):
ctrl = win32gui.GetDlgItem(hwnd, 0x417)

self.copy_table(ctrl)

data = None
retry = 0
#self.empty_clipboard()
while not data and retry < retry_time:
retry += 1
time.sleep(sleep_time)
Expand Down Expand Up @@ -518,6 +535,7 @@ def copy_table(self, hwnd):
win32gui.SetForegroundWindow(hwnd)
os.system('echo off | clip')
hot_key(['ctrl', 'c'])
time.sleep(0.2)
self.input_ocr()

def input_ocr(self):
Expand All @@ -535,8 +553,8 @@ def input_ocr(self):

def capture_window(self, hwnd, file_name):
left, top, right, bottom = win32gui.GetWindowRect(hwnd)
width = right - left
height = bottom - top
width = right - left + 40
height = bottom - top + 10

hdc = win32gui.GetWindowDC(hwnd)
dc = win32ui.CreateDCFromHandle(hdc)
Expand Down