Auto Json Update

Author: ayon-ssp

gen_data/utils.py

@@ -1,13 +1,66 @@
 import json
+import arch as archAlt
+archAltName = archAlt.ReverseTranslate()
+
+def add_suite(JsonFile: dict, suite: str) -> None:
+    """ Add suite to the JsonFile
+
+    Args:
+        JsonFile (dict): The JsonFile
+        suite (str): The suite to add
+    """
+
+    JsonFile["suites"].append(suite)
+    JsonFile[suite] = {
+        "varients": []
+    }
+
+
+def add_varient(JsonFile: dict, suite: str, varient: str, Name: str, FirendlyName: str) -> None:
+    """ Add varient to the JsonFile"""
+    
+    JsonFile[suite]["varients"].append(varient)
+    JsonFile[suite][varient] = {
+        # adding default arch to the varients
+        "arch": [],
+        "Name": Name,
+        "FirendlyName": FirendlyName
+    }
+
+def add_arch(JsonFile: dict, suite: str, varient: str, arch:list[str]) -> None:
+    """ Add arch to the JsonFile
+
+    Args:
+        JsonFile (dict): The JsonFile
+        suite (str): The suite to add archs
+        varient (str): The varient to archs
+        arch (list[str]): The arch to add
+    """
+    
+    #revArchLst = {'armhf': ['armhf', 'arm'], 'aarch64': ['arm64', 'aarch64'], 'amd64': ['amd64', 'x86_64']}
+    revArchLst = archAltName[arch]
+    for revArch in revArchLst:
+        JsonFile[suite][varient]["arch"].append(revArch)
+        JsonFile[suite][varient][f"{arch}url"] = ""
+        JsonFile[suite][varient][f"{arch}sha"] = ""
+
 
 def resolv_data(
        json_data: dict,
        suite: str,
        variant: str,
-       arch: list = ...,
+       arch: list[str],
        Name: str = ...,
        FriendlyName: str = ...,
     ) -> dict:
-    # TODO yet to be implemented
-    pass
-
+    if suite not in json_data["suites"]:
+        add_suite(json_data,suite)
+    
+    if variant not in json_data[suite]["varients"]:
+        add_varient(json_data, suite, variant, Name, FriendlyName)
+    
+    for arc in arch:
+        if arch not in json_data[suite][variant]["arch"]:
+            add_arch(json_data, suite, variant, arc)
+    
+    return json_data