simplify envs

modules/restart.py

@@ -1,15 +1,7 @@
 import requests
 from huggingface_hub import HfApi
 from modules.smtp import send_email
-from utils import (
-    HF_DOMAIN,
-    MS_DOMAIN,
-    HF_TK,
-    MS_HEADER,
-    TIMEOUT,
-    PUB_REPOS,
-    PRI_REPOS,
-)
+from utils import HF_DOMAIN, MS_DOMAIN, HF_TK, MS_HEADER, TIMEOUT, REPOS
 
 HF_API = HfApi(token=HF_TK)
 
@@ -25,8 +17,8 @@ def restart_private_studio(repo: str):
 
 
 def restart_repos():
-    if PUB_REPOS:
-        repos = PUB_REPOS.split(";")
+    if REPOS:
+        repos = REPOS.split(";")
         for repo in repos:
             private_repo = repo.strip()
             print(f"Restarting {private_repo} space & studio...")
@@ -34,12 +26,14 @@ def restart_repos():
             HF_API.restart_space(private_repo)
 
 
-def self_restart():
+def self_restart(me="kakamond/keeps_alive"):
     try:
-        if PRI_REPOS:
-            repos = PRI_REPOS.split(";")
-            for repo in repos:
-                HF_API.restart_space(repo)
+        spaces = HF_API.list_spaces(author=me.split("/")[0])
+        for space in spaces:
+            if space.id != me:
+                HF_API.restart_space(space.id)
+
+        HF_API.restart_space(me)
 
     except Exception as e:
         send_email(f"Failed to self-restart: {e}")

modules/smtp.py

@@ -1,5 +1,5 @@
 import requests
-from utils import EMAIL, SMTP, TAG, HOST, PORT, API
+from utils import EMAIL, SMTP, TAG, HOST, PORT
 
 
 def email_api(
@@ -21,7 +21,7 @@ def email_api(
     </html>
     """
     response = requests.get(
-        API,
+        "http://api.mmp.cc/api/mail",
         params={
             "host": host,
             "Port": port,

utils.py

@@ -2,9 +2,7 @@ import os
 
 PERIOD = os.getenv("period")
 USERS = os.getenv("users")
-PUB_REPOS = os.getenv("pub_repos")
-PRI_REPOS = os.getenv("pri_repos")
-API = os.getenv("api")
+REPOS = os.getenv("repos")
 EMAIL = os.getenv("email")
 SMTP = os.getenv("smtp")
 HF_TK = os.getenv("hftk")