tinybird 0.0.1.dev90__tar.gz → 0.0.1.dev91__tar.gz

{tinybird-0.0.1.dev90 → tinybird-0.0.1.dev91}/PKG-INFO

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

{tinybird-0.0.1.dev90 → tinybird-0.0.1.dev91}/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.dev90'
-__revision__ = 'd11bb13'
+__version__ = '0.0.1.dev91'
+__revision__ = 'b965ddf'

{tinybird-0.0.1.dev90 → tinybird-0.0.1.dev91}/tinybird/tb/cli.py

@@ -8,6 +8,7 @@ import tinybird.tb.modules.auth
 import tinybird.tb.modules.build
 import tinybird.tb.modules.cli
 import tinybird.tb.modules.common
+import tinybird.tb.modules.connection
 import tinybird.tb.modules.copy
 import tinybird.tb.modules.create
 import tinybird.tb.modules.datasource

{tinybird-0.0.1.dev90 → tinybird-0.0.1.dev91}/tinybird/tb/modules/cli.py

@@ -412,7 +412,7 @@ async def create_ctx_client(ctx: Context, config: Dict[str, Any], cloud: bool, b
         return _get_tb_client(config.get("token", None), config["host"], staging=staging)
     build = command in commands_always_build or build
     if not build and command not in commands_always_local and command not in commands_always_build:
-        click.echo(FeedbackManager.gray(message="Running against Tinybird Local\n"))
+        click.echo(FeedbackManager.gray(message="Running against Tinybird Local"))
     return await get_tinybird_local_client(config, build=build, staging=staging)
 
 

tinybird-0.0.1.dev91/tinybird/tb/modules/connection.py

@@ -0,0 +1,125 @@
+# This is a command file for our CLI. Please keep it clean.
+#
+# - If it makes sense and only when strictly necessary, you can create utility functions in this file.
+# - But please, **do not** interleave utility functions and command definitions.
+
+from typing import Any, Dict, List, Optional
+
+import click
+from click import Context
+
+from tinybird.client import TinyB
+from tinybird.tb.modules.cli import cli
+from tinybird.tb.modules.common import (
+    DataConnectorType,
+    _get_setting_value,
+    coro,
+    echo_safe_humanfriendly_tables_format_smart_table,
+)
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+DATA_CONNECTOR_SETTINGS: Dict[DataConnectorType, List[str]] = {
+    DataConnectorType.KAFKA: [
+        "kafka_bootstrap_servers",
+        "kafka_sasl_plain_username",
+        "kafka_sasl_plain_password",
+        "cli_version",
+        "endpoint",
+        "kafka_security_protocol",
+        "kafka_sasl_mechanism",
+        "kafka_schema_registry_url",
+        "kafka_ssl_ca_pem",
+    ],
+    DataConnectorType.GCLOUD_SCHEDULER: ["gcscheduler_region"],
+    DataConnectorType.SNOWFLAKE: [
+        "account",
+        "username",
+        "password",
+        "role",
+        "warehouse",
+        "warehouse_size",
+        "stage",
+        "integration",
+    ],
+    DataConnectorType.BIGQUERY: ["account"],
+    DataConnectorType.GCLOUD_STORAGE: [
+        "gcs_private_key_id",
+        "gcs_client_x509_cert_url",
+        "gcs_project_id",
+        "gcs_client_id",
+        "gcs_client_email",
+        "gcs_private_key",
+    ],
+    DataConnectorType.GCLOUD_STORAGE_HMAC: [
+        "gcs_hmac_access_id",
+        "gcs_hmac_secret",
+    ],
+    DataConnectorType.GCLOUD_STORAGE_SA: ["account_email"],
+    DataConnectorType.AMAZON_S3: [
+        "s3_access_key_id",
+        "s3_secret_access_key",
+        "s3_region",
+    ],
+    DataConnectorType.AMAZON_S3_IAMROLE: [
+        "s3_iamrole_arn",
+        "s3_iamrole_region",
+        "s3_iamrole_external_id",
+    ],
+    DataConnectorType.AMAZON_DYNAMODB: [
+        "dynamodb_iamrole_arn",
+        "dynamodb_iamrole_region",
+        "dynamodb_iamrole_external_id",
+    ],
+}
+
+SENSITIVE_CONNECTOR_SETTINGS = {
+    DataConnectorType.KAFKA: ["kafka_sasl_plain_password"],
+    DataConnectorType.GCLOUD_SCHEDULER: [
+        "gcscheduler_target_url",
+        "gcscheduler_job_name",
+        "gcscheduler_region",
+    ],
+    DataConnectorType.GCLOUD_STORAGE_HMAC: ["gcs_hmac_secret"],
+    DataConnectorType.AMAZON_S3: ["s3_secret_access_key"],
+    DataConnectorType.AMAZON_S3_IAMROLE: ["s3_iamrole_arn"],
+    DataConnectorType.AMAZON_DYNAMODB: ["dynamodb_iamrole_arn"],
+}
+
+
+@cli.group()
+@click.pass_context
+def connection(ctx: Context) -> None:
+    """Connection commands."""
+
+
+@connection.command(name="ls")
+@click.option("--service", help="Filter by service")
+@click.pass_context
+@coro
+async def connection_ls(ctx: Context, service: Optional[DataConnectorType] = None) -> None:
+    """List connections."""
+    obj: Dict[str, Any] = ctx.ensure_object(dict)
+    client: TinyB = obj["client"]
+
+    connections = await client.connections(connector=service, skip_bigquery=True)
+    columns = []
+    table = []
+
+    click.echo(FeedbackManager.info_connections())
+
+    if not service:
+        sensitive_settings = []
+        columns = ["service", "name", "id", "connected_datasources"]
+    else:
+        sensitive_settings = SENSITIVE_CONNECTOR_SETTINGS.get(service, [])
+        columns = ["service", "name", "id", "connected_datasources"]
+        if connector_settings := DATA_CONNECTOR_SETTINGS.get(service):
+            columns += connector_settings
+
+    for connection in connections:
+        row = [_get_setting_value(connection, setting, sensitive_settings) for setting in columns]
+        table.append(row)
+
+    column_names = [c.replace("kafka_", "") for c in columns]
+    echo_safe_humanfriendly_tables_format_smart_table(table, column_names=column_names)
+    click.echo("\n")

{tinybird-0.0.1.dev90 → tinybird-0.0.1.dev91}/tinybird.egg-info/PKG-INFO

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

{tinybird-0.0.1.dev90 → tinybird-0.0.1.dev91}/tinybird.egg-info/SOURCES.txt

@@ -32,6 +32,7 @@ tinybird/tb/modules/cicd.py
 tinybird/tb/modules/cli.py
 tinybird/tb/modules/common.py
 tinybird/tb/modules/config.py
+tinybird/tb/modules/connection.py
 tinybird/tb/modules/copy.py
 tinybird/tb/modules/create.py
 tinybird/tb/modules/datasource.py