e-Gov電子申請のコツ

import os
import time
import urllib.parse
import pythoncom
import wmi
import win32api
import win32gui
import win32con
import win32process
import win32com.client

def find_toplevel_windows(pid):
    def callback(hwnd, l):
        _, p = win32process.GetWindowThreadProcessId(hwnd)
        if p == pid: l.append(hwnd)
        return True
    toplevel_windows = []
    win32gui.EnumWindows(callback, toplevel_windows)
    return toplevel_windows

def find_ie_windows(toplevel_windows):
    def callback(hwnd, l):
        classname = win32gui.GetClassName(hwnd)
        if classname == 'Internet Explorer_Server': l.append(hwnd)
        return True
    ie_windows = []
    for toplevel_window in toplevel_windows:
        win32gui.EnumChildWindows(toplevel_window, callback, ie_windows)
    return ie_windows

def get_html_document(ie_window):
    msg = win32gui.RegisterWindowMessage('WM_HTML_GETOBJECT')
    _, result = win32gui.SendMessageTimeout(ie_window, msg, 0, 0, win32con.SMTO_ABORTIFHUNG, 1000)
    ob = pythoncom.ObjectFromLresult(result, pythoncom.IID_IDispatch, 0)
    return win32com.client.dynamic.Dispatch(ob)

if __name__ == '__main__':
    app_path = r'C:\Program Files (x86)\eGovClient\eGovClient.exe'
    gbiz_username = 'xxxx'
    gbiz_password = 'xxxx'

    # 起動
    win32api.WinExec(app_path)
    time.sleep(3) # 起動待ち

    # PID取得
    c = wmi.WMI()
    rows = c.query(f'SELECT ProcessId FROM Win32_Process WHERE Name = "{os.path.basename(app_path)}"')
    if len(rows) != 1:
        print('Error', 'Unexpected number of processes:', len(rows))
        exit(1)
    pid = rows[0].ProcessId

    # "Internet Explorer_Server" を探す
    toplevel_windows = find_toplevel_windows(pid)
    ie_windows = find_ie_windows(toplevel_windows)
    if len(ie_windows) != 1:
        print('Error', 'Unexpected number of ie_windows:', len(ie_windows))
        exit(1)
    ie_window = ie_windows[0]

    # ウィンドウサイズを変えておく
    for toplevel_window in toplevel_windows:
        if 'e-Govアカウントログイン' in win32gui.GetWindowText(toplevel_window):
            left, top, right, bottom = win32gui.GetWindowRect(toplevel_window)
            width = right - left
            height = bottom - top
            win32gui.MoveWindow(toplevel_window, left, top, width//2, height, True)

    # 「GビズIDでログイン」を選択
    # querySelectorもgetElementByIdも使えないので頑張る
    # https://stackoverflow.com/questions/39205598/getelementbyid-takes-exactly-1-argument-2-given
    doc = get_html_document(ie_window)
    parsedUrl = urllib.parse.urlparse(doc.URL)
    if not (parsedUrl.netloc == 'account.e-gov.go.jp' and parsedUrl.path == '/auth/auth'):
        print('Error', 'Unexpected URL (eGov):', doc.URL)
        exit(1)
    for e in doc.documentElement.getElementsByTagName('li'):
        if e.id == 'gbizid':
            e.getElementsByTagName('a')[0].click()
            time.sleep(3) # 遷移待ち
            break

    # GビズID入力
    # URLが想定外（遷移できなかった or すでに二段階認証に行った）なら何もしない
    doc = get_html_document(ie_window)
    parsedUrl = urllib.parse.urlparse(doc.URL)
    if parsedUrl.netloc == 'gbiz-id.go.jp' and parsedUrl.path == '/oauth/login':
        elements = {'username': None, 'password': None, 'submit': None}
        for e in doc.documentElement.getElementsByTagName('input'):
            if e.id == 'username': elements['username'] = e
            if e.id == 'password': elements['password'] = e
            if e.type == 'submit': elements['submit'] = e
        if all([ x is not None for x in elements.values()]):
            elements['username'].value = gbiz_username
            elements['password'].value = gbiz_password
            elements['submit'].click()