mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-21 21:58:04 +00:00
34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
|
from typing import Optional, Dict
|
||
|
|
|
||
|
|
from core.services.devices import DevicesService
|
||
|
|
from modules.apihelper.utility.helpers import get_device_id, hex_digest
|
||
|
|
|
||
|
|
|
||
|
|
class DevicesMethods:
|
||
|
|
def __init__(self):
|
||
|
|
self.service: Optional[DevicesService] = None
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def get_default_device_header(account_id: int, headers: Dict = None) -> Dict[str, str]:
|
||
|
|
headers = headers or {}
|
||
|
|
headers["x-rpc-device_id"] = get_device_id(str(account_id))
|
||
|
|
headers["x-rpc-device_fp"] = hex_digest(headers["x-rpc-device_id"])[:13]
|
||
|
|
headers["x-rpc-device_name"] = "Xiaomi"
|
||
|
|
return headers
|
||
|
|
|
||
|
|
async def update_device_headers(self, account_id: int, headers: Dict = None) -> Dict[str, str]:
|
||
|
|
account_id = account_id or 0
|
||
|
|
if not self.service:
|
||
|
|
return self.get_default_device_header(account_id, headers)
|
||
|
|
device = await self.service.get(account_id)
|
||
|
|
if not device:
|
||
|
|
return self.get_default_device_header(account_id, headers)
|
||
|
|
headers = headers or {}
|
||
|
|
headers["x-rpc-device_id"] = device.device_id
|
||
|
|
headers["x-rpc-device_fp"] = device.device_fp
|
||
|
|
headers["x-rpc-device_name"] = device.device_name or "Xiaomi"
|
||
|
|
return headers
|
||
|
|
|
||
|
|
|
||
|
|
devices_methods = DevicesMethods()
|