tinybird 0.0.1.dev235__tar.gz → 0.0.1.dev237__tar.gz

{tinybird-0.0.1.dev235 → tinybird-0.0.1.dev237}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev235
+Version: 0.0.1.dev237
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird

{tinybird-0.0.1.dev235 → tinybird-0.0.1.dev237}/tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/forward/commands'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev235'
-__revision__ = 'c22959d'
+__version__ = '0.0.1.dev237'
+__revision__ = '7aa96d2'

{tinybird-0.0.1.dev235 → tinybird-0.0.1.dev237}/tinybird/tb/modules/local.py

@@ -25,11 +25,11 @@ def stop_tinybird_local(docker_client: DockerClient) -> None:
         pass
 
 
-def remove_tinybird_local(docker_client: DockerClient) -> None:
+def remove_tinybird_local(docker_client: DockerClient, persist_data: bool) -> None:
     """Remove the Tinybird container."""
     try:
         container = docker_client.containers.get(TB_CONTAINER_NAME)
-        if click.confirm(
+        if persist_data or click.confirm(
             FeedbackManager.warning(
                 message="△ This step will remove all your data inside Tinybird Local. Are you sure? [y/N]:"
             ),
@@ -130,7 +130,7 @@ def remove() -> None:
     """Remove Tinybird Local"""
     click.echo(FeedbackManager.highlight(message="» Removing Tinybird Local..."))
     docker_client = get_docker_client()
-    remove_tinybird_local(docker_client)
+    remove_tinybird_local(docker_client, persist_data=False)
     click.echo(FeedbackManager.success(message="✓ Tinybird Local removed"))
 
 
@@ -141,11 +141,16 @@ def remove() -> None:
     is_flag=True,
     help="Use local AWS credentials from your environment and pass them to the Tinybird docker container",
 )
-def start(use_aws_creds: bool) -> None:
+@click.option(
+    "--volumes-path",
+    default=None,
+    help="Path to the volumes directory. If not provided, the container data won't be persisted.",
+)
+def start(use_aws_creds: bool, volumes_path: str) -> None:
     """Start Tinybird Local"""
     click.echo(FeedbackManager.highlight(message="» Starting Tinybird Local..."))
     docker_client = get_docker_client()
-    start_tinybird_local(docker_client, use_aws_creds)
+    start_tinybird_local(docker_client, use_aws_creds, volumes_path)
     click.echo(FeedbackManager.success(message="✓ Tinybird Local is ready!"))
 
 
@@ -156,13 +161,18 @@ def start(use_aws_creds: bool) -> None:
     is_flag=True,
     help="Use local AWS credentials from your environment and pass them to the Tinybird docker container",
 )
-def restart(use_aws_creds: bool) -> None:
+@click.option(
+    "--volumes-path",
+    default=None,
+    help="Path to the volumes directory. If not provided, the container data won't be persisted.",
+)
+def restart(use_aws_creds: bool, volumes_path: str) -> None:
     """Restart Tinybird Local"""
     click.echo(FeedbackManager.highlight(message="» Restarting Tinybird Local..."))
     docker_client = get_docker_client()
-    remove_tinybird_local(docker_client)
+    remove_tinybird_local(docker_client, volumes_path is not None)
     click.echo(FeedbackManager.info(message="✓ Tinybird Local stopped"))
-    start_tinybird_local(docker_client, use_aws_creds)
+    start_tinybird_local(docker_client, use_aws_creds, volumes_path)
     click.echo(FeedbackManager.success(message="✓ Tinybird Local is ready!"))
 
 

{tinybird-0.0.1.dev235 → tinybird-0.0.1.dev237}/tinybird/tb/modules/local_common.py

@@ -106,7 +106,6 @@ def get_test_workspace_name(path: str) -> str:
 
 def get_local_tokens() -> Dict[str, str]:
     try:
-        # ruff: noqa: ASYNC210
         return requests.get(f"{TB_LOCAL_ADDRESS}/tokens").json()
     except Exception:
         # Check if tinybird-local is running using docker client (some clients use podman and won't have docker cmd)
@@ -217,6 +216,7 @@ def get_local_tokens() -> Dict[str, str]:
 def start_tinybird_local(
     docker_client: DockerClient,
     use_aws_creds: bool,
+    volumes_path: Optional[str] = None,
 ) -> None:
     """Start the Tinybird container."""
     pull_show_prompt = False
@@ -252,6 +252,13 @@ def start_tinybird_local(
         if container:
             container.remove(force=True)
 
+        volumes = {}
+        if volumes_path:
+            volumes = {
+                f"{volumes_path}/data": {"bind": "/var/lib/clickhouse", "mode": "rw"},
+                f"{volumes_path}/metadata": {"bind": "/redis-data", "mode": "rw"},
+            }
+
         container = docker_client.containers.run(
             TB_IMAGE_NAME,
             name=TB_CONTAINER_NAME,
@@ -260,6 +267,7 @@ def start_tinybird_local(
             remove=False,
             platform="linux/amd64",
             environment=environment,
+            volumes=volumes,
         )
 
     click.echo(FeedbackManager.info(message="* Waiting for Tinybird Local to be ready..."))

{tinybird-0.0.1.dev235 → tinybird-0.0.1.dev237}/tinybird/tb/modules/login.py

@@ -179,14 +179,14 @@ def login(host: Optional[str], auth_host: str, workspace: str, interactive: bool
             )
             click.echo(FeedbackManager.info(message=auth_url))
 
-            async def poll_for_tokens():
+            def poll_for_tokens():
                 while True:
                     params = {
                         "apiHost": host,
                         "cliCode": one_time_code,
                         "method": "code",
                     }
-                    response = requests.get(f"{auth_host}/api/cli-login?{urlencode(params)}")  # noqa: ASYNC210
+                    response = requests.get(f"{auth_host}/api/cli-login?{urlencode(params)}")
 
                     try:
                         if response.status_code == 200:
@@ -194,12 +194,12 @@ def login(host: Optional[str], auth_host: str, workspace: str, interactive: bool
                             user_token = data.get("user_token", "")
                             workspace_token = data.get("workspace_token", "")
                             if user_token and workspace_token:
-                                await authenticate_with_tokens(data, host, cli_config)
+                                authenticate_with_tokens(data, host, cli_config)
                                 break
                     except Exception:
                         pass
 
-                    time.sleep(2)  # noqa: ASYNC251
+                    time.sleep(2)
 
             poll_for_tokens()
             return

{tinybird-0.0.1.dev235 → tinybird-0.0.1.dev237}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev235
+Version: 0.0.1.dev237
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird