PaiGramTeam/SIMNet
3
0
Fork 0
mirror of https://github.com/PaiGramTeam/SIMNet.git synced 2024-09-28 03:45:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

✨ Add custom device header

Browse Source
This commit is contained in:
omg-xtao 2023-06-24 02:45:26 +08:00 committed by GitHub
parent 424c24a10b
commit 0161ec695c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
simnet/client/base.py
View File

@ -50,7 +50,7 @@ class BaseClient(AsyncContextManager["BaseClient"]):
"""
game: Optional[Game] = None
_device_id = str(uuid.uuid3(uuid.NAMESPACE_URL, "SIMNet"))
__device_id = str(uuid.uuid3(uuid.NAMESPACE_URL, "SIMNet"))
def __init__(
self,
@ -61,6 +61,8 @@ class BaseClient(AsyncContextManager["BaseClient"]):
region: Region = Region.OVERSEAS,
lang: str = "en-us",
timeout: Optional[TimeoutTypes] = None,
device_id: Optional[str] = None,
device_fp: Optional[str] = None,
) -> None:
"""Initialize the client with the given parameters."""
if timeout is None:
@ -78,6 +80,8 @@ class BaseClient(AsyncContextManager["BaseClient"]):
self.client = AsyncClient(cookies=cookies, timeout=timeout)
self.region = region
self.lang = lang
self._device_id = device_id
self._device_fp = device_fp
@property
def cookies(self) -> Cookies:
@ -96,15 +100,29 @@ class BaseClient(AsyncContextManager["BaseClient"]):
@property
def device_id(self) -> str:
"""Get the device id used for the client."""
if self._device_id:
return self._device_id
if self.account_id is not None:
return str(uuid.uuid3(uuid.NAMESPACE_URL, str(self.account_id)))
return self._device_id
return self.__device_id
@device_id.setter
def device_id(self, device_id: str) -> None:
"""Set the device id used for the client."""
self._device_id = device_id
@property
def device_fp(self) -> str:
"""Get the device fingerprint used for the client."""
if self._device_fp:
return self._device_fp
return hex_digest(self.device_id)[:13]
@device_fp.setter
def device_fp(self, device_fp: str) -> None:
"""Set the device fingerprint used for the client."""
self._device_fp = device_fp
@property
def app_version(self) -> str:
"""Get the app version used for the client."""