Mypi fix (#231)

danielperna84 · Jan 25, 2022 · be5c2b0 · be5c2b0
1 parent 66c362c
commit be5c2b0
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 14 deletions.
diff --git a/hahomematic/central_unit.py b/hahomematic/central_unit.py
@@ -519,10 +519,7 @@ async def get_install_mode(self, interface_id: str) -> int:
         return 0
 
     async def get_value(
-        self,
-        interface_id: str,
-        channel_address: str,
-        parameter: str
+        self, interface_id: str, channel_address: str, parameter: str
     ) -> Any | None:
         """Get a single value on paramset VALUES."""
 

diff --git a/hahomematic/client.py b/hahomematic/client.py
@@ -300,10 +300,14 @@ async def put_paramset(
         """Set paramsets manually."""
         try:
             if rx_mode:
-                await self._proxy.putParamset(channel_address, paramset_key, value, rx_mode)
+                await self._proxy.putParamset(
+                    channel_address, paramset_key, value, rx_mode
+                )
             else:
                 await self._proxy.putParamset(channel_address, paramset_key, value)
-            _LOGGER.debug("put_paramset: %s, %s, %s", channel_address, paramset_key, value)
+            _LOGGER.debug(
+                "put_paramset: %s, %s, %s", channel_address, paramset_key, value
+            )
         except BaseHomematicException as hhe:
             _LOGGER.warning(
                 "put_paramset failed: %s (%s) %s, %s, %s",
@@ -314,7 +318,9 @@ async def put_paramset(
                 value,
             )
 
-    async def fetch_paramset_description(self, channel_address: str, paramset: str) -> None:
+    async def fetch_paramset_description(
+        self, channel_address: str, paramset: str
+    ) -> None:
         """
         Fetch a specific paramset and add it to the known ones.
         """
@@ -395,7 +401,9 @@ async def get_all_paramsets(
         all_paramsets: dict[str, dict[str, Any]] = {}
         for device_description in device_descriptions:
             all_paramsets.update(
-                await self.get_paramset_descriptions(device_description=device_description)
+                await self.get_paramset_descriptions(
+                    device_description=device_description
+                )
             )
         return all_paramsets
 

diff --git a/hahomematic/hmcli.py b/hahomematic/hmcli.py
@@ -5,12 +5,13 @@
 import argparse
 import json
 import sys
+from typing import Any
 from xmlrpc.client import ServerProxy
 
 from hahomematic.helpers import build_api_url, get_tls_context
 
 
-def main():
+def main() -> None:
     """Main function."""
     parser = argparse.ArgumentParser(
         description="Commandline tool to query HomeMatic hubs via XML-RPC."
@@ -87,6 +88,7 @@ def main():
                 print(res)
             sys.exit(0)
         elif args.paramset == "VALUES" and args.value:
+            value: Any
             if args.type == "int":
                 value = int(args.value)
             elif args.type == "float":
@@ -98,11 +100,13 @@ def main():
             proxy.setValue(args.address, args.parameter, value)
             sys.exit(0)
         elif args.paramset == "MASTER" and args.value is None:
-            res = proxy.getParamset(args.address, args.paramset).get(args.parameter)
-            if args.json:
-                print(json.dumps({args.parameter: res}))
-            else:
-                print(res)
+            paramset: dict[str, Any] | None
+            if paramset := proxy.getParamset(args.address, args.paramset):  # type: ignore
+                if param_value := paramset.get(args.parameter):
+                    if args.json:
+                        print(json.dumps({args.parameter: param_value}))
+                    else:
+                        print(param_value)
             sys.exit(0)
         elif args.paramset == "MASTER" and args.value:
             if args.type == "int":