|
|
|
|
| import os
|
| import time
|
| from http.cookies import SimpleCookie
|
| from threading import Thread
|
|
|
| import requests
|
|
|
| from utils import COMMON_HEADERS
|
|
|
|
|
| class SunoCookie:
|
| def __init__(self):
|
| self.cookie = SimpleCookie()
|
| self.session_id = None
|
| self.token = None
|
| self.is_disabled = False
|
|
|
| def load_cookie(self, cookie_str):
|
| self.cookie.load(cookie_str)
|
|
|
| def get_cookie(self):
|
| return ";".join([f"{i}={self.cookie.get(i).value}" for i in self.cookie.keys()])
|
|
|
| def set_session_id(self, session_id):
|
| self.session_id = session_id
|
|
|
| def get_session_id(self):
|
| return self.session_id
|
|
|
| def get_token(self):
|
| return self.token
|
|
|
| def set_token(self, token: str):
|
| self.token = token
|
|
|
| def disable(self):
|
| self.is_disabled = True
|
|
|
| def enable(self):
|
| self.is_disabled = False
|
|
|
| def load_env_cookies():
|
| suno_auths = {}
|
| for key, value in os.environ.items():
|
| if key.startswith("SESSION_ID"):
|
| try:
|
| i = int(key[10:])
|
| cookie = os.getenv(f"COOKIE{i}")
|
| if cookie:
|
| suno_auth = SunoCookie()
|
| suno_auth.set_session_id(value)
|
| suno_auth.load_cookie(cookie)
|
| suno_auths[i] = suno_auth
|
| except ValueError:
|
| continue
|
| return suno_auths
|
|
|
| def update_token(suno_cookie: SunoCookie):
|
| headers = {"cookie": suno_cookie.get_cookie()}
|
| headers.update(COMMON_HEADERS)
|
| session_id = suno_cookie.get_session_id()
|
|
|
| resp = requests.post(
|
| url=f"https://clerk.suno.com/v1/client/sessions/{session_id}/tokens?_clerk_js_version=5.22.3",
|
| headers=headers,
|
| )
|
|
|
| resp_headers = dict(resp.headers)
|
| set_cookie = resp_headers.get("Set-Cookie")
|
| suno_cookie.load_cookie(set_cookie)
|
| token = resp.json().get("jwt")
|
| suno_cookie.set_token(token)
|
|
|
|
|
|
|
|
|
| def keep_alive(suno_cookie: SunoCookie):
|
| while True:
|
| try:
|
| update_token(suno_cookie)
|
| except Exception as e:
|
| print(e)
|
| finally:
|
| time.sleep(5)
|
|
|
| def start_keep_alive(suno_auths):
|
| for suno_auth in suno_auths.values():
|
| t = Thread(target=keep_alive, args=(suno_auth,))
|
| t.start()
|
|
|
| suno_auths = load_env_cookies()
|
| start_keep_alive(suno_auths) |
