Add support for creating full device reader

Author: glopesdev

harp/reader.py

@@ -8,7 +8,7 @@
 _camel_to_snake_regex = re.compile(r"(?<!^)(?=[A-Z])")
 
 
-class Reader:
+class RegisterReader:
     register: Register
     read: Callable[[str], DataFrame]
 
@@ -17,6 +17,15 @@ def __init__(self, register: Register, read: Callable[[str], DataFrame]) -> None
         self.read = read
 
 
+class DeviceReader:
+    model: Model
+    registers: dict[str, RegisterReader]
+
+    def __init__(self, model: Model, registers: dict[str, RegisterReader]) -> None:
+        self.model = model
+        self.registers = registers
+
+
 def compose(f, g):
     return lambda *a, **kw: f(g(*a, **kw))
 
@@ -33,7 +42,14 @@ def create_bitreader(mask):
     return lambda xs: ((xs & mask) != 0)
 
 
-def create_reader(device: Model, name: str):
+def create_reader(device: Model):
+    reg_readers = {
+        create_register_reader(device, name) for name in device.registers.keys()
+    }
+    return DeviceReader(device, reg_readers)
+
+
+def create_register_reader(device: Model, name: str):
     register = device.registers[name]
     reader = read
 
@@ -50,22 +66,22 @@ def unpack(df):
                 return DataFrame({n: f(df[0]) for n, f in lookup}, index=df.index)
 
             reader = compose(unpack, reader)
-            return Reader(register, reader)
+            return RegisterReader(register, reader)
 
         groupMask = device.groupMasks.get(key)
         if groupMask is not None:
             name = id_camel_to_snake(name)
             lookup = {value.root: name for name, value in groupMask.values.items()}
             reader = partial(reader, columns=[name])
             reader = compose(lambda df: df.map(lambda x: lookup[x]), reader)
-            return Reader(register, reader)
+            return RegisterReader(register, reader)
 
     if register.payloadSpec is not None:
         columns = register.payloadSpec.keys()
         columns = keys_camel_to_snake(columns)
         reader = partial(reader, columns=columns)
-        return Reader(register, reader)
+        return RegisterReader(register, reader)
 
     columns = [id_camel_to_snake(name)]
     reader = partial(reader, columns=columns)
-    return Reader(register, reader)
+    return RegisterReader(register, reader)