tinybird 0.0.1.dev235__py3-none-any.whl → 0.0.1.dev237__py3-none-any.whl

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/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/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/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.dist-info → tinybird-0.0.1.dev237.dist-info}/METADATA

@@ -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.dist-info → tinybird-0.0.1.dev237.dist-info}/RECORD

@@ -17,7 +17,7 @@ tinybird/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1w
 tinybird/datafile/parse_connection.py,sha256=tRyn2Rpr1TeWet5BXmMoQgaotbGdYep1qiTak_OqC5E,1825
 tinybird/datafile/parse_datasource.py,sha256=ssW8QeFSgglVFi3sDZj_HgkJiTJ2069v2JgqnH3CkDE,1825
 tinybird/datafile/parse_pipe.py,sha256=xf4m0Tw44QWJzHzAm7Z7FwUoUUtr7noMYjU1NiWnX0k,3880
-tinybird/tb/__cli__.py,sha256=b4nj8uwvsmk28yWwIoYdrpRZlirPtMdv4UPFgQiG-ZQ,247
+tinybird/tb/__cli__.py,sha256=vmMQxvQ7ux44Mfpu0EDdpMaLEOM4WmWMZ3dbG8YeI5s,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=_5hIbxDFdLjMq0GiXpdoMIz7TdTpQ0DHfAS9MKdlST4,975
 tinybird/tb/client.py,sha256=cHJky5SF-z_epBajNovo_gaIRdMsU5KXx7vtVl4sdgE,53957
@@ -42,9 +42,9 @@ tinybird/tb/modules/infra.py,sha256=JE9oLIyF4bi_JBoe-BgZ5HhKp_lQgSihuSV1KIS02Qs,
 tinybird/tb/modules/job.py,sha256=wBsnu8UPTOha2rkLvucgmw4xYv73ubmui3eeSIF68ZM,3107
 tinybird/tb/modules/llm.py,sha256=QbHRcMLgFmLKEh4zVb2ctR_5tIGUGdFJrAiRCDtMxDw,1572
 tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCfxvg,3469
-tinybird/tb/modules/local.py,sha256=YIzmv0Hm63WhXGRAEFcpPWB0IVrZl1yrvY1VSKl_fD8,5986
-tinybird/tb/modules/local_common.py,sha256=BrtNjF5UhoD522LhpNBVHfIkJMFaGuXYpvv7I4mm-dw,17171
-tinybird/tb/modules/login.py,sha256=vZ7chla0mR9KlagUtLQLW_DxW9xkF3ZNTobavsfppyM,12545
+tinybird/tb/modules/local.py,sha256=HTLUFb8xqR8_PEFkvZV1UB-uk2L66_DAh6eflce6_FE,6446
+tinybird/tb/modules/local_common.py,sha256=_WODjW3oPshgsZ1jDFFx2nr0zrLi3Gxz5dlahWPobM8,17464
+tinybird/tb/modules/login.py,sha256=glqj5RWH26AseEoBl8XfrSDEjQTdko17i_pVWOIMoGc,12497
 tinybird/tb/modules/logout.py,sha256=sniI4JNxpTrVeRCp0oGJuQ3yRerG4hH5uz6oBmjv724,1009
 tinybird/tb/modules/materialization.py,sha256=0O2JUCxLzz-DrXTUewVHlIyC6-Kyymw0hGXXDicMSHE,5403
 tinybird/tb/modules/mock.py,sha256=rHJtbxIKpwnWcSK4nqb-tFlObmyn2G7hR7LNw3aeFWs,5321
@@ -82,8 +82,8 @@ tinybird/tb_cli_modules/config.py,sha256=IsgdtFRnUrkY8-Zo32lmk6O7u3bHie1QCxLwgp4
 tinybird/tb_cli_modules/exceptions.py,sha256=pmucP4kTF4irIt7dXiG-FcnI-o3mvDusPmch1L8RCWk,3367
 tinybird/tb_cli_modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
 tinybird/tb_cli_modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09K1A,10449
-tinybird-0.0.1.dev235.dist-info/METADATA,sha256=bqFXMTlHLyOTGGaRSBujVPJu530jgHjhfAVGQz4KpCk,1682
-tinybird-0.0.1.dev235.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev235.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev235.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev235.dist-info/RECORD,,
+tinybird-0.0.1.dev237.dist-info/METADATA,sha256=Wx48Mkja_8CCOe1o8FJFVlz9M36gjsZHPKZK86YVQDo,1682
+tinybird-0.0.1.dev237.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev237.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev237.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev237.dist-info/RECORD,,