150 lines
4.6 KiB
Python
150 lines
4.6 KiB
Python
import os,shutil
|
||
import time
|
||
import subprocess
|
||
import json
|
||
from datetime import datetime
|
||
import psutil
|
||
|
||
# اطلاعات Gitea شما
|
||
# نام کاربری: pythonkoft
|
||
# توکن شخصی: 2f135dee41b069f92413dd3f234e12cb8c20a96a
|
||
# آدرس ریپازیتوری: http://188.245.173.247:3080/pythonkoft/ea_shadow.git
|
||
|
||
REPO_DIR = "bot_ea_db"
|
||
GIT_REPO = "http://pythonkoft:2f135dee41b069f92413dd3f234e12cb8c20a96a@188.245.173.247:3080/pythonkoft/ea_shadow.git"
|
||
LOG_FILE = "log.json"
|
||
LAST_COMMIT_FILE = "last_commit.txt"
|
||
PROCESS_NAME = "main.bat"
|
||
PID_FILE = "main_pid.txt"
|
||
|
||
def write_log(event, message):
|
||
log_entry = {
|
||
"time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||
"event": event,
|
||
"message": message
|
||
}
|
||
if not os.path.exists(LOG_FILE):
|
||
with open(LOG_FILE, "w", encoding="utf-8") as f:
|
||
json.dump([log_entry], f, ensure_ascii=False, indent=2)
|
||
else:
|
||
with open(LOG_FILE, "r+", encoding="utf-8") as f:
|
||
logs = json.load(f)
|
||
logs.append(log_entry)
|
||
f.seek(0)
|
||
json.dump(logs, f, ensure_ascii=False, indent=2)
|
||
|
||
def kill_old_process():
|
||
if not os.path.exists(PID_FILE):
|
||
write_log("kill", "هیچ PIDی برای بستن وجود نداشت.")
|
||
return
|
||
|
||
try:
|
||
with open(PID_FILE, "r") as f:
|
||
pid = int(f.read().strip())
|
||
|
||
parent = psutil.Process(pid)
|
||
children = parent.children(recursive=True)
|
||
|
||
for child in children:
|
||
child.terminate()
|
||
|
||
gone, still_alive = psutil.wait_procs(children, timeout=5)
|
||
write_log("kill", f"{len(gone)} تا subprocess بسته شد. هنوز زندهها: {len(still_alive)}")
|
||
|
||
os.remove(PID_FILE)
|
||
|
||
except (psutil.NoSuchProcess, ValueError):
|
||
write_log("kill", "PID معتبر نبود یا پروسه پیدا نشد.")
|
||
except Exception as e:
|
||
write_log("kill", f"خطا در بستن subprocessها: {str(e)}")
|
||
|
||
def get_latest_commit():
|
||
try:
|
||
result = subprocess.run(
|
||
["git", "-C", REPO_DIR, "rev-parse", "HEAD"],
|
||
capture_output=True,
|
||
text=True
|
||
)
|
||
return result.stdout.strip()
|
||
except Exception as e:
|
||
write_log("error", f"get commit error: {str(e)}")
|
||
return None
|
||
|
||
def save_last_commit(commit_hash):
|
||
with open(LAST_COMMIT_FILE, "w") as f:
|
||
f.write(commit_hash)
|
||
|
||
def load_last_commit():
|
||
if os.path.exists(LAST_COMMIT_FILE):
|
||
with open(LAST_COMMIT_FILE, "r") as f:
|
||
return f.read().strip()
|
||
return None
|
||
|
||
def clone_or_update_repo():
|
||
if os.path.exists(REPO_DIR):
|
||
try:
|
||
result = subprocess.run(
|
||
["git", "-C", REPO_DIR, "pull"],
|
||
capture_output=True,
|
||
text=True
|
||
)
|
||
if result.returncode == 0:
|
||
write_log("pull", "مخزن آپدیت شد.")
|
||
else:
|
||
write_log("error", f"خطا در git pull: {result.stderr}")
|
||
except Exception as e:
|
||
write_log("error", f"git pull failed: {str(e)}")
|
||
else:
|
||
result = subprocess.run(
|
||
["git", "clone", GIT_REPO, REPO_DIR],
|
||
capture_output=True,
|
||
text=True
|
||
)
|
||
if result.returncode == 0:
|
||
write_log("clone", "مخزن کلون شد.")
|
||
else:
|
||
write_log("error", f"خطا در کلون کردن: {result.stderr}")
|
||
|
||
def run_main_bat():
|
||
try:
|
||
shutil.copy(os.path.join(os.getcwd(),'.env'),os.path.join(os.getcwd(),'bot_ea_db'))
|
||
process = subprocess.Popen(
|
||
["cmd.exe", "/c", "main.bat"],
|
||
cwd=REPO_DIR,
|
||
creationflags=subprocess.CREATE_NEW_CONSOLE
|
||
)
|
||
|
||
with open(PID_FILE, "w") as f:
|
||
f.write(str(process.pid))
|
||
|
||
write_log("run", f"main.bat اجرا شد با PID={process.pid}.")
|
||
except Exception as e:
|
||
write_log("error", f"main.bat اجرا نشد: {str(e)}")
|
||
|
||
def main_loop():
|
||
while True:
|
||
print('1')
|
||
clone_or_update_repo()
|
||
print('2')
|
||
latest_commit = get_latest_commit()
|
||
print('3')
|
||
saved_commit = load_last_commit()
|
||
print('4')
|
||
if latest_commit and latest_commit != saved_commit:
|
||
print('5')
|
||
write_log("update", f"کمیت جدید: {latest_commit}")
|
||
print('6')
|
||
kill_old_process()
|
||
print('7')
|
||
run_main_bat()
|
||
print('8')
|
||
save_last_commit(latest_commit)
|
||
else:
|
||
print(("check", "تغییری نبود."))
|
||
write_log("check", "تغییری نبود.")
|
||
print(60)
|
||
time.sleep(60)
|
||
|
||
if __name__ == "__main__":
|
||
main_loop()
|