53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
|
import subprocess, os, sys, asyncio
|
||
|
|
|
||
|
|
pid_file = r'C:\Users\whdwo\OneDrive\바탕 화면\stockbot_v3\logs\bot.pid'
|
||
|
|
project_dir = r'C:\Users\whdwo\OneDrive\바탕 화면\stockbot_v3'
|
||
|
|
|
||
|
|
# 1-a) PID 파일로 종료
|
||
|
|
if os.path.exists(pid_file):
|
||
|
|
try:
|
||
|
|
pid = int(open(pid_file).read().strip())
|
||
|
|
subprocess.run(['taskkill', '/PID', str(pid), '/F'], capture_output=True)
|
||
|
|
print(f'PID 파일 종료: {pid}')
|
||
|
|
except Exception as e:
|
||
|
|
print(f'PID 파일 종료 실패: {e}')
|
||
|
|
os.remove(pid_file)
|
||
|
|
|
||
|
|
# 1-b) Get-CimInstance로 잔존 프로세스 스캔
|
||
|
|
r = subprocess.run(
|
||
|
|
['powershell', '-Command',
|
||
|
|
'Get-CimInstance Win32_Process | Where-Object { $_.CommandLine -like "*app/main.py*" -or $_.CommandLine -like "*app\\main.py*" } | Select-Object -ExpandProperty ProcessId'],
|
||
|
|
capture_output=True, text=True
|
||
|
|
)
|
||
|
|
pids = [p.strip() for p in r.stdout.strip().splitlines() if p.strip().isdigit()]
|
||
|
|
for pid in pids:
|
||
|
|
subprocess.run(['taskkill', '/PID', pid, '/F'], capture_output=True)
|
||
|
|
print(f'잔존 프로세스 종료: {pid}')
|
||
|
|
|
||
|
|
if not pids:
|
||
|
|
print('실행 중인 봇 없음 — 새로 시작합니다')
|
||
|
|
|
||
|
|
# 2. 봇 시작
|
||
|
|
os.chdir(project_dir)
|
||
|
|
proc = subprocess.Popen(
|
||
|
|
[sys.executable, 'app/main.py'],
|
||
|
|
creationflags=subprocess.DETACHED_PROCESS | subprocess.CREATE_NEW_PROCESS_GROUP,
|
||
|
|
stdout=open('logs/bot_stderr.log', 'a', encoding='utf-8'),
|
||
|
|
stderr=subprocess.STDOUT,
|
||
|
|
close_fds=True,
|
||
|
|
)
|
||
|
|
with open('logs/bot.pid', 'w') as f:
|
||
|
|
f.write(str(proc.pid))
|
||
|
|
print(f'봇 시작 완료 PID={proc.pid}')
|
||
|
|
|
||
|
|
# 3. Discord 알림
|
||
|
|
sys.path.insert(0, '.')
|
||
|
|
from app.main import load_env
|
||
|
|
load_env()
|
||
|
|
from app.monitor.notifier import send
|
||
|
|
mode = os.getenv('KIS_MOCK', 'true')
|
||
|
|
dry = os.getenv('DRY_RUN', 'true')
|
||
|
|
label = '[모의투자]' if mode == 'true' else '[실거래]'
|
||
|
|
asyncio.run(send(f'{label} 자동매매 봇 시작 (DRY_RUN={dry})'))
|
||
|
|
print('Discord 알림 전송 완료')
|