g0ldyy/comet
1
0
Fork 0
mirror of https://github.com/g0ldyy/comet.git synced 2026-01-12 01:16:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Enhance MediaFusionConfig by simplifying password handling and improving caching mechanism

Browse Source
This commit is contained in:
g0ldyy
2025-08-27 16:44:30 +02:00
parent 288cbcb042
commit 09fee71544
1 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comet/utils/mediafusion.py
+14 -12
View File
@@ -17,7 +17,7 @@ def associate_mediafusion_urls_passwords(
if passwords is None:
password = None
elif isinstance(passwords, str):
password = passwords
password = passwords or None
elif isinstance(passwords, list) and len(passwords) > 0:
password = passwords[0]
else:
@@ -28,12 +28,13 @@ def associate_mediafusion_urls_passwords(
if passwords is None:
passwords_list = [None] * len(urls)
elif isinstance(passwords, str):
passwords_list = [passwords] * len(urls)
passwords_list = [passwords or None] * len(urls)
elif isinstance(passwords, list):
passwords_list = []
for i in range(len(urls)):
if i < len(passwords):
passwords_list.append(passwords[i])
pw = passwords[i] or None
passwords_list.append(pw)
else:
passwords_list.append(None)
@@ -56,28 +57,29 @@ def encode_mediafusion_api_password(api_password: str):
class MediaFusionConfig:
def __init__(self):
self._password_cache = {}
self.password_cache = {}
self.default_headers = {
"encoded_user_data": encode_mediafusion_api_password("")
}
self.precompute_encodings()
self._precompute_encodings()
print(self._password_cache)
def _precompute_encodings(self):
def precompute_encodings(self):
urls = settings.MEDIAFUSION_URL
passwords = settings.MEDIAFUSION_API_PASSWORD
url_password_pairs = associate_mediafusion_urls_passwords(urls, passwords)
for _, password in url_password_pairs:
if password is not None and password not in self._password_cache:
self._password_cache[password] = encode_mediafusion_api_password(
if password is not None and password not in self.password_cache:
self.password_cache[password] = encode_mediafusion_api_password(
password
)
def get_headers_for_password(self, api_password: str | None):
if api_password is None:
return {}
return self.default_headers
return {"encoded_user_data": self._password_cache[api_password]}
return {"encoded_user_data": self.password_cache[api_password]}
mediafusion_config = MediaFusionConfig()