feat(em511): Add the driver with some functions

also added some tests, should run!

Signed-off-by: Bobo Bäck Engström <bobo@id8-engineering.io>

Author: Dexter9532

src/em511/__init__.py

@@ -1,6 +1,9 @@
 """Hello package."""
+from .em511 import Em511
+
 
 
 def hello() -> str:
     """Hello docstring."""
     return "Hello from python-em511!"
+

src/em511/em511.py

@@ -0,0 +1,120 @@
+
+"""Driver class for EM511"""
+class Em511:
+    """Em511 driver class"""
+
+    # ---------------------------------------------------------------------
+    # Initialization & Connection
+    # ---------------------------------------------------------------------
+
+    def __init__(self, device_address: int, client, auto_connect=True) -> None:
+        """Constructor"""
+        self.device_address = device_address
+        self.client = client
+        self.connected = False
+        if auto_connect:
+            self.connect()
+
+    #---------------------------------------------------------------------
+
+    def connect(self) -> bool:
+        """Try to connect Modbus client."""
+        if not self.client.connect():
+            print("Could not open COM port", flush=True)
+            self.connected = False
+            return False
+        self.connected = True
+        return True
+    
+    #---------------------------------------------------------------------
+
+    def _read_registers(self, address: int, count: int) -> list[int] | None:
+        """Internal helper to read Modbus registers safely."""
+        if not self.connected and not self.connect():
+            print("Connection failed", flush=True)
+            return None
+        
+        result = self.client.read_input_registers(address=address, count=count, device_id=self.device_address)
+        if result.isError():
+            print(f"Read error at {address}: {result}", flush=True)
+            return None
+        return result.registers
+    
+    #---------------------------------------------------------------------
+
+    def _unpack(self, regs: list[int], reg_type: str) -> float | None:
+        """Internal helper to unpuck register list"""
+        if reg_type == "int16":
+            return regs[0]
+        
+        elif reg_type == "int32":
+            return (regs[1] << 16) + regs[0]
+        else:
+            return None
+        
+    # ---------------------------------------------------------------------
+    # Core Read Functions
+    # ---------------------------------------------------------------------
+
+    def read_V(self) -> float | None:
+        """Return V value"""
+        regs = self._read_registers(0, 2)
+        if not regs:
+            return None
+        return self._unpack(regs, "int32") / 10
+    
+    #---------------------------------------------------------------------
+
+    def read_A(self) -> float | None:
+        """Return A value"""
+        regs = self._read_registers(2, 2)
+        if not regs:
+            return None
+        return self._unpack(regs, "int32") / 1000
+    
+    #---------------------------------------------------------------------
+
+    def read_W(self) -> float | None:
+        """Return W value"""
+        regs = self._read_registers(4, 2)
+        if not regs:
+            return None
+        return self._unpack(regs, "int32") / 10
+    
+    #---------------------------------------------------------------------
+
+    def read_W_dmd(self) -> float | None:
+        """Return W Dmd value"""
+        regs = self._read_registers(10, 2)
+        if not regs:
+            return None
+        return self._unpack(regs, "int32") / 10
+    
+    #---------------------------------------------------------------------
+
+    def read_W_dmd_peak(self) -> float | None:
+        """Return W dmd peak value"""
+        regs = self._read_registers(12, 2)
+        if not regs:
+            return None
+        return self._unpack(regs, "int32") / 10
+    
+    #---------------------------------------------------------------------
+
+    def read_Hz(self) -> float | None:
+        """Return Hz value"""
+        regs = self._read_registers(15, 1)
+        if not regs:
+            return None
+        return self._unpack(regs, "int16") / 10
+    
+    #---------------------------------------------------------------------
+
+    def read_kWh_tot(self) -> float | None:
+        """Return kWh value"""
+        regs = self._read_registers(16, 2)
+        if not regs:
+            return None
+        return self._unpack(regs, "int32") / 10
+    
+    #---------------------------------------------------------------------

src/em511/my_functions.py

@@ -0,0 +1,2 @@
+"""My functions file, not for the user"""
+

src/em511/test_hello.py

@@ -1,7 +1,37 @@
-# noqa: D100
-import em511
+"""Test file for driver"""
+from em511 import Em511
+from pymodbus.client import ModbusSerialClient
+from time import sleep
 
+"""Init client"""
+client = ModbusSerialClient(
+    port="/dev/ttyUSB0",
+    baudrate=9600,
+    parity="E",
+    stopbits=1,
+    bytesize=8,
+    timeout=1,
+)
+print("client params is set")
 
-def test_hello() -> None:
-    """Test hello function."""
-    assert em511.hello() == "Hello from python-em511!"  # noqa: S101
+"""Init constructor"""
+AC_meter = Em511(102, client)
+
+def test():
+
+    if client.connect():
+        print("Client is connected", flush=True)
+        print("Testing Functions", flush=True)
+        print("V: ",AC_meter.read_V())
+        print("A: ",AC_meter.read_A())
+        print("W: ",AC_meter.read_W())
+        print("W dmd: ",AC_meter.read_W_dmd())
+        print("W dmd peak: ",AC_meter.read_W_dmd_peak())
+        print("Hz : ",AC_meter.read_Hz())
+        print("kWh tot : ",AC_meter.read_kWh_tot())
+
+    else:
+        print("Could not open COM port", flush=True)
+                   
+if __name__ == "__main__":
+    test()