18
18
from binascii import unhexlify
19
19
from datetime import timedelta
20
20
from enum import Enum , unique
21
- from logging import debug , info
21
+ from logging import getLogger
22
22
from socket import AF_INET
23
23
from types import TracebackType
24
24
from typing import Optional , Set , Tuple , Type , final
40
40
SwitcherStateResponse ,
41
41
)
42
42
43
+ logger = getLogger (__name__ )
44
+
43
45
44
46
@unique
45
47
class Command (Enum ):
@@ -88,24 +90,24 @@ async def __aexit__(
88
90
89
91
async def connect (self ) -> None :
90
92
"""Connect to asynchronous socket and get reader and writer object."""
91
- info ("connecting to the switcher device" )
93
+ logger . info ("connecting to the switcher device" )
92
94
self ._reader , self ._writer = await open_connection (
93
95
host = self ._ip_address ,
94
96
port = self ._port ,
95
97
family = AF_INET ,
96
98
)
97
99
98
100
self ._connected = True
99
- info ("switcher device connected" )
101
+ logger . info ("switcher device connected" )
100
102
101
103
async def disconnect (self ) -> None :
102
104
"""Disconnect from asynchronous socket."""
103
105
if hasattr (self , "_writer" ) and self ._writer :
104
- info ("disconnecting from the switcher device" )
106
+ logger . info ("disconnecting from the switcher device" )
105
107
self ._writer .close ()
106
108
await self ._writer .wait_closed ()
107
109
else :
108
- info ("switcher device not connected" )
110
+ logger . info ("switcher device not connected" )
109
111
self ._connected = False
110
112
111
113
@property
@@ -128,7 +130,7 @@ async def _login(self) -> Tuple[str, SwitcherLoginResponse]:
128
130
packet = packets .LOGIN_PACKET .format (timestamp )
129
131
signed_packet = sign_packet_with_crc_key (packet )
130
132
131
- debug ("sending a login packet" )
133
+ logger . debug ("sending a login packet" )
132
134
self ._writer .write (unhexlify (signed_packet ))
133
135
response = await self ._reader .read (1024 )
134
136
return timestamp , SwitcherLoginResponse (response )
@@ -153,7 +155,7 @@ async def _get_full_state(self) -> Tuple[str, str, SwitcherStateResponse]:
153
155
)
154
156
signed_packet = sign_packet_with_crc_key (packet )
155
157
156
- debug ("sending a get state packet" )
158
+ logger . debug ("sending a get state packet" )
157
159
self ._writer .write (unhexlify (signed_packet ))
158
160
state_resp = await self ._reader .read (1024 )
159
161
try :
@@ -202,7 +204,7 @@ async def control_device(
202
204
)
203
205
signed_packet = sign_packet_with_crc_key (packet )
204
206
205
- debug ("sending a control packet" )
207
+ logger . debug ("sending a control packet" )
206
208
self ._writer .write (unhexlify (signed_packet ))
207
209
response = await self ._reader .read (1024 )
208
210
return SwitcherBaseResponse (response )
@@ -228,7 +230,7 @@ async def set_auto_shutdown(self, full_time: timedelta) -> SwitcherBaseResponse:
228
230
)
229
231
signed_packet = sign_packet_with_crc_key (packet )
230
232
231
- debug ("sending a set auto shutdown packet" )
233
+ logger . debug ("sending a set auto shutdown packet" )
232
234
self ._writer .write (unhexlify (signed_packet ))
233
235
response = await self ._reader .read (1024 )
234
236
return SwitcherBaseResponse (response )
@@ -253,7 +255,7 @@ async def set_device_name(self, name: str) -> SwitcherBaseResponse:
253
255
)
254
256
signed_packet = sign_packet_with_crc_key (packet )
255
257
256
- debug ("sending a set name packet" )
258
+ logger . debug ("sending a set name packet" )
257
259
self ._writer .write (unhexlify (signed_packet ))
258
260
response = await self ._reader .read (1024 )
259
261
return SwitcherBaseResponse (response )
@@ -273,7 +275,7 @@ async def get_schedules(self) -> SwitcherGetSchedulesResponse:
273
275
)
274
276
signed_packet = sign_packet_with_crc_key (packet )
275
277
276
- debug ("sending a get schedules packet" )
278
+ logger . debug ("sending a get schedules packet" )
277
279
self ._writer .write (unhexlify (signed_packet ))
278
280
response = await self ._reader .read (1024 )
279
281
return SwitcherGetSchedulesResponse (response )
@@ -296,7 +298,7 @@ async def delete_schedule(self, schedule_id: str) -> SwitcherBaseResponse:
296
298
)
297
299
signed_packet = sign_packet_with_crc_key (packet )
298
300
299
- debug ("sending a delete schedule packet" )
301
+ logger . debug ("sending a delete schedule packet" )
300
302
self ._writer .write (unhexlify (signed_packet ))
301
303
response = await self ._reader .read (1024 )
302
304
return SwitcherBaseResponse (response )
@@ -335,7 +337,7 @@ async def create_schedule(
335
337
)
336
338
signed_packet = sign_packet_with_crc_key (packet )
337
339
338
- debug ("sending a create schedule packet" )
340
+ logger . debug ("sending a create schedule packet" )
339
341
self ._writer .write (unhexlify (signed_packet ))
340
342
response = await self ._reader .read (1024 )
341
343
return SwitcherBaseResponse (response )
0 commit comments