tinybird 0.0.1.dev89__tar.gz → 0.0.1.dev90__tar.gz

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

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

{tinybird-0.0.1.dev89 → tinybird-0.0.1.dev90}/tinybird/client.py

@@ -288,6 +288,25 @@ class TinyB:
             ds = [x for x in ds if x["name"].startswith(branch)]
         return ds
 
+    async def secrets(self) -> List[Dict[str, Any]]:
+        response = await self._req("/v0/variables")
+        return response["variables"]
+
+    async def get_secret(self, name: str) -> Optional[Dict[str, Any]]:
+        return await self._req(f"/v0/variables/{name}")
+
+    async def create_secret(self, name: str, value: str):
+        response = await self._req("/v0/variables", method="POST", data={"name": name, "value": value})
+        return response
+
+    async def update_secret(self, name: str, value: str):
+        response = await self._req(f"/v0/variables/{name}", method="PUT", data={"value": value})
+        return response
+
+    async def delete_secret(self, name: str):
+        response = await self._req(f"/v0/variables/{name}", method="DELETE")
+        return response
+
     async def get_connections(self, service: Optional[str] = None):
         params = {}
 

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

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

@@ -22,6 +22,7 @@ import tinybird.tb.modules.materialization
 import tinybird.tb.modules.mock
 import tinybird.tb.modules.pipe
 import tinybird.tb.modules.playground
+import tinybird.tb.modules.secret
 import tinybird.tb.modules.tag
 import tinybird.tb.modules.test
 import tinybird.tb.modules.token

tinybird-0.0.1.dev90/tinybird/tb/modules/secret.py

@@ -0,0 +1,85 @@
+import re
+from typing import Optional
+
+import click
+
+from tinybird.client import TinyB
+from tinybird.tb.modules.cli import cli
+from tinybird.tb.modules.common import coro, echo_safe_humanfriendly_tables_format_smart_table
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+
+@cli.group()
+@click.pass_context
+def secret(ctx):
+    """Secret commands."""
+
+
+@secret.command(name="ls")
+@click.option("--match", default=None, help="Retrieve any secrets matching the pattern. For example, --match _test")
+@click.pass_context
+@coro
+async def secret_ls(ctx: click.Context, match: Optional[str]):
+    """List secrets"""
+
+    client: TinyB = ctx.ensure_object(dict)["client"]
+    secrets = await client.secrets()
+    columns = ["name", "created_at", "updated_at"]
+    table_human_readable = []
+    table_machine_readable = []
+    pattern = re.compile(match) if match else None
+
+    for secret in secrets:
+        name = secret["name"]
+
+        if pattern and not pattern.search(name):
+            continue
+
+        created_at = secret["created_at"]
+        updated_at = secret["updated_at"]
+
+        table_human_readable.append((name, created_at, updated_at))
+        table_machine_readable.append({"name": name, "created at": created_at, "updated at": updated_at})
+
+    click.echo(FeedbackManager.info(message="** Secrets:"))
+    echo_safe_humanfriendly_tables_format_smart_table(table_human_readable, column_names=columns)
+    click.echo("\n")
+
+
+@secret.command(name="set")
+@click.argument("name")
+@click.argument("value")
+@click.pass_context
+@coro
+async def secret_set(ctx: click.Context, name: str, value: str):
+    """Create or update secrets"""
+    try:
+        click.echo(FeedbackManager.highlight(message=f"\n» Setting secret '{name}'..."))
+        client: TinyB = ctx.ensure_object(dict)["client"]
+        existing_secret = None
+        try:
+            existing_secret = await client.get_secret(name)
+        except Exception:
+            pass
+        if existing_secret:
+            await client.update_secret(name, value)
+        else:
+            await client.create_secret(name, value)
+        click.echo(FeedbackManager.success(message=f"\n✓ Secret '{name}' set"))
+    except Exception as e:
+        click.echo(FeedbackManager.error(message=f"✗ Error: {e}"))
+
+
+@secret.command(name="rm")
+@click.argument("name")
+@click.pass_context
+@coro
+async def secret_rm(ctx: click.Context, name: str):
+    """Delete a secret"""
+    try:
+        click.echo(FeedbackManager.highlight(message=f"\n» Deleting secret '{name}'..."))
+        client: TinyB = ctx.ensure_object(dict)["client"]
+        await client.delete_secret(name)
+        click.echo(FeedbackManager.success(message=f"\n✓ Secret '{name}' deleted"))
+    except Exception as e:
+        click.echo(FeedbackManager.error(message=f"✗ Error: {e}"))

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

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

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

@@ -53,6 +53,7 @@ tinybird/tb/modules/pipe.py
 tinybird/tb/modules/playground.py
 tinybird/tb/modules/project.py
 tinybird/tb/modules/regions.py
+tinybird/tb/modules/secret.py
 tinybird/tb/modules/shell.py
 tinybird/tb/modules/table.py
 tinybird/tb/modules/tag.py