Move generator script

generate.py → gen/generate.py

@@ -4,17 +4,22 @@
 
 import json, re, sys, copy
 
-def to_snake_case(text):
-    text = re.sub(r'(?<!^)(?=[A-Z])', '_', text).lower()
-    return text.replace("/", "_")
-
 resource_key_mapping = {
     "routing_interface_id": "interface_id",
     "switch_access_policie_id": "access_policy_number",
     "switch_routing_multicast_rendezvous_point_id": "rendezvous_point_id",
     "routing_static_route_id": "static_route_id",
 }
 
+URL = sys.argv[1]
+OPEN_API_SPEC_PATH = "/meraki/openapi/spec3.json"
+
+
+def to_snake_case(text):
+    text = re.sub(r"(?<!^)(?=[A-Z])", "_", text).lower()
+    return text.replace("/", "_")
+
+
 def generate_loop(path, depth, prev_var, path_from_networks):
     if len(path) == 1:
         ids = ""
@@ -24,14 +29,33 @@ def generate_loop(path, depth, prev_var, path_from_networks):
             path_from_networks = path_from_networks[:-1]
         for p in path_from_networks:
             id_path = items_from_networks + [p]
-            resource_key = to_snake_case(p)[:-1]+"_id"
+            resource_key = to_snake_case(p)[:-1] + "_id"
             resource_key = resource_key_mapping.get(resource_key, resource_key)
             ret_resource_keys.append(resource_key)
-            ids += "{spaces}{resource_key} = meraki_network_{val}.net_{val}[\"${{domain.name}}/${{organization.name}}/${{network.name}}/{access_path}\"].{resource_key}\n".format(spaces=" "*(depth+2)*2, resource_key=resource_key, val="_".join([to_snake_case(i) for i in id_path]), access_path="/".join([to_snake_case(i)+"/${"+to_snake_case(i)[:-1]+".name}" for i in id_path]))
+            ids += '{spaces}{resource_key} = meraki_network_{val}.net_{val}["${{domain.name}}/${{organization.name}}/${{network.name}}/{access_path}"].{resource_key}\n'.format(
+                spaces=" " * (depth + 2) * 2,
+                resource_key=resource_key,
+                val="_".join([to_snake_case(i) for i in id_path]),
+                access_path="/".join(
+                    [
+                        to_snake_case(i) + "/${" + to_snake_case(i)[:-1] + ".name}"
+                        for i in id_path
+                    ]
+                ),
+            )
             items_from_networks += [p]
-        return """
+        return (
+            """
     {spaces}network_id = meraki_network.network["${{domain.name}}/${{organization.name}}/${{network.name}}"].id\n{ids}
-    {spaces}data = try({prev_var}, null)""".format(spaces=" "*depth*2, prev_var=prev_var if path[0] == "" else prev_var+"."+to_snake_case(path[0]), ids=ids[:-1]), ret_resource_keys
+    {spaces}data = try({prev_var}, null)""".format(
+                spaces=" " * depth * 2,
+                prev_var=prev_var
+                if path[0] == ""
+                else prev_var + "." + to_snake_case(path[0]),
+                ids=ids[:-1],
+            ),
+            ret_resource_keys,
+        )
 
     step = path[0]
     local_var = to_snake_case(step[:-1] if step[-1] == "s" else step)
@@ -41,14 +65,28 @@ def generate_loop(path, depth, prev_var, path_from_networks):
     else:
         lbrace = "["
         rbrace = "]"
-    content, ret_resource_keys = generate_loop(path[1:], depth+1, local_var, path_from_networks)
-    return """
+    content, ret_resource_keys = generate_loop(
+        path[1:], depth + 1, local_var, path_from_networks
+    )
+    return (
+        """
     {spaces}for {local_var} in try({prev_var}.{step}, []) : {lbrace}{spaces}{content}
-    {spaces}{rbrace} if try({prev_var}.{step}, null) != null""".format(local_var=local_var, prev_var=prev_var, step=to_snake_case(step), content=content, spaces=" "*depth*2, lbrace=lbrace, rbrace=rbrace), ret_resource_keys
+    {spaces}{rbrace} if try({prev_var}.{step}, null) != null""".format(
+            local_var=local_var,
+            prev_var=prev_var,
+            step=to_snake_case(step),
+            content=content,
+            spaces=" " * depth * 2,
+            lbrace=lbrace,
+            rbrace=rbrace,
+        ),
+        ret_resource_keys,
+    )
+
 
 template = """
 
-### marcin generated
+### template generated
 ### resource name: {RESOURCE_NAME}
 ### values: {VALUES}
 
@@ -66,39 +104,49 @@ def generate_loop(path, depth, prev_var, path_from_networks):
 }}
 """
 
+
 def generate_template(resource, values, content, ret_resource_keys):
     values_render = ""
     for v in values:
         v_terra = "_".join([to_snake_case(x) for x in v])
         v_dots = ".".join([to_snake_case(x) for x in v])
         if v_terra not in ret_resource_keys:
-            values_render += "  {v_terra} = try(each.value.data.{v_dots}, local.defaults.meraki.networks.{r}.{v_dots}, null)\n".format(v_terra=v_terra, v_dots=v_dots, r=resource)
+            values_render += "  {v_terra} = try(each.value.data.{v_dots}, local.defaults.meraki.networks.{r}.{v_dots}, null)\n".format(
+                v_terra=v_terra, v_dots=v_dots, r=resource
+            )
     keys_render = ""
     for k in ret_resource_keys:
         keys_render += "  {k} = each.value.{k}\n".format(k=k)
-    return template.format(RESOURCE_NAME=resource, VALUES=values, values_render=values_render, content=content, keys_render=keys_render)
+    return template.format(
+        RESOURCE_NAME=resource,
+        VALUES=values,
+        values_render=values_render,
+        content=content,
+        keys_render=keys_render,
+    )
 
-with open("/Users/maparafi/openapi/openapi/spec3.json") as f:
+
+with open(OPEN_API_SPEC_PATH) as f:
     j = json.loads(f.read())
-
-URL = sys.argv[1]
 
 res = {}
 spec_url = ""
 
+
 def fields(d, p):
     ret = []
     if d["type"] == "object":
         if "properties" not in d:
             # ret.append(copy.deepcopy(p))
             return ret
         for prop, propValue in d["properties"].items():
-           propRet = fields(propValue, p+[prop])
-           ret += propRet
+            propRet = fields(propValue, p + [prop])
+            ret += propRet
     else:
         ret.append(copy.deepcopy(p))
     return ret
 
+
 METHOD = "put"
 
 for k, v in j["paths"].items():
@@ -108,10 +156,14 @@ def fields(d, p):
         spec_url = k
 
 path = [x.split("}")[-1].strip("/").replace("/", "_") for x in spec_url.split("{")]
-res_name = "_".join([to_snake_case(x) for x in spec_url.split("/") if not x.startswith("{")][1:])
+res_name = "_".join(
+    [to_snake_case(x) for x in spec_url.split("/") if not x.startswith("{")][1:]
+)
 path = ["domains", "organizations"] + path
 path_from_networks = path[3:-1]
 print(path_from_networks)
-res_fields = fields(res[METHOD]["requestBody"]["content"]["application/json"]["schema"], [])
+res_fields = fields(
+    res[METHOD]["requestBody"]["content"]["application/json"]["schema"], []
+)
 content, ret_resource_keys = generate_loop(path, 0, "local.meraki", path_from_networks)
 print(generate_template(res_name, res_fields, content, ret_resource_keys))