[heatpump] fix fonction de correction des adresses

pyheatpump/cli.py

 
 from pyheatpump.logger import logger_init
 from pyheatpump.models import *
+from pyheatpump.lib import shift_response
 
 CONTEXT_SETTINGS={
     'default_map':{'run': {}}
 
     from .models.heatpump import Heatpump
 
-    def shift_response(d):
-        def shift_dict(d):
-            d_k = list(map(int, d.keys()))
-            if not len(d_k):
-                return d
-
-            d_k.sort()
-            offset = d_k[0]-1
-            return {
-                str(k - offset): d[str(k)]
-                for k in d_k
-                if str(k) in d
-            }
-
-        if 'Analog' in d.keys():
-            d.update({"Analog": shift_dict(d['Analog'])})
-
-        if 'Integer' in d.keys():
-            d.update({"Integer": shift_dict(d['Integer'])})
-
-        if 'Digital' in d.keys():
-            d.update({"Digital": shift_dict(d['Digital'])})
-
-        return d
-
-
     if not since:
         last_update = 0
     else:

pyheatpump/heatpump.py

 from starlette.routing import Router, Route
 from starlette.responses import JSONResponse
 from .config import config
+from .lib import shift_response
 
 from pyheatpump.models import *
 
         time = datetime.fromisoformat(request.params['time'])
         h = Heatpump(config.get('heatpump', 'mac_address'), request.params['time'])
 
-    return JSONResponse(h.__dict__())
+    return JSONResponse(shift_response(h.__dict__()))
 
 
 ROUTES=[

pyheatpump/lib.py

+def shift_response(d):
+    def shift_dict(d_):
+        d_k = list(map(int, d_.keys()))
+        if not len(d_k):
+            return d_
+
+        d_k.sort()
+        offset = d_k[0]-1
+
+        d_res = {
+            str(k - offset): d_[k]
+            for k in d_k
+            if (k in d_ and isinstance(d_[k], int))
+        }
+
+        return d_res
+
+    if 'Analog' in d.keys():
+        d.update({"Analog": shift_dict(d['Analog'])})
+
+    if 'Integer' in d.keys():
+        d.update({"Integer": shift_dict(d['Integer'])})
+
+    if 'Digital' in d.keys():
+        d.update({"Digital": shift_dict(d['Digital'])})
+
+    return d