...

_archive/openapi/openapi.py

@@ -0,0 +1,50 @@
+from openapi_python_client.schema import OpenAPI
+
+import json
+import yaml
+
+from generator.server.vlang.vlang import VlangCodeGenerator
+
+class OpenApiCodeGenerator:
+    def __init__(self, lang: str, spec_file: str, output_dir: str):
+        self.lang = lang
+        self.spec_file = spec_file
+        self.output_dir = output_dir
+
+    def _read_file(self):
+        """
+        Read the OpenAPI spec file.
+        """
+        if self.spec_file.endswith(".json"):
+            with open(self.spec_file, "r") as file:
+                return file.read()  # Return raw JSON string
+        elif self.spec_file.endswith(".yaml"):
+            with open(self.spec_file, "r") as file:
+                # Convert YAML to JSON string for compatibility
+                return json.dumps(yaml.safe_load(file))
+        else:
+            raise ValueError("Unsupported file format")
+
+
+    def generate(self):
+        """
+        Main generation logic for code based on the OpenAPI spec.
+        """
+        file_content = self._read_file()
+        openapi = OpenAPI.model_validate_json(file_content)
+        if self.lang == "vlang":
+            vlang_code_generator = VlangCodeGenerator(
+                python_code=openapi, output_dir=self.output_dir
+            )
+            vlang_code_generator.generate()
+        elif self.lang == "python":
+            print("Python code generation not implemented yet.")
+
+
+if __name__ == "__main__":
+    s = OpenApiCodeGenerator(
+        lang="vlang",
+        spec_file="/home/thunder/work/codescalers/github/hero_server_python/lib/openapi/schema.json",
+        output_dir="./output"
+    )
+    s.generate()