FOSS/FragmentAPI
0
0
Fork 0
mirror of https://github.com/bohd4nx/FragmentAPI.git synced 2026-07-25 06:14:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: cast cookies to dict for type safety and improve error handling in FragmentClient

Browse Source
This commit is contained in:
bohd4nx
2026-03-20 20:22:33 +02:00
parent 3e14a01c92
commit 75ee76b60e
3 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyfragment/client.py
+3 -2
View File
@@ -1,4 +1,5 @@
import json
from typing import cast
from pyfragment.methods.premium import purchase_premium
from pyfragment.methods.stars import purchase_stars
@@ -71,7 +72,7 @@ class FragmentClient:
except Exception as exc:
raise CookieError(CookieError.READ_FAILED.format(exc=exc)) from exc
missing_keys = [k for k in REQUIRED_COOKIE_KEYS if not str(cookies.get(k, "")).strip()]
missing_keys = [k for k in REQUIRED_COOKIE_KEYS if not str(cast(dict, cookies).get(k, "")).strip()]
if missing_keys:
raise CookieError(CookieError.MISSING_KEYS.format(keys=", ".join(missing_keys)))
@@ -85,7 +86,7 @@ class FragmentClient:
self.seed: str = seed.strip()
self.api_key: str = api_key.strip()
self.cookies: dict = cookies
self.cookies: dict = cast(dict, cookies)
self.wallet_version: WalletVersion = version # type: ignore[assignment]
self.timeout: float = timeout