tinybird 0.0.1.dev57__tar.gz → 0.0.1.dev58__tar.gz

{tinybird-0.0.1.dev57 → tinybird-0.0.1.dev58}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird
-Version: 0.0.1.dev57
+Version: 0.0.1.dev58
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird

{tinybird-0.0.1.dev57 → tinybird-0.0.1.dev58}/tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/cli/introduction.html'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev57'
-__revision__ = '0af16b9'
+__version__ = '0.0.1.dev58'
+__revision__ = '35a8d6e'

{tinybird-0.0.1.dev57 → tinybird-0.0.1.dev58}/tinybird/tb/modules/deployment.py

@@ -153,12 +153,18 @@ def deployment_group() -> None:
     default=False,
     help="Auto-promote the deployment. Only works if --wait is enabled. Disabled by default.",
 )
+@click.option(
+    "--check/--no-check",
+    is_flag=True,
+    default=False,
+    help="Validate the deployment before creating it. Disabled by default.",
+)
 @click.pass_context
-def deployment_create(ctx: click.Context, wait: bool, auto: bool) -> None:
+def deployment_create(ctx: click.Context, wait: bool, auto: bool, check: bool) -> None:
     """
     Validate and deploy the project server side.
     """
-    create_deployment(ctx, wait, auto)
+    create_deployment(ctx, wait, auto, check)
 
 
 @deployment_group.command(name="ls")
@@ -246,15 +252,21 @@ def deployment_rollback(ctx: click.Context, wait: bool) -> None:
     default=False,
     help="Auto-promote the deployment. Only works if --wait is enabled. Disabled by default.",
 )
+@click.option(
+    "--check",
+    is_flag=True,
+    default=False,
+    help="Validate the deployment before creating it. Disabled by default.",
+)
 @click.pass_context
-def deploy(ctx: click.Context, wait: bool, auto: bool) -> None:
+def deploy(ctx: click.Context, wait: bool, auto: bool, check: bool) -> None:
     """
     Deploy the project.
     """
-    create_deployment(ctx, wait, auto)
+    create_deployment(ctx, wait, auto, check)
 
 
-def create_deployment(ctx: click.Context, wait: bool, auto: bool) -> None:
+def create_deployment(ctx: click.Context, wait: bool, auto: bool, check: Optional[bool] = None) -> None:
     # TODO: This code is duplicated in build_server.py
     # Should be refactored to be shared
     MULTIPART_BOUNDARY_DATA_PROJECT = "data_project://"
@@ -281,11 +293,23 @@ def create_deployment(ctx: click.Context, wait: bool, auto: bool) -> None:
     deployment = None
     try:
         HEADERS = {"Authorization": f"Bearer {TINYBIRD_API_KEY}"}
-
-        r = requests.post(TINYBIRD_API_URL, files=files, headers=HEADERS)
+        params = {}
+        if check:
+            click.echo(FeedbackManager.highlight(message="\n» Validating deployment...\n"))
+            params["check"] = "true"
+        r = requests.post(TINYBIRD_API_URL, files=files, headers=HEADERS, params=params)
         result = r.json()
         logging.debug(json.dumps(result, indent=2))
 
+        if check:
+            print_changes(result, project)
+            status = result.get("result")
+            if status == "success":
+                click.echo(FeedbackManager.success(message="\n✓ Deployment is valid"))
+            else:
+                click.echo(FeedbackManager.error(message="\n✗ Deployment is not valid"))
+            return
+
         deploy_result = result.get("result")
         if deploy_result == "success":
             click.echo(FeedbackManager.success(message="Deployment submitted successfully"))
@@ -308,6 +332,8 @@ def create_deployment(ctx: click.Context, wait: bool, auto: bool) -> None:
                     click.echo(FeedbackManager.error(message=f"{deploy_error.get('error')}"))
         else:
             click.echo(FeedbackManager.error(message=f"Unknown build result {deploy_result}"))
+    except Exception as e:
+        click.echo(FeedbackManager.error_exception(error=e))
     finally:
         for fd in fds:
             fd.close()
@@ -343,3 +369,29 @@ def create_deployment(ctx: click.Context, wait: bool, auto: bool) -> None:
 
         if auto:
             promote_deployment(client.host, HEADERS, wait=wait)
+
+
+def print_changes(result: dict, project: Project) -> None:
+    deployment = result.get("deployment", {})
+    columns = ["status", "name", "path"]
+    resources: list[list[str]] = []
+
+    for ds in deployment.get("new_datasources", []):
+        resources.append(["new", ds, project.get_resource_path(ds, "datasource")])
+
+    for p in deployment.get("new_pipes", []):
+        resources.append(["new", p, project.get_resource_path(p, "pipe")])
+
+    for ds in deployment.get("changed_datasources", []):
+        resources.append(["modified", ds, project.get_resource_path(ds, "datasource")])
+
+    for p in deployment.get("changed_pipes", []):
+        resources.append(["modified", p, project.get_resource_path(p, "pipe")])
+
+    for ds in deployment.get("deleted_datasources", []):
+        resources.append(["deleted", ds, project.get_resource_path(ds, "datasource")])
+
+    for p in deployment.get("deleted_pipes", []):
+        resources.append(["deleted", p, project.get_resource_path(p, "pipe")])
+
+    echo_safe_humanfriendly_tables_format_smart_table(resources, column_names=columns)

{tinybird-0.0.1.dev57 → tinybird-0.0.1.dev58}/tinybird/tb/modules/project.py

@@ -32,6 +32,12 @@ class Project:
                 project_files.append(project_file)
         return project_files
 
+    def get_resource_path(self, resource_name: str, resource_type: str) -> Optional[str]:
+        full_path = next((p for p in self.get_project_files() if p.endswith(resource_name + f".{resource_type}")), "")
+        if not full_path:
+            return None
+        return Path(full_path).relative_to(self.path)
+
     def get_vendor_files(self) -> List[str]:
         vendor_files: List[str] = []
         for project_file in glob.glob(f"{self.vendor_path}/**/*.datasource", recursive=True):

{tinybird-0.0.1.dev57 → tinybird-0.0.1.dev58}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird
-Version: 0.0.1.dev57
+Version: 0.0.1.dev58
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird