codeassist command added

backend/src/zango/apps/shared/tenancy/management/commands/codeassist.py

@@ -0,0 +1,89 @@
+import json
+import os
+import requests
+import tarfile
+from time import sleep
+import subprocess
+import shutil
+
+from django.core.management.base import BaseCommand
+
+from zango.apps.shared.tenancy.models import TenantModel, Domain
+
+
+class Command(BaseCommand):
+
+    def add_arguments(self, parser):
+        super().add_arguments(parser)
+        parser.add_argument(
+            "--codeassist_endpoint",
+            help="The endpoint URL for codeassist.",
+            required=True,
+        )
+        parser.add_argument(
+            "--jsonspec", help="The schema name to be used.", required=True
+        )
+
+    def handle(self, *args, **options):
+        try:
+            spec = json.load(open(options["jsonspec"], "r"))
+            app_name = spec["app_name"]
+            domain = spec["domain"]
+            try:
+                TenantModel.objects.get(name=app_name)
+                print("App already exists, codeassist will not be applied")
+                return
+            except TenantModel.DoesNotExist:
+                pass
+            app, task_id = TenantModel.create(
+                name=app_name,
+                schema_name=app_name,
+                description="",
+                tenant_type="app",
+                status="staged",
+                timezone="Asia/Kolkata",
+                date_format="%d/%m/%Y",
+                datetime_format="%d %b %Y %I:%M %p",
+            )
+            while app.status == "staged":
+                print("Waiting for app to start")
+                sleep(1)
+                app = TenantModel.objects.get(id=app.id)
+                if app.status == "deployed":
+                    try:
+                        domain = Domain.objects.create(domain=domain, tenant=app)
+                    except Exception as e:
+                        print(e)
+                    print("App started successfully, proceeding with codeassist")
+                    break
+            resp = requests.post(
+                f"{options['codeassist_endpoint']}/generate-app",
+                json=spec,
+                headers={"Content-Type": "application/json"},
+            )
+            if resp.status_code == 200:
+                shutil.rmtree(f"workspaces/{app_name}")
+
+                file_path = f"{app_name}.tar"
+                with open(file_path, "wb") as f:
+                    for chunk in resp.iter_content(chunk_size=1024):
+                        if chunk:
+                            f.write(chunk)
+
+                with tarfile.open(file_path) as tar:
+                    tar.extractall(path=f"workspaces/")
+                    os.remove(file_path)
+            else:
+                raise Exception("Codeassist failed")
+            try:
+                subprocess.run(["python", "manage.py", "ws_makemigration", app_name])
+                subprocess.run(["zango", "update-apps", "--app_name", app_name])
+            except Exception as e:
+                print(e)
+        except json.JSONDecodeError as e:
+            print(f"Error while parsing jsonspec: {e}")
+            return
+        except Exception as e:
+            import traceback
+
+            traceback.print_exc()