tinybird 0.0.1.dev102__py3-none-any.whl → 0.0.1.dev103__py3-none-any.whl

tinybird/tb/modules/local.py

@@ -3,6 +3,7 @@ import re
 import subprocess
 import time
 
+import boto3
 import click
 
 import docker
@@ -51,6 +52,8 @@ def start_tinybird_local(
         if container:
             container.remove(force=True)
 
+        environment = get_local_aws_credentials()
+
         container = docker_client.containers.run(
             TB_IMAGE_NAME,
             name=TB_CONTAINER_NAME,
@@ -58,6 +61,7 @@ def start_tinybird_local(
             ports={"80/tcp": TB_LOCAL_PORT},
             remove=False,
             platform="linux/amd64",
+            environment=environment,
         )
 
     click.echo(FeedbackManager.info(message="* Waiting for Tinybird Local to be ready..."))
@@ -89,6 +93,32 @@ def get_docker_client():
         )
 
 
+def get_local_aws_credentials():
+    credentials: dict[str, str] = {}
+    try:
+        # Get the boto3 session and credentials
+        session = boto3.Session()
+        creds = session.get_credentials()
+
+        if creds:
+            # Create environment variables for the container based on boto credentials
+            credentials["AWS_ACCESS_KEY_ID"] = creds.access_key
+            credentials["AWS_SECRET_ACCESS_KEY"] = creds.secret_key
+
+            # Add session token if it exists (for temporary credentials)
+            if creds.token:
+                credentials["AWS_SESSION_TOKEN"] = creds.token
+
+            # Add region if available
+            if session.region_name:
+                credentials["AWS_DEFAULT_REGION"] = session.region_name
+    except Exception:
+        # TODO (rbarbadillo): We should handle this better. If users don't have AWS credentials, most times it's fine
+        # but if they want to use S3, they'll need a warning.
+        pass
+    return credentials
+
+
 def stop_tinybird_local(docker_client):
     """Stop the Tinybird container."""
     try:

tinybird/tb/modules/workspace.py

@@ -12,10 +12,8 @@ from tinybird.client import TinyB
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import (
     _get_workspace_plan_name,
-    ask_for_organization_interactively,
-    ask_for_organization_name,
+    ask_for_organization,
     coro,
-    create_organization_and_add_workspaces,
     create_workspace_interactive,
     create_workspace_non_interactive,
     echo_safe_humanfriendly_tables_format_smart_table,
@@ -24,7 +22,6 @@ from tinybird.tb.modules.common import (
     get_user_token,
     is_valid_starterkit,
     print_current_workspace,
-    sort_organizations_by_user,
     switch_workspace,
 )
 from tinybird.tb.modules.config import CLIConfig
@@ -128,27 +125,9 @@ async def create_workspace(
     organization_name = None
     organizations = await get_organizations_by_user(config, user_token)
 
-    if organization_id:
-        organization = next((org for org in organizations if org.get("id") == organization_id), None)
-        if not organization:
-            raise CLIWorkspaceException(FeedbackManager.error_organization_not_found(organization_id=organization_id))
-        organization_name = organization.get("name")
-    else:
-        if len(organizations) == 0:
-            organization_name = await ask_for_organization_name(config)
-            organization = await create_organization_and_add_workspaces(config, organization_name, user_token)
-            organization_id = organization.get("id")
-        elif len(organizations) == 1:
-            organization_id = organizations[0].get("id")
-            organization_name = organizations[0].get("name")
-        else:
-            sorted_organizations = sort_organizations_by_user(organizations, user_email=config.get_user_email())
-            current_organization = await ask_for_organization_interactively(sorted_organizations)
-            if current_organization:
-                organization_id = current_organization.get("id")
-                organization_name = current_organization.get("name")
-            else:
-                return
+    organization_id, organization_name = await ask_for_organization(organizations, organization_id)
+    if not organization_id:
+        return
 
     # If we have at least workspace_name, we start the non interactive
     # process, creating an empty workspace

{tinybird-0.0.1.dev102.dist-info → tinybird-0.0.1.dev103.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev102
+Version: 0.0.1.dev103
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird
@@ -9,6 +9,7 @@ Requires-Python: >=3.9, <3.14
 Description-Content-Type: text/x-rst
 Requires-Dist: aiofiles==24.1.0
 Requires-Dist: anthropic==0.42.0
+Requires-Dist: boto3
 Requires-Dist: click<8.2,>=8.1.6
 Requires-Dist: clickhouse-toolset==0.34.dev0
 Requires-Dist: colorama==0.4.6
@@ -32,29 +33,11 @@ Requires-Dist: wheel
 Requires-Dist: packaging<24,>=23.1
 Requires-Dist: llm>=0.19
 Requires-Dist: thefuzz==0.22.1
-Provides-Extra: bigquery
-Requires-Dist: gsutil==4.58; extra == "bigquery"
-Requires-Dist: google-api-python-client==2.0.2; extra == "bigquery"
-Requires-Dist: google-auth==1.27.1; extra == "bigquery"
-Requires-Dist: google-auth-httplib2==0.1.0; extra == "bigquery"
-Requires-Dist: google-cloud-storage==2.4.0; extra == "bigquery"
-Requires-Dist: google-cloud-bigquery==2.11.0; extra == "bigquery"
-Provides-Extra: snowflake
-Requires-Dist: snowflake-connector-python~=3.12.3; extra == "snowflake"
-Requires-Dist: gsutil==4.58; extra == "snowflake"
-Requires-Dist: google-api-python-client==2.0.2; extra == "snowflake"
-Requires-Dist: google-auth==1.27.1; extra == "snowflake"
-Requires-Dist: google-auth-httplib2==0.1.0; extra == "snowflake"
-Requires-Dist: google-cloud-storage==2.4.0; extra == "snowflake"
-Requires-Dist: oauth2client==3.0.0; extra == "snowflake"
-Requires-Dist: chardet<4,>=3.0.2; extra == "snowflake"
-Requires-Dist: pyOpenSSL<20.0.0,>=16.2.0; extra == "snowflake"
 Dynamic: author
 Dynamic: author-email
 Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
-Dynamic: provides-extra
 Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary

{tinybird-0.0.1.dev102.dist-info → tinybird-0.0.1.dev103.dist-info}/RECORD

@@ -4,7 +4,7 @@ tinybird/config.py,sha256=5UP_UZ2Qtlm5aOH5W7SbtN8r7X-8u3-r853joKqU5zs,6072
 tinybird/connectors.py,sha256=7Gjms7b5MAaBFGi3xytsJurCylprONpFcYrzp4Fw2Rc,15241
 tinybird/context.py,sha256=FfqYfrGX_I7PKGTQo93utaKPDNVYWelg4Hsp3evX5wM,1291
 tinybird/datatypes.py,sha256=XNypumfqNjsvLJ5iNXnbVHRvAJe0aQwI3lS6Cxox-e0,10979
-tinybird/feedback_manager.py,sha256=YSjtFDJvc8y66j2J0iIkb3SVzDdYAJbzFL-JPQ26pak,68761
+tinybird/feedback_manager.py,sha256=a_ZhFX2zcB7vRknIcmHKMdQbb0c7TqlTBQ_5hPuWh88,69267
 tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
 tinybird/prompts.py,sha256=RgV1B7-b6-1xXStNlhKikYSCG1WNaddDVeSp8hSgBlE,33753
 tinybird/sql.py,sha256=J35bhdpuu84HW2tiLp-cs_nzkRwPhiy1yPcFhcWMCR4,46248
@@ -15,28 +15,28 @@ tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
 tinybird/tornado_template.py,sha256=jjNVDMnkYFWXflmT8KU_Ssbo5vR8KQq3EJMk5vYgXRw,41959
 tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj0,3890
 tinybird/ch_utils/engine.py,sha256=BZuPM7MFS7vaEKK5tOMR2bwSAgJudPrJt27uVEwZmTY,40512
-tinybird/tb/__cli__.py,sha256=wj1R7TGTdVnIOEJgofqjYqrfWRx9JGWFQnBLQy53RqQ,252
+tinybird/tb/__cli__.py,sha256=tRfbprNM4lcNzOdUF9bZBJUiPalFJgLPA5lSLBIJUYQ,252
 tinybird/tb/cli.py,sha256=H_HaZhkimKgkryYXpBjHfY9Qtg-ZORiONU3psDNpzDk,1135
 tinybird/tb/modules/auth.py,sha256=L1IatO2arRSzys3t8px8xVt8uPWUL5EVD0sFzAV_uVU,9022
 tinybird/tb/modules/build.py,sha256=h5drdmDFX8NHts9dA2Zepao7KSgMAl3DZGyFufVZP78,11085
-tinybird/tb/modules/cicd.py,sha256=ldjswh70j0lQpReZGb6YbOcSIVYhrMUQFdJTfAfownU,5671
-tinybird/tb/modules/cli.py,sha256=9VrTDe6HlarG8h6wn_u7PgaRTCMZ4Hi_qiE3U2KHo2k,17912
-tinybird/tb/modules/common.py,sha256=QhSPoLKz3J892z3qi0BBEOxLKmBU6f0MK7pmKIFIkVk,81427
+tinybird/tb/modules/cicd.py,sha256=cCq2NaW_YYlODoJLQGJE1cPNrHGgOqRP6Jkq554h3f8,5679
+tinybird/tb/modules/cli.py,sha256=XtdwMI5uPoKfhlpSuCuhzODRJfTCx1oXHz_bn3x1OBg,16047
+tinybird/tb/modules/common.py,sha256=e5g4hAELeb2QdIxrn9lJEtBQUV96peHKhNKptVEKBbg,83198
 tinybird/tb/modules/config.py,sha256=FqdLpLaKpYubqw3xkB4EX06ufZYDgGRxONR_9i-y-KE,11416
 tinybird/tb/modules/connection.py,sha256=WKeDxbTpSsQ1PUmsT730g3S5RT2PtR5mPpVEanD1nbM,3933
 tinybird/tb/modules/copy.py,sha256=MAVqKip8_QhOYq99U_XuqSO6hCLJEh5sFtbhcXtI3SI,5802
-tinybird/tb/modules/create.py,sha256=VvY5C0GNM_j3ZPK4ba8yq3cephJ356PaAsBWR-zoPXs,14054
+tinybird/tb/modules/create.py,sha256=FuJYuFqWy2kgSfxSwkQ5nExZgEYALyq_z9dc1d-FRN0,14078
 tinybird/tb/modules/datasource.py,sha256=dNCK9iCR2xPLfwqqwg2ixyE6NuoVEiJU2mBZBmOYrVY,16906
 tinybird/tb/modules/deployment.py,sha256=KMpIahNGUdea3KcH0jTMEnzQ-7zg7M4esd2zTfvaC-k,19337
 tinybird/tb/modules/endpoint.py,sha256=EhVoGAXsFz-83Fiwj1gI-I73iRRvL49d0W81un7hvPE,12080
 tinybird/tb/modules/exceptions.py,sha256=4A2sSjCEqKUMqpP3WI00zouCWW4uLaghXXLZBSw04mY,3363
 tinybird/tb/modules/feedback_manager.py,sha256=7nNiOx7OMebiheLED1r0d75SbuXCNxyBmF4e20rCBNc,69511
 tinybird/tb/modules/fmt.py,sha256=qpf9APqKTKL2uphNgdbj4OMVyLkAxZn6dn4eHF99L5g,3553
-tinybird/tb/modules/infra.py,sha256=vb6WmVHFuU1ZhtQMh2wOzRd2x8g7CLt_2gR3EL_L2E4,22867
+tinybird/tb/modules/infra.py,sha256=4grSpfcm0gOHiMCYoIh26jx2mzeEEqgyMX8PetRMAWI,33246
 tinybird/tb/modules/job.py,sha256=956Pj8BEEsiD2GZsV9RKKVM3I_CveOLgS82lykO5ukk,2963
 tinybird/tb/modules/llm.py,sha256=AC0VSphTOM2t-v1_3NLvNN_FIbgMo4dTyMqIv5nniPo,835
 tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCfxvg,3469
-tinybird/tb/modules/local.py,sha256=revEEYP-Oq5nqURgmZmc0kt1_tEjM31dyUtcH5YbOuc,5842
+tinybird/tb/modules/local.py,sha256=SS13uDtH6xhhE9fseQH5cGtHWw9nsrMF1gD5wkGEVcw,6936
 tinybird/tb/modules/local_common.py,sha256=RN5OEncHdq7ua4AZ--WgKtaFuEsLvIhq_ROHJadRXXA,3188
 tinybird/tb/modules/login.py,sha256=dAaBBn_ZIQRm9BFl6Uw_HgZa2qoLuAO6mBOxob_17ms,6253
 tinybird/tb/modules/logout.py,sha256=ULooy1cDBD02-r7voZmhV7udA0ML5tVuflJyShrh56Y,1022
@@ -55,13 +55,13 @@ tinybird/tb/modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09
 tinybird/tb/modules/test.py,sha256=FUU-drY8mdjNoKsw16O5ZqvYvZqzycrZBEpSwbhGDUE,11456
 tinybird/tb/modules/token.py,sha256=OhqLFpCHVfYeBCxJ0n7n2qoho9E9eGcUfHgL7R1MUVQ,13485
 tinybird/tb/modules/watch.py,sha256=poNJOUNDESDNn80H2dHvE6X6pIu-t9MZFi59_TxVN2U,8822
-tinybird/tb/modules/workspace.py,sha256=GUXSROaPn2LpR2ZTYLIlj-wZnAbiFkib9m3Cag-Kbeg,7791
+tinybird/tb/modules/workspace.py,sha256=pk9_6fNxXGM1JPQcjTz7Sj77qn9ML27lGIyYEHPwORE,6611
 tinybird/tb/modules/workspace_members.py,sha256=Vb5XEaKmkfONyfg2MS5EcpwolMvv7GLwFS5m2EuobT8,8726
 tinybird/tb/modules/datafile/build.py,sha256=jhfIJ2xt0N13XsLPe3iMQIyCPApHS13_Df2LfISYtT8,50952
 tinybird/tb/modules/datafile/build_common.py,sha256=rT7VJ5mnQ68R_8US91DAtkusfvjWuG_NObOzNgtN_ko,4562
 tinybird/tb/modules/datafile/build_datasource.py,sha256=CCU3eQ8Rax9RgHHfbAXDRL6rQ49N35h_GDQnGrUUUzA,17379
 tinybird/tb/modules/datafile/build_pipe.py,sha256=w-Wd08gZYAEcak9FdBijVfIU2_Wn_PPdgAZddPpoGTo,11382
-tinybird/tb/modules/datafile/common.py,sha256=ZRRx-9EXeHMOz6Lu9Y1TSWELF0ftfUHLLQAANjJZ334,82241
+tinybird/tb/modules/datafile/common.py,sha256=HUuhEQ7lC5pqOxUw0NT76qwKlKjSlVAvaIIaXdgqMss,82327
 tinybird/tb/modules/datafile/diff.py,sha256=-0J7PsBO64T7LOZSkZ4ZFHHCPvT7cKItnJkbz2PkndU,6754
 tinybird/tb/modules/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1wnI,556
 tinybird/tb/modules/datafile/fixture.py,sha256=si-9LB-LdKQSWDtVW82xDrHtFfko5bgBG1cvjqqrcPU,1064
@@ -81,8 +81,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.dev102.dist-info/METADATA,sha256=b33GgOxL3oZcglnCCC5gDCW-o6CQY67vZ1NalxV2QvQ,2586
-tinybird-0.0.1.dev102.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev102.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev102.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev102.dist-info/RECORD,,
+tinybird-0.0.1.dev103.dist-info/METADATA,sha256=JB1oi-8Up8AsrEXhKLGoR0zXe7_8nTG9tG0UAQ45Wf4,1612
+tinybird-0.0.1.dev103.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev103.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev103.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev103.dist-info/RECORD,,