[2026-05-27] 포맷 후 복구 설치 스크립트 추가

2026-05-27 16:53:52 +09:00
parent 04577c63f1
commit 29db1bfcab
135 changed files with 2909 additions and 251 deletions
@@ -0,0 +1,113 @@
+"""
+워치독 태스크를 XML로 직접 등록
+관리자 권한 없이 현재 사용자 세션으로 등록
+"""
+import subprocess, sys, os, tempfile
+
+PROJECT = r'C:\Users\whdwo\OneDrive\바탕 화면\stockbot_v3'
+SCRIPT  = fr'{PROJECT}\scripts\run_watchdog.ps1'
+
+xml = f'''<?xml version="1.0" encoding="UTF-16"?>
+<Task version="1.3" xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task">
+  <RegistrationInfo>
+    <URI>\\StockBot\\StockBot_Watchdog</URI>
+  </RegistrationInfo>
+  <Principals>
+    <Principal id="Author">
+      <LogonType>InteractiveToken</LogonType>
+      <RunLevel>LeastPrivilege</RunLevel>
+    </Principal>
+  </Principals>
+  <Settings>
+    <DisallowStartIfOnBatteries>false</DisallowStartIfOnBatteries>
+    <StopIfGoingOnBatteries>false</StopIfGoingOnBatteries>
+    <ExecutionTimeLimit>PT3M</ExecutionTimeLimit>
+    <MultipleInstancesPolicy>IgnoreNew</MultipleInstancesPolicy>
+    <StartWhenAvailable>true</StartWhenAvailable>
+    <IdleSettings>
+      <Duration>PT10M</Duration>
+      <WaitTimeout>PT1H</WaitTimeout>
+      <StopOnIdleEnd>true</StopOnIdleEnd>
+      <RestartOnIdle>false</RestartOnIdle>
+    </IdleSettings>
+  </Settings>
+  <Triggers>
+    <CalendarTrigger>
+      <StartBoundary>2026-05-27T09:00:00+09:00</StartBoundary>
+      <Repetition>
+        <Interval>PT5M</Interval>
+        <Duration>PT6H10M</Duration>
+        <StopAtDurationEnd>true</StopAtDurationEnd>
+      </Repetition>
+      <ScheduleByWeek>
+        <WeeksInterval>1</WeeksInterval>
+        <DaysOfWeek>
+          <Monday />
+          <Tuesday />
+          <Wednesday />
+          <Thursday />
+          <Friday />
+        </DaysOfWeek>
+      </ScheduleByWeek>
+    </CalendarTrigger>
+  </Triggers>
+  <Actions Context="Author">
+    <Exec>
+      <Command>powershell.exe</Command>
+      <Arguments>-NonInteractive -ExecutionPolicy Bypass -File "{SCRIPT}"</Arguments>
+      <WorkingDirectory>{PROJECT}</WorkingDirectory>
+    </Exec>
+  </Actions>
+</Task>'''
+
+# UTF-16 LE BOM으로 임시 파일 저장 (Task Scheduler XML 요구사항)
+with tempfile.NamedTemporaryFile(suffix='.xml', delete=False, mode='wb') as f:
+    f.write(b'\xff\xfe')  # UTF-16 LE BOM
+    f.write(xml.encode('utf-16-le'))
+    tmp = f.name
+
+try:
+    # 1단계: XML로 등록
+    ps_cmd = f'Register-ScheduledTask -TaskName "StockBot_Watchdog" -TaskPath "\\\\StockBot\\\\" -Xml (Get-Content -Path \'{tmp}\' -Raw -Encoding Unicode) -Force | Out-Null; Write-Host "1단계 완료"'
+    r = subprocess.run(
+        ['powershell', '-NonInteractive', '-ExecutionPolicy', 'Bypass', '-Command', ps_cmd],
+        capture_output=True, text=True, encoding='utf-8', errors='replace'
+    )
+    print(r.stdout.strip())
+
+    # 2단계: 한글 경로 깨짐 교정 (setup_scheduler.ps1과 동일 방식)
+    # 등록된 태스크의 Arguments에서 garbled 문자열 추출 → 정상 한글로 치환 후 재등록
+    fix_cmd = r'''
+$task = Get-ScheduledTask -TaskName "StockBot_Watchdog" -TaskPath "\StockBot\"
+$stored = $task.Actions[0].Arguments
+if ($stored -match [regex]::Escape("바탕 화면")) {
+    Write-Host "경로 정상"
+} else {
+    $xml = Export-ScheduledTask -TaskName "StockBot_Watchdog" -TaskPath "\StockBot\"
+    if ($stored -match 'OneDrive\\(.+?)\\stockbot') {
+        $garbled = $Matches[1]
+        $fixedXml = $xml.Replace($garbled, "바탕 화면")
+        Register-ScheduledTask -TaskName "StockBot_Watchdog" -TaskPath "\StockBot\" -Xml $fixedXml -Force | Out-Null
+        Write-Host "한글 경로 교정 완료"
+    } else {
+        Write-Host "교정 패턴 미일치 — 수동 확인 필요"
+    }
+}
+'''
+    r2 = subprocess.run(
+        ['powershell', '-NonInteractive', '-ExecutionPolicy', 'Bypass', '-Command', fix_cmd],
+        capture_output=True, text=True, encoding='utf-8', errors='replace'
+    )
+    print(r2.stdout.strip())
+    if r2.returncode != 0:
+        print('STDERR:', r2.stderr.strip())
+
+    # 최종 확인
+    check = subprocess.run(
+        ['powershell', '-NonInteractive', '-ExecutionPolicy', 'Bypass', '-Command',
+         'Get-ScheduledTask -TaskPath "\\StockBot\\" | Format-Table TaskName, State -AutoSize'],
+        capture_output=True, text=True, encoding='utf-8', errors='replace'
+    )
+    print(check.stdout)
+finally:
+    os.unlink(tmp)