Fix type checks/linting

Author: OCopping

src/fastcs/connections/modbus_connection.py

@@ -1,19 +1,17 @@
 import asyncio
-
 from dataclasses import dataclass
 from typing import Optional
 
-from pymodbus.client.base import ModbusBaseClient
 from pymodbus.client import (
     AsyncModbusSerialClient,
     AsyncModbusTcpClient,
     AsyncModbusUdpClient,
+    ModbusBaseClient,
 )
-from pymodbus.exceptions import ModbusException
 
+from pymodbus.exceptions import ModbusException
 from pymodbus.framer import Framer
-from pymodbus.pdu import ExceptionResponse
-
+from pymodbus.pdu import ExceptionResponse, ModbusResponse
 
 # Constants
 CR = "\r"
@@ -27,10 +25,9 @@ class ModbusConnectionSettings:
     port: int = 7001
     slave: int = 0
 
-class ModbusConnection:
 
+class ModbusConnection:
     def __init__(self, settings: ModbusConnectionSettings) -> None:
-
         self.host, self.port, self.slave = settings.host, settings.port, settings.slave
         self.running: bool = False
 
@@ -39,11 +36,10 @@ def __init__(self, settings: ModbusConnectionSettings) -> None:
     async def connect(self, framer=Framer.SOCKET):
         raise NotImplementedError
 
-
     def disconnect(self):
         self._client.close()
 
-    async def _read(self, address: int, count: int = 2) -> Optional[str]:
+    async def _read(self, address: int, count: int = 2) -> Optional[ModbusResponse]:
         # address -= 1  # modbus spec starts from 0 not 1
         try:
             # address_hex = hex(address)
@@ -61,6 +57,8 @@ async def _read(self, address: int, count: int = 2) -> Optional[str]:
             # Received Modbus library exception
             # THIS IS NOT A PYTHON EXCEPTION, but a valid modbus message
             self.disconnect()
+        else:
+            return rr
 
     async def send(self, address: int, value: int) -> None:
         """Send a request.
@@ -72,13 +70,13 @@ async def send(self, address: int, value: int) -> None:
         await self._client.write_registers(address, value, slave=self.slave)
         resp = await self._read(address, 2)
 
-class ModbusSerialConnection(ModbusConnection):
 
+class ModbusSerialConnection(ModbusConnection):
     def __init__(self, settings: ModbusConnectionSettings) -> None:
         super().__init__(settings)
 
     async def connect(self, framer=Framer.SOCKET):
-        self._client: AsyncModbusSerialClient = AsyncModbusSerialClient(
+        self._client = AsyncModbusSerialClient(
             str(self.port),
             framer=framer,
             timeout=10,
@@ -95,13 +93,13 @@ async def connect(self, framer=Framer.SOCKET):
         await self._client.connect()
         assert self._client.connected
 
-class ModbusTcpConnection(ModbusConnection):
 
+class ModbusTcpConnection(ModbusConnection):
     def __init__(self, settings: ModbusConnectionSettings) -> None:
         super().__init__(settings)
 
     async def connect(self, framer=Framer.SOCKET):
-        self._client: AsyncModbusTcpClient = AsyncModbusTcpClient(
+        self._client = AsyncModbusTcpClient(
             self.host,
             self.port,
             framer=framer,
@@ -116,13 +114,13 @@ async def connect(self, framer=Framer.SOCKET):
         await self._client.connect()
         assert self._client.connected
 
-class ModbusUdpConnection(ModbusConnection):
 
+class ModbusUdpConnection(ModbusConnection):
     def __init__(self, settings: ModbusConnectionSettings) -> None:
         super().__init__(settings)
 
     async def connect(self, framer=Framer.SOCKET):
-        self._client: AsyncModbusUdpClient = AsyncModbusUdpClient(
+        self._client = AsyncModbusUdpClient(
             self.host,
             self.port,
             framer=framer,
@@ -135,4 +133,4 @@ async def connect(self, framer=Framer.SOCKET):
         )
 
         await self._client.connect()
-        assert self._client.connected
+        assert self._client.connected