added version numbering

exddc · Jun 17, 2024 · 0ceb51d · 0ceb51d
1 parent b9afb44
commit 0ceb51d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/agent/__init__.py b/agent/__init__.py
@@ -0,0 +1,3 @@
+"""Init file for the doorbell agent package."""
+
+__version__ = "0.2.0"
diff --git a/agent/agent.py b/agent/agent.py
@@ -13,6 +13,7 @@
 import doorbell_agent
 import indoor_unit_agent
 import web_server
+from . import __version__
 
 # Load environment variables
 dotenv.load_dotenv()
@@ -46,7 +47,7 @@ def __init__(self):
     def run(self):
         """Run the agent and all modules"""
 
-        LOGGER.info("%s agent started.", self.__agent_type)
+        LOGGER.info("%s agent started with version: %s", self.__agent_type, __version__)
         self._agent.run()
         for module in self._modules:
             module.run()

diff --git a/agent/mqtt_agent.py b/agent/mqtt_agent.py
@@ -9,6 +9,7 @@
 import logger
 import dotenv
 import paho.mqtt.client
+from . import __version__
 
 # Load environment variables
 dotenv.load_dotenv()
@@ -187,7 +188,11 @@ def _publish_heartbeat(self) -> None:
                 self.publish(
                     f"status/{os.environ.get('AGENT_LOCATION')}",
                     json.dumps(
-                        {"state": "online", "date": datetime.datetime.now().isoformat()}
+                        {
+                            "state": "online",
+                            "date": datetime.datetime.now().isoformat(),
+                            "version": __version__,
+                        }
                     ),
                     False,
                 )