import time
# 获取同花顺副屏幕1句柄
import win32api
import win32con

import win32_util
import win32gui


def get_ths_second_screen_menu_hwnd():
    hwnds = win32_util.search_window("副屏1")
    if hwnds:
        hwnd = hwnds[0]
        hwnd = win32gui.FindWindowEx(hwnd, None, "AfxFrameOrView100s", None)
        temp = None
        for i in range(0, 100):
            temp = win32gui.FindWindowEx(hwnd, temp, "AfxWnd100s", None)
            if win32gui.IsWindowVisible(temp):
                left, top, right, bottom = win32gui.GetWindowRect(temp)
                if bottom - top > right - left:
                    break
        if temp is None:
            return None
        hwnd = temp
        hwnd = win32gui.FindWindowEx(hwnd, None, "#32770", None)
        hwnd = win32gui.FindWindowEx(hwnd, None, "block_list_page", None)
        if hwnd:
            return hwnd

    return None


def get_trade_refesh_hwnd():
    hwnds = win32_util.search_window("网上股票交易系统")
    if hwnds:
        for hwnd in hwnds:
            content_hwnd = win32gui.FindWindowEx(hwnd, None, "AfxMDIFrame140s", None)
            temp = win32gui.FindWindowEx(content_hwnd, None, "#32770", None)
            for i in range(10):
                if win32gui.IsWindowVisible(temp):
                    break
                temp = win32gui.FindWindowEx(content_hwnd, temp, "#32770", None)
            if not temp:
                continue
            t1 = win32gui.FindWindowEx(temp, None, None, "过滤")
            if not win32gui.IsWindowVisible(t1):
                continue

            tool_hwnd = win32gui.FindWindowEx(hwnd, None, "ToolbarWindow32", None)
            temp = win32gui.FindWindowEx(tool_hwnd, None, "#32770", None)
            if temp:
                nq = win32gui.FindWindowEx(temp, None, None, "内嵌")
                if nq and win32gui.IsWindowVisible(nq):
                    continue
            return tool_hwnd
    return None


# 获取交易句柄
def get_trade_hwnd():
    hwnds = win32_util.search_window("网上股票交易系统")
    if hwnds:
        hwnd = hwnds[0]
        return hwnd
    return None


def get_flash_trade_hwnds():
    """
    获取闪电交易句柄
    :return:
    """
    buy_win_list = []
    hWndList = []
    win32gui.EnumWindows(lambda hWnd, param: param.append(hWnd), hWndList)
    for hwnd in hWndList:
        clsname = win32gui.GetClassName(hwnd)
        if clsname == '#32770' and win32gui.IsWindowVisible(hwnd):
            pos = win32gui.GetWindowRect(hwnd)
            width = pos[2] - pos[0]
            height = pos[3] - pos[1]
            if 500 > width > 100 and 500 > height > 50:
                # 查找确定按钮
                try:
                    buy_win = win32gui.GetDlgItem(hwnd, 0x000003EE)
                    if buy_win > 0:
                        win_name = win32_util.getText(buy_win)
                        if win_name == '一键买入[B]' or win_name == '一键卖出[S]':
                            buy_win_list.append(hwnd)
                except Exception as e:
                    print(e)
    return buy_win_list


def get_ths_main_content_hwnd():
    hwnds = win32_util.search_window("同花顺", max_count=1)
    if hwnds:
        hwnd = hwnds[0]
        child = None
        for i in range(0, 20):
            child = win32gui.FindWindowEx(hwnd, child, None, None)
            if child:
                left, top, right, bottom = win32gui.GetWindowRect(child)
                if right - left > 1000 and bottom - top > 800:
                    return child
    return None


__cache_hwnd = {}


def get_ths_main_content_hwnd_cached():
    """
    获取同花顺主页句柄
    :return:
    """
    target_hwnd = __cache_hwnd.get("ths_main_content_hwnd")
    if target_hwnd and win32_util.is_visible(target_hwnd):
        return target_hwnd
    target_hwnd = get_ths_main_content_hwnd()
    if target_hwnd:
        __cache_hwnd["ths_main_content_hwnd"] = target_hwnd
    return target_hwnd


# 批量点击事件
def betch_click(hwnd, ps, space_time=0.5):
    for p in ps:
        win32_util.visual_click(hwnd, p)
        time.sleep(space_time)


# 将代码加入自选
def input_code(hwnd, num_str):
    for c in num_str:
        code = -1
        if 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)
        time.sleep(0.1)


def add_code_to_zixuan(num_str):
    hwnds = win32_util.search_window("看盘页面")
    if hwnds:
        origin_hwnd = hwnds[0]
        for w in hwnds:
            title = win32_util.getText(w)
            if title.find("同花顺") >= 0:
                origin_hwnd = w
                break
        input_code(origin_hwnd, num_str[:1])
        # Afx:00400000:0
        hwnds = []
        try:
            hwnd = win32gui.GetDesktopWindow()
            temp = None
            while True:
                if temp and win32gui.IsWindowVisible(temp):
                    classname = win32gui.GetClassName(temp)
                    if classname and classname.find("Afx:") == 0 and classname.endswith(":0"):
                        hwnds.append(temp)
                temp = win32gui.FindWindowEx(hwnd, temp, None, None)
                if not temp:
                    break
        except:
            pass
        hwnd = win32gui.FindWindowEx(hwnds[0], None, "Edit", "")
        print(hwnd)
        input_code(hwnd, num_str[1:])
        time.sleep(0.5)
        win32gui.SendMessage(hwnds[0], win32con.WM_KEYDOWN, win32con.VK_RETURN, 0)
        win32gui.PostMessage(hwnds[0], win32con.WM_KEYUP, win32con.VK_RETURN, 0)
        # time.sleep(0.5)
        # win32gui.SetForegroundWindow(origin_hwnd)
        # win32api.keybd_event(win32con.VK_INSERT, 0, 0, 0)  # 按下键
        # win32api.keybd_event(win32con.VK_INSERT, 0, win32con.KEYEVENTF_KEYUP, 0)


if __name__ == "__main__":
    print(get_flash_trade_hwnds())