Skip to content
Snippets Groups Projects
Verified Commit 4273bb7c authored by Ole Binder's avatar Ole Binder
Browse files

Chore: add type hints

parent 8f1207f2
Branches
No related tags found
1 merge request!51Fix broken pipe #153
......@@ -44,12 +44,12 @@ class CoreClient:
skip_log_commands = ("aura_engine.status", "mixer.volume")
instance = None
instance: CoreClient
logger: logging.Logger
config: confuse.Configuration
connection = None
event_dispatcher = None
conn = None
conn: CoreConnection
def __init__(self, event_dispatcher):
"""
......@@ -80,7 +80,7 @@ class CoreClient:
if not self.conn.is_connected():
self.conn.open()
except CoreConnectionError as e:
self.logger.critical(SU.red(e.message))
self.logger.critical(SU.red(str(e)))
@synchronized
def disconnect(self):
......@@ -231,10 +231,10 @@ class CoreConnection:
logger: logging.Logger
config: confuse.Configuration
socket_path = None
socket = None
connected = None
message = None
socket_path: str
socket: socket.socket
connected: bool
message: str
def __init__(self):
"""
......@@ -248,7 +248,6 @@ class CoreConnection:
self.connected = False
self.message = ""
self.socket = None
def is_connected(self):
"""
......@@ -286,7 +285,7 @@ class CoreConnection:
"""
if self.connected:
message = "quit\r"
self.socket.sendall(message.decode(CoreConnection.ENCODING))
self.socket.sendall(message.encode(CoreConnection.ENCODING))
self.socket.close()
self.connected = False
......@@ -378,6 +377,7 @@ class CoreConnection:
msg = "Unknown error while socket.read()"
self.logger.error(SU.red(msg), e)
return self.message
# FIXME: return str if no tests depend on this None.
return None
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment