format

Author: Surajsm60720

node/node/app.py

@@ -3,7 +3,10 @@
 from node.read import read_gpio_sensors
 from node.transmit import send_to_rabbitmq
 
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+)
+
 
 async def run():
     while True:
@@ -13,12 +16,14 @@ async def run():
         logging.info("Data sent to RabbitMQ")
         await asyncio.sleep(1)
 
+
 def main():
     try:
         asyncio.run(run())
     except KeyboardInterrupt:
         logging.info("Program terminated by user")
     return 0
 
+
 if __name__ == "__main__":
     main()

node/node/read/__init__.py

@@ -1,3 +1,3 @@
 from .read import read_gpio_sensors
 
-__all__ = ['read_gpio_sensors']
+__all__ = ["read_gpio_sensors"]

node/node/transmit/transmit.py

@@ -3,38 +3,47 @@
 from node.logger import ops_logger
 from node.settings import CONFIG
 
+
 def send_to_rabbitmq(data):
-  connection = None
-  try:
-    ops_logger.info("Attempting to connect to RabbitMQ")
-    if CONFIG["rabbitmq_user"] and CONFIG["rabbitmq_password"]:
-      credentials = pika.PlainCredentials(CONFIG["rabbitmq_user"], CONFIG["rabbitmq_password"])
-    else:
-      credentials = pika.PlainCredentials("guest", "guest")
-    connection = pika.BlockingConnection(pika.ConnectionParameters(host=CONFIG["rabbitmq_host"], port=CONFIG["rabbitmq_port"], credentials=credentials))
-    channel = connection.channel()
-    ops_logger.info("Connected to RabbitMQ")
+    connection = None
+    try:
+        ops_logger.info("Attempting to connect to RabbitMQ")
+        if CONFIG["rabbitmq_user"] and CONFIG["rabbitmq_password"]:
+            credentials = pika.PlainCredentials(
+                CONFIG["rabbitmq_user"], CONFIG["rabbitmq_password"]
+            )
+        else:
+            credentials = pika.PlainCredentials("guest", "guest")
+        connection = pika.BlockingConnection(
+            pika.ConnectionParameters(
+                host=CONFIG["rabbitmq_host"],
+                port=CONFIG["rabbitmq_port"],
+                credentials=credentials,
+            )
+        )
+        channel = connection.channel()
+        ops_logger.info("Connected to RabbitMQ")
 
-    ops_logger.info(f"Declaring queue: {CONFIG['rabbitmq_queue']}")
-    channel.queue_declare(queue=CONFIG["rabbitmq_queue"], durable=True)
+        ops_logger.info(f"Declaring queue: {CONFIG['rabbitmq_queue']}")
+        channel.queue_declare(queue=CONFIG["rabbitmq_queue"], durable=True)
 
-    ops_logger.info(f"Publishing data to queue: {CONFIG['rabbitmq_queue']}")
-    channel.basic_publish(
-      exchange='',
-      routing_key=CONFIG["rabbitmq_queue"],
-      body=json.dumps(data)
-    )
+        ops_logger.info(f"Publishing data to queue: {CONFIG['rabbitmq_queue']}")
+        channel.basic_publish(
+            exchange="",
+            routing_key=CONFIG["rabbitmq_queue"],
+            body=json.dumps(data),
+        )
 
-    ops_logger.info(f"Sent data to RabbitMQ: {data}")
-  except pika.exceptions.AMQPConnectionError as e:
-    ops_logger.error(f"RabbitMQ connection error")
-  except pika.exceptions.AMQPChannelError as e:
-    ops_logger.error(f"RabbitMQ channel error")
-  except Exception as e:
-    ops_logger.error(f"RabbitMQ error")
-  finally:
-    try:
-      connection.close()
-      ops_logger.info("Closed RabbitMQ connection")
+        ops_logger.info(f"Sent data to RabbitMQ: {data}")
+    except pika.exceptions.AMQPConnectionError as e:
+        ops_logger.error("RabbitMQ connection error")
+    except pika.exceptions.AMQPChannelError as e:
+        ops_logger.error("RabbitMQ channel error")
     except Exception as e:
-      ops_logger.error(f"Error closing RabbitMQ connection: {e}")
+        ops_logger.error("RabbitMQ error")
+    finally:
+        try:
+            connection.close()
+            ops_logger.info("Closed RabbitMQ connection")
+        except Exception as e:
+            ops_logger.error(f"Error closing RabbitMQ connection: {e}")