feat: fully working functionality for spinning up lab instances

Author: syntaxsdev

appwrite_lab/_orchestrator.py

@@ -43,11 +43,25 @@ def __init__(self, state: State, backend: str = "auto"):
             Path(__file__).parent / "templates" / "environment" / "dotenv"
         )
 
-    def get_labs(self):
+    def get_labs(self, collapsed: bool = False):
         """
         Get all labs.
         """
-        return self.state.get("labs", {})
+        labs: dict = self.state.get("labs", {})
+        if collapsed:
+            headers = ["Name", "Version", "URL", "Admin Email", "Project ID", "API Key"]
+            return headers, [
+                [
+                    val["name"],
+                    val["version"],
+                    val["url"],
+                    val["admin_email"],
+                    val["project_id"],
+                    val["api_key"],
+                ]
+                for val in labs.values()
+            ]
+        return labs
 
     def get_running_pods(self):
         """
@@ -267,7 +281,6 @@ def deploy_playwright_automation(
                 if result_file.exists():
                     with open(result_file, "r") as f:
                         data = f.read()
-                        print("RESULT FILE", data)
                         return data
                 else:
                     return None

appwrite_lab/cli/list_menu.py

@@ -1,5 +1,5 @@
 from appwrite_lab import get_global_labs
-from appwrite_lab.utils import print_table, console
+from appwrite_lab.utils import print_table, print_2d_table, console
 from appwrite_lab._orchestrator import get_template_versions
 import typer
 
@@ -11,12 +11,12 @@
 @list_menu.command(name="labs")
 def get_labs():
     """List all ephemeral Appwrite instances."""
-    pods = labs.orchestrator.get_labs()
-    console.print(pods)
-    # print_table(pods, ["Name", "Version", "Status"])
+    headers, pods = labs.orchestrator.get_labs(collapsed=True)
+    print_table(pods, headers)
+
 
 @list_menu.command()
 def versions():
     """List all available Appwrite versions."""
     versions = get_template_versions()
-    print_table(versions, ["Version"])
+    print_table([versions], ["Version"])

appwrite_lab/models.py

@@ -45,4 +45,4 @@ def generate_missing_config(self):
                 random.choices(string.ascii_letters + string.digits, k=16)
             )
         if not self.project_name:
-            self.project_name = "Default_{self.project_id}"
+            self.project_name = f"Default_{self.project_id}"

appwrite_lab/playwright/functions/create_user_and_api_key.py

@@ -55,8 +55,7 @@ async def create_user_and_api_key(playwright: Playwright) -> str:
     await browser.close()
 
     try:
-        with open("result.txt", "w") as f:
-            print("API KEY", api_key)
+        with open("/playwright/result.txt", "w") as f:
             f.write(api_key)
     except Exception as e:
         print("ERROR", e)

appwrite_lab/utils.py

@@ -18,12 +18,19 @@ def ensure_dir():
     return True
 
 
-def print_table(data: list[dict], headers: list[str]):
-    """Print a table of data."""
+def print_table(data: list[list], headers: list[str]):
+    """Print a 2D table of data.
+
+    Args:
+        data: A list of lists of data.
+        headers: A list of headers.
+    """
     console = Console()
-    table = Table(*headers, style="purple")
+    table = Table(style="purple")
+    for header in headers:
+        table.add_column(header, overflow="fold", style="purple")
     for row in data:
-        table.add_row(row)
+        table.add_row(*row)
     console.print(table)