add version parameter to the script

Author: JonasScholl

src/videoipath_automation_tool/apps/inventory/model/drivers.py

@@ -1,11 +1,11 @@
 from abc import ABC
-from typing import Dict, Literal, Type, TypeVar, Union, Optional
+from typing import Dict, Literal, Optional, Type, TypeVar, Union
 
 from pydantic import BaseModel, Field
 
 # Notes:
 # - The name of the custom settings model follows the naming convention: CustomSettings_<driver_organization>_<driver_name>_<driver_version> => "." and "-" are replaced by "_"!
-# - src/videoipath_automation_tool/apps/inventory/model/driver_schema/2024.1.4.json is used as reference to define the custom settings model!
+# - Schema 2024.4.12.json is used as reference to define the custom settings model!
 # - The "driver_id" attribute is necessary for the discriminator, which is used to determine the correct model for the custom settings in DeviceConfiguration!
 # - The "alias" attribute is used to map the attribute to the correct key (with driver organization & name) in the JSON payload for the API!
 # - "DriverLiteral" is used to provide a list of all possible drivers in the IDEs IntelliSense!

src/videoipath_automation_tool/scripts/generate_all.py

@@ -1,16 +1,27 @@
-from videoipath_automation_tool.scripts.generate_driver_models import DEFAULT_OUTPUT_FILE, DEFAULT_SCHEMA_FILE, parser
+import argparse
+import os
+
 from videoipath_automation_tool.scripts.generate_driver_models import main as generate_driver_models
 from videoipath_automation_tool.scripts.generate_overloads import main as generate_overloads
+from videoipath_automation_tool.utils.script_utils import ROOT_DIR
+
+parser = argparse.ArgumentParser(description="Generate all version-specific code for a given VideoIPath version")
+parser.add_argument("version", help="Version of VideoIPath to use", default="2024.4.12", nargs="?")
+
 
+def main():
+    args = parser.parse_args()
+    schema_file = os.path.join(ROOT_DIR, "apps", "inventory", "model", "driver_schema", f"{args.version}.json")
+
+    if not os.path.exists(schema_file):
+        print(
+            f"VideoIPath version {args.version} is currently not supported. Please create an issue on https://github.com/SWR-MoIP/VideoIPath-Automation-Tool/issues to request support for this version."
+        )
+        exit(1)
 
-def main(
-    schema_file: str = DEFAULT_SCHEMA_FILE,
-    output_file: str = DEFAULT_OUTPUT_FILE,
-):
-    generate_driver_models(schema_file, output_file)
+    generate_driver_models(schema_file)
     generate_overloads()
 
 
 if __name__ == "__main__":
-    args = parser.parse_args()
-    main(args.schema_file, args.output_file)
+    main()

src/videoipath_automation_tool/scripts/generate_driver_models.py

@@ -4,7 +4,8 @@
 
 from videoipath_automation_tool.utils.script_utils import ROOT_DIR, load_module
 
-DEFAULT_SCHEMA_FILE = os.path.join(ROOT_DIR, "apps", "inventory", "model", "driver_schema", "2024.3.3.json")
+DEFAULT_VERSION = "2024.4.12"
+DEFAULT_SCHEMA_FILE = os.path.join(ROOT_DIR, "apps", "inventory", "model", "driver_schema", f"{DEFAULT_VERSION}.json")
 DEFAULT_OUTPUT_FILE = os.path.join(ROOT_DIR, "apps", "inventory", "model", "drivers.py")
 
 parser = argparse.ArgumentParser(description="Generate Pydantic models from driver schema")
@@ -134,7 +135,7 @@ def main(
 
 # Notes:
 # - The name of the custom settings model follows the naming convention: CustomSettings_<driver_organization>_<driver_name>_<driver_version> => "." and "-" are replaced by "_"!
-# - {schema_file} is used as reference to define the custom settings model!
+# - Schema {schema_file.split("/")[-1]} is used as reference to define the custom settings model!
 # - The "driver_id" attribute is necessary for the discriminator, which is used to determine the correct model for the custom settings in DeviceConfiguration!
 # - The "alias" attribute is used to map the attribute to the correct key (with driver organization & name) in the JSON payload for the API!
 # - "DriverLiteral" is used to provide a list of all possible drivers in the IDEs IntelliSense!