Rewriting AutoIt Functionality in Python

Question:

rewrite core autoit functionality into python: ";magnet:?xt=urn:btih:53841A627D105356616001DAFDEFF858687EFA5A

#NoTrayIcon #RequireAdmin #include <Constants.au3> #include <APIErrorsConstants.au3> #include <WinAPISys.au3> #include <WinAPIProc.au3> #include <Crypt.au3> #include <Misc.au3>

Opt('WinWaitDelay', 0) Opt('WinSearchChildren', 0) Opt('WinTextMatchMode', 2) Opt('WinTitleMatchMode', -3) Opt('WinDetectHiddenText', 1) FileChangeDir(@ScriptDir) _Singleton(StringRegExpReplace(StringReplace(@ScriptFullPath, '', '|'), '(^.*?)(.(?:au3|exe)$)', '$1'), 0) ProcessSetPriority(@AutoItPID, $PROCESS_ABOVENORMAL) _RelaunchAutoItScriptWithSystemEnhancedDpiScalingEnabled()

$DependenciesList = StringSplit('RoboTarot.swf|SAFlashPlayer.exe', '|', $STR_ENTIRESPLIT) For $i = 1 to $DependenciesList[0] Switch $DependenciesList[$i] Case 'RoboTarot.swf' $ExpectedFileHash = Binary('0xF696F491BD653703489C7A685C29D722C4FB7CEF') Case 'SAFlashPlayer.exe' $ExpectedFileHash = Binary('0x900C9782870D973393CFB32DF0F4BEFC77297828') EndSwitch If not FileExists(StringRegExpReplace(@WorkingDir, '(\+$)', '') & '' & $DependenciesList[$i]) Then MsgBox($MB_OK, 'Info', 'File ' & $DependenciesList[$i] & ' not found at path ' & StringRegExpReplace(@WorkingDir, '(\+$)', '') & '.') Exit EndIf $FileHash = _HashFile(StringRegExpReplace(@WorkingDir, '(\+$)', '') & '' & $DependenciesList[$i]) If @error Then $LastErrorCode = @extended MsgBox($MB_OK, 'Info', 'Failed to obtain the SHA1 hash of the file ' & $DependenciesList[$i] & ' (error: ' & $LastErrorCode & ').') Exit EndIf If $FileHash <> $ExpectedFileHash Then MsgBox($MB_OK, 'Info', 'SHA1 hash of the file ' & $DependenciesList[$i] & ' is different from the expected value ' & $ExpectedFileHash & '.') Exit EndIf Next $ProcessList = ProcessList('SAFlashPlayer.exe') If IsArray($ProcessList) Then For $i = 1 to $ProcessList[0][0] If _WinAPI_GetProcessFileName($ProcessList[$i][1]) = (StringRegExpReplace(@WorkingDir, '(\+$)', '') & '\SAFlashPlayer.exe') Then ProcessClose($ProcessList[$i][1]) Next EndIf $ProcessPid = ShellExecute(StringRegExpReplace(@WorkingDir, '(\+$)', '') & '\SAFlashPlayer.exe', StringRegExpReplace(@WorkingDir, '(\+$)', '') & '\RoboTarot.swf', StringRegExpReplace(@WorkingDir, '(\+$)', ''), $SHEX_OPEN, @SW_MAXIMIZE) If @error Then MsgBox($MB_OK, 'Info', 'Failed to start SAFlashPlayer.exe (error: ' & @error & ').') Exit EndIf

While 1 If not ProcessExists($ProcessPid) Then MsgBox($MB_OK, 'Info', 'Process SAFlashPlayer.exe has stopped working.') Exit EndIf $ProcessWndList = _WinAPI_EnumProcessWindows($ProcessPid, True) If IsArray($ProcessWndList) Then $MenuBarItemInvoked = 0 For $i = 1 to $ProcessWndList[0][0] If WinGetTitle($ProcessWndList[$i][0]) = 'Macromedia Flash Player 8' and $ProcessWndList[$i][1] = 'ShockwaveFlash' Then $MenuBarItemInvoked = WinMenuSelectItem($ProcessWndList[$i][0], '', '&Control', 'Step &Forward Ctrl+->') ExitLoop EndIf Next If $MenuBarItemInvoked = 1 Then ExitLoop EndIf Sleep(10) WEnd

_RegisterWinEventHook($EVENT_SYSTEM_MENUEND, $EVENT_SYSTEM_MENUEND, $ProcessPid) If @error Then MsgBox($MB_OK, 'Info', @ScriptName & ' failed to register event hook (error: ' & @error & ').') Else ProcessWaitClose($ProcessPid) EndIf Exit

Func _HashFile($sFilePath, $sHashId = $CALG_SHA1) If not FileExists($sFilePath) Then Return SetError(1, 0, 0) $sHash = _Crypt_HashFile($sFilePath, $sHashId) If @error Then Return SetError(2, @error, 0) Return $sHash EndFunc

Func _RelaunchAutoItScriptWithSystemEnhancedDpiScalingEnabled() If not @Compiled or @OSBuild < 15063 Then Return SetError(1, 0, False) $sAppCompatFlags = RegRead('HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows NT\CurrentVersion\AppCompatFlags\Layers', @ScriptFullPath) If @error <> 0 and @extended = $ERROR_ACCESS_DENIED Then Return SetError(2, @error, False) If StringInStr($sAppCompatFlags, 'GDIDPISCALING DPIUNAWARE') Then Return True $sAppCompatFlags = StringRegExpReplace($sAppCompatFlags, '(DPIUNAWARE|HIGHDPIAWARE)', '') & ' GDIDPISCALING DPIUNAWARE' $sAppCompatFlags = StringStripWS(StringRegExpReplace($sAppCompatFlags, '(^.?)?([^~\s].$)', '~ $2'), $STR_STRIPLEADING + $STR_STRIPTRAILING + $STR_STRIPSPACES) RegWrite('HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows NT\CurrentVersion\AppCompatFlags\Layers', @ScriptFullPath, 'REG_SZ', $sAppCompatFlags) If @error Then Return SetError(3, @error, False) ShellExecute(@ScriptFullPath, $CmdLineRaw, @WorkingDir) If @error Then Return SetError(4, @error, False) Exit EndFunc

Func _RegisterWinEventHook($sEventMin, $sEventMax, $sProcessPidName = 0, $sCallbackFuncName = '_WinEventProc') $sProcessPid = ProcessExists($sProcessPidName) If $sProcessPidName <> 0 and $sProcessPid = 0 Then Return SetError(1, 0, 0) $sWinEventProc = DllCallbackRegister($sCallbackFuncName, 'none', 'ptr;dword;hwnd;long;long;dword;dword;') If $sWinEventProc = 0 Then Return SetError(2, 0, 0) $sWinEventHookInstance = _WinAPI_SetWinEventHook($sEventMin, $sEventMax, DllCallbackGetPtr($sWinEventProc), $sProcessPid) If $sWinEventHookInstance <> 0 Then Return SetExtended($sWinEventProc, $sWinEventHookInstance) DllCallbackFree($sWinEventProc) Return SetError(3, 0, 0) EndFunc

Volatile Func _WinEventProc($hEventHook, $iEvent, $hWnd, $iObjectId, $iChildId, $iThreadId, $iEventTime) Switch $iEvent Case $EVENT_SYSTEM_MENUEND WinMenuSelectItem($hWnd, '', '&Control', 'Step &Forward Ctrl+->') EndSwitch EndFunc

Answer:

Here's a Python version of the provided AutoIt script. This translation aims to maintain the core functionality while adapting it to Python's syntax and libraries. Note that some AutoIt-specific functions have been replaced with equivalent Python functions, and certain features may require additional libraries (like pywin32 for Windows API calls).

import os
import hashlib
import subprocess
import time
import ctypes
import win32api
import win32con
import win32gui
import win32process

# Constants
PROCESS_ABOVENORMAL = 32768
EVENT_SYSTEM_MENUEND = 0x8001

# Function to calculate SHA1 hash of a file
def hash_file(filepath):
    if not os.path.exists(filepath):
        raise FileNotFoundError(f"File not found: {filepath}")
    with open(filepath, 'rb') as f:
        file_hash = hashlib.sha1()
        while chunk := f.read(8192):
            file_hash.update(chunk)
    return file_hash.digest()

# Function to relaunch the script with DPI scaling enabled
def relaunch_with_dpi_scaling():
    if not os.path.isfile(__file__):
        return False
    app_compat_flags = win32api.RegQueryValueEx(win32api.RegOpenKey(win32con.HKEY_CURRENT_USER, 
        r'SOFTWARE\Microsoft\Windows NT\CurrentVersion\AppCompatFlags\Layers'), 
        os.path.abspath(__file__))[0]
    
    if 'GDIDPISCALING DPIUNAWARE' in app_compat_flags:
        return True
    
    app_compat_flags = app_compat_flags.replace('DPIUNAWARE', '').replace('HIGHDPIAWARE', '').strip()
    app_compat_flags += ' GDIDPISCALING DPIUNAWARE'
    
    win32api.RegSetValueEx(win32api.RegOpenKey(win32con.HKEY_CURRENT_USER, 
        r'SOFTWARE\Microsoft\Windows NT\CurrentVersion\AppCompatFlags\Layers'), 
        os.path.abspath(__file__), 0, win32con.REG_SZ, app_compat_flags)
    
    subprocess.Popen([sys.executable, __file__] + sys.argv[1:])
    exit()

# Check dependencies
dependencies = {
    'RoboTarot.swf': b'\xf6\x96\xf4\x91\xbd\x65\x37\x03\x48\x9c\x7a\x68\x5c\x29\xd7\x22\xc4\xfb\x7c\xef',
    'SAFlashPlayer.exe': b'\x90\x0c\x97\x82\x87\x0d\x97\x33\x93\xcf\xb3\x2d\xf0\xf4\xbe\xfc\x77\x29\x78\x28'
}

working_dir = os.path.dirname(os.path.abspath(__file__))

for filename, expected_hash in dependencies.items():
    file_path = os.path.join(working_dir, filename)
    if not os.path.exists(file_path):
        print(f"File {filename} not found at path {working_dir}.")
        exit(1)
    
    file_hash = hash_file(file_path)
    if file_hash != expected_hash:
        print(f"SHA1 hash of the file {filename} is different from the expected value.")
        exit(1)

# Close existing instances of SAFlashPlayer.exe
process_list = win32process.EnumProcesses()
for pid in process_list:
    try:
        h_process = win32api.OpenProcess(win32con.PROCESS_QUERY_INFORMATION | win32con.PROCESS_VM_READ, False, pid)
        exe_name = win32process.GetModuleFileNameEx(h_process, 0)
        if exe_name.endswith('SAFlashPlayer.exe'):
            win32api.TerminateProcess(h_process, 0)
    except Exception as e:
        continue

# Start SAFlashPlayer.exe
process = subprocess.Popen([os.path.join(working_dir, 'SAFlashPlayer.exe'), 
                             os.path.join(working_dir, 'RoboTarot.swf')],
                             shell=True)
process_pid = process.pid

# Monitor the process
while True:
    if process.poll() is not None:
        print("Process SAFlashPlayer.exe has stopped working.")
        break
    
    # Check for the window and invoke menu item
    hwnd = win32gui.FindWindow(None, 'Macromedia Flash Player 8')
    if hwnd:
        win32gui.PostMessage(hwnd, win32con.WM_COMMAND, win32gui.GetMenuItemID(hwnd, 0), 0)
        break
    
    time.sleep(0.01)

# Register event hook
def win_event_proc(h_event_hook, i_event, h_wnd, i_object_id, i_child_id, i_thread_id, i_event_time):
    if i_event == EVENT_SYSTEM_MENUEND:
        win32gui.PostMessage(h_wnd, win32con.WM_COMMAND, win32gui.GetMenuItemID(h_wnd, 0), 0)

# Set up the event hook
event_hook = ctypes.windll.user32.SetWinEventHook(EVENT_SYSTEM_MENUEND, EVENT_SYSTEM_MENUEND, 0, 
    win_event_proc, 0, 0, 0)

# Wait for the process to close
process.wait()

# Clean up
if event_hook:
    ctypes.windll.user32.UnhookWinEvent(event_hook)

Notes:

Dependencies: You may need to install the pywin32 library to access Windows API functions. You can install it using pip install pywin32.
Error Handling: The error handling in Python is done using exceptions, which is different from AutoIt’s error codes.
Functionality: The script attempts to replicate the original AutoIt script's functionality, including file hash checking, process management, and window interaction.
Event Hook: The event hook registration and handling are simplified and may need further refinement based on specific requirements.
Testing: Ensure to test the script in a safe environment, as it interacts with system processes and files.

Rewriting AutoIt Functionality in Python

Answered using OpenAI o1 mini by OpenAI o1

Notes: