@@ -152,7 +152,7 @@ def __init__(
152152 self ._token = self ._database .get_auth_token () or token
153153 if headers is None :
154154 headers = self ._default_headers ()
155- self .user_agent = headers
155+ self .headers = headers
156156 self ._validate_device_type ()
157157 self ._send_fake_telemetry : bool = send_fake_telemetry
158158 self ._session_id : int = int (time .time () * 1000 )
@@ -202,10 +202,10 @@ def _default_headers() -> UserAgentPayload:
202202 return UserAgentPayload (device_type = "WEB" )
203203
204204 def _validate_device_type (self ) -> None :
205- if self .user_agent .device_type not in self .allowed_device_types :
205+ if self .headers .device_type not in self .allowed_device_types :
206206 raise ValueError (
207207 f"{ self .__class__ .__name__ } does not support "
208- f"device_type={ self .user_agent .device_type } "
208+ f"device_type={ self .headers .device_type } "
209209 )
210210
211211 async def _wait_forever (self ) -> None :
@@ -312,7 +312,7 @@ async def start(self) -> None:
312312 self .logger .info ("Client starting" )
313313 while not self ._stop_event .is_set ():
314314 try :
315- await self .connect (self .user_agent )
315+ await self .connect (self .headers )
316316
317317 if self .registration :
318318 if not self .first_name :
@@ -325,7 +325,7 @@ async def start(self) -> None:
325325 if self ._token is None :
326326 await self ._login ()
327327
328- await self ._sync (self .user_agent )
328+ await self ._sync (self .headers )
329329 await self ._post_login_tasks (sync = False )
330330
331331 wait_task = asyncio .create_task (self ._wait_forever ())
0 commit comments