python/screen.py

import math

import win32con
import win32gui


def get_all_windows():
    windows = []
    win32gui.EnumWindows(lambda hwnd, windows: windows.append(hwnd), windows)
    return windows


def get_window_title(hwnd):
    return win32gui.GetWindowText(hwnd)


def find_chrome():
    windows = get_all_windows()
    target = []
    for hwnd in windows:
        title = get_window_title(hwnd)
        if "Google Chrome" in title:
            target.append(hwnd)
    return target


def arrange_master():
    chrome = find_chrome()
    hwnd = chrome[0]
    win32gui.SetWindowPos(hwnd, win32con.HWND_TOP, 0, 0, 500, 600, 0)
    win32gui.ShowWindow(hwnd, win32con.SW_SHOWNORMAL)
    win32gui.SetForegroundWindow(hwnd)


def arrange_slave():
    chrome = find_chrome()
    for i, hwnd in enumerate(chrome):
        if i > 5:
            return
        # width = math.ceil(i / 3)
        # height = math.floor(i / 2)
        win32gui.SetWindowPos(hwnd, win32con.HWND_TOP, i * 500, 0, 500, 600, 0)
        win32gui.ShowWindow(hwnd, win32con.SW_SHOWNORMAL)
        win32gui.SetForegroundWindow(hwnd)
    return len(chrome)


# arrange_master()