support self restart assigned private repos

modules/restart.py

@@ -1,6 +1,6 @@
 import requests
 from modules.smtp import send_email
-from utils import HF_DOMAIN, MS_DOMAIN, HF_HEADER, MS_HEADER, TIMEOUT
+from utils import HF_DOMAIN, MS_DOMAIN, HF_HEADER, MS_HEADER, TIMEOUT, REPOS
 
 
 def restart_private_studio(repo: str):
@@ -25,7 +25,15 @@ def restart_private_space(repo: str):
 
 def self_restart():
     try:
+        if REPOS:
+            repos = REPOS.split(";")
+            for repo in repos:
+                private_repo = repo.strip()
+                print(f"Restarting {private_repo}...")
+                restart_private_space(private_repo)
+
         restart_private_space("kakamond/keeps_alive")
+
     except Exception as e:
         send_email(f"Failed to self-restart: {e}")
 

utils.py

@@ -2,6 +2,7 @@ import os
 
 PERIOD = os.getenv("period")
 USERS = os.getenv("users")
+REPOS = os.getenv("repos")
 API = os.getenv("api")
 EMAIL = os.getenv("email")
 SMTP = os.getenv("smtp")