Add black dependency and format code with it

dev_requirements.txt

@@ -1,2 +1,3 @@
 pytest
-pytest-cov
+pytest-cov
+black

src/blockperf/cli.py

@@ -34,8 +34,8 @@ def setup_logger(debug: bool):
             },
             "extra": {
                 "format": "%(asctime)-16s %(name)-8s %(filename)-12s %(lineno)-6s %(funcName)-30s %(levelname)-8s %(message)s",
-                "datefmt": "%m-%d %H:%M:%S"
-            }
+                "datefmt": "%m-%d %H:%M:%S",
+            },
         },
         "handlers": {
             "console": {
@@ -54,16 +54,8 @@ def setup_logger(debug: bool):
             #    "backupCount": 2,
             # }
         },
-        "loggers": {
-            "blockperf": {
-                "level": "DEBUG",
-                "handlers": []
-            }
-        },
-        "root": {
-            "level": level,
-            "handlers": ["console"]
-        }
+        "loggers": {"blockperf": {"level": "DEBUG", "handlers": []}},
+        "root": {"level": level, "handlers": ["console"]},
     }
     dictConfig(logger_config)
 
@@ -74,8 +66,7 @@ def setup_argparse():
     parser.add_argument(
         "command", help="Command to run blockperf with", choices=["run"]
     )
-    parser.add_argument(
-        "--debug", help="Write more debug output", action="store_true")
+    parser.add_argument("--debug", help="Write more debug output", action="store_true")
     return parser.parse_args()
 
 

src/blockperf/mqtt.py

@@ -5,6 +5,7 @@
 import json
 from paho.mqtt.client import MQTTMessageInfo
 from paho.mqtt.properties import Properties as Properties
+
 try:
     import paho.mqtt.client as mqtt
     from paho.mqtt.properties import Properties
@@ -31,7 +32,7 @@ def __init__(
         client_keyfile: str,
         host: str,
         port: int,
-        keepalive: int
+        keepalive: int,
     ) -> None:
         super().__init__(protocol=mqtt.MQTTv5)
         self.tls_set(
@@ -86,7 +87,8 @@ def publish(self, topic: str, payload: dict):
             logger.info("Publishing sample to %s", topic)
             # call the actuall clients publish method and receive the message_info
             message_info: MQTTMessageInfo = super().publish(
-                topic=topic, payload=json_payload, properties=publish_properties)
+                topic=topic, payload=json_payload, properties=publish_properties
+            )
             # The message_info might not yet have been published,
             # wait_for_publish() blocks until TIMEOUT for that message to be published
             message_info.wait_for_publish(PUBLISH_TIMEOUT)

src/blockperf/nodelogs.py

@@ -157,8 +157,12 @@ def __init__(self, event_data: dict) -> None:
         ):
             self.local_addr = self.data.get("peer", {}).get("local", {}).get("addr", "")
             self.local_port = self.data.get("peer", {}).get("local", {}).get("port", "")
-            self.remote_addr = self.data.get("peer", {}).get("remote", {}).get("addr", "")
-            self.remote_port = self.data.get("peer", {}).get("remote", {}).get("port", "")
+            self.remote_addr = (
+                self.data.get("peer", {}).get("remote", {}).get("addr", "")
+            )
+            self.remote_port = (
+                self.data.get("peer", {}).get("remote", {}).get("port", "")
+            )
 
     def __repr__(self):
         _kind = self.kind.value