feat: add config path as an parameter

SlaviXG · Jun 20, 2024 · 7821156 · 7821156
1 parent 691593f
commit 7821156
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/commander.py b/commander.py
@@ -1,3 +1,4 @@
+import argparse
 import paho.mqtt.client as mqtt
 import configparser
 import os
@@ -68,7 +69,11 @@ def init_commander(config_path: os.path) -> BaseCommander:
 
 
 if __name__ == '__main__':
-    commander = init_commander('config.ini')
+    parser = argparse.ArgumentParser(description="Commander for sending commands to clients.")
+    parser.add_argument('--config', type=str, default='config.ini', help='Path to the configuration file.')
+    args = parser.parse_args()
+
+    commander = init_commander(args.config)
     commander.connect()
 
     try:

diff --git a/operator.py b/operator.py
@@ -1,3 +1,4 @@
+import argparse
 import paho.mqtt.client as mqtt
 import configparser
 import os
@@ -166,8 +167,12 @@ def run_realtime_mode(self):
 
 
 if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description="Operator for managing commands and clients.")
+    parser.add_argument('--config', type=str, default='config.ini', help='Path to the configuration file.')
+    args = parser.parse_args()
+
     config = configparser.ConfigParser()
-    config.read('config.ini')
+    config.read(args.config)
     broker = os.getenv('MQTT_BROKER') or config['mqtt']['broker']
     port = int(config['mqtt']['port'])
     command_topic = config['mqtt']['command_topic']

diff --git a/sut.py b/sut.py
@@ -1,3 +1,4 @@
+import argparse
 import paho.mqtt.client as mqtt
 import configparser
 import subprocess
@@ -150,8 +151,12 @@ def stop(self):
 
 
 if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description="System Under Test (SUT) for processing commands.")
+    parser.add_argument('--config', type=str, default='config.ini', help='Path to the configuration file.')
+    args = parser.parse_args()
+
     config = configparser.ConfigParser()
-    config.read('config.ini')
+    config.read(args.config)
     broker = os.getenv('MQTT_BROKER') if os.getenv('MQTT_BROKER') is not None else config['mqtt']['broker']
     port = int(config['mqtt']['port'])
     command_topic = config['mqtt']['command_topic']