tinybird 0.0.1.dev230__py3-none-any.whl → 0.0.1.dev232__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.dev230'
-__revision__ = '45e2fd2'
+__version__ = '0.0.1.dev232'
+__revision__ = 'd58086c'

tinybird/tb/modules/connection.py

@@ -398,6 +398,15 @@ async def connection_create_kafka(ctx: Context) -> Tuple[str, str, str, str, str
         except Exception as e:
             raise CLIConnectionException(FeedbackManager.error(message=str(e)))
 
+    security_protocol_options = ["SASL_SSL", "PLAINTEXT"]
+    security_protocol = click.prompt(
+        FeedbackManager.highlight(message="? Security Protocol (SASL_SSL, PLAINTEXT) [SASL_SSL]"),
+        type=click.Choice(security_protocol_options),
+        show_default=False,
+        show_choices=False,
+        default="SASL_SSL",
+    )
+
     sasl_mechanism_options = ["PLAIN", "SCRAM-SHA-256", "SCRAM-SHA-512"]
     sasl_mechanism = click.prompt(
         FeedbackManager.highlight(message="? SASL Mechanism (PLAIN, SCRAM-SHA-256, SCRAM-SHA-512) [PLAIN]"),
@@ -455,7 +464,7 @@ async def connection_create_kafka(ctx: Context) -> Tuple[str, str, str, str, str
     )
     validate_kafka_auto_offset_reset(auto_offset_reset)
     click.echo(FeedbackManager.gray(message="» Validating connection..."))
-    security_protocol = "SASL_SSL"
+
     topics = list_kafka_topics(bootstrap_servers, key, secret, security_protocol, sasl_mechanism)
 
     if topics is None:
@@ -470,11 +479,11 @@ async def connection_create_kafka(ctx: Context) -> Tuple[str, str, str, str, str
         tb_secret_key=tb_secret_key,
         secret=secret,
         tb_secret_secret=tb_secret_secret,
-        security_protocol="SASL_SSL",
+        security_protocol=security_protocol,
         sasl_mechanism=sasl_mechanism,
         folder=project.folder,
     )
-    click.echo(FeedbackManager.info(message=f"/connections/{name}.connection"))
+    click.echo(FeedbackManager.info_file_created(file=f"connections/{name}.connection"))
     click.echo(FeedbackManager.success(message="✓ Connection created!"))
     return (
         name,

tinybird/tb/modules/datasource.py

@@ -788,7 +788,6 @@ async def datasource_create(
     connection_required = datasource_type in ("kafka", "s3", "gcs")
 
     if connection_required:
-        click.echo(FeedbackManager.gray(message="\n» Creating .connection file..."))
 
         def get_connection_files():
             connection_files = []
@@ -801,6 +800,8 @@ async def datasource_create(
             return connection_files
 
         connection_files = get_connection_files()
+
+        click.echo(FeedbackManager.gray(message="\n» Selecting connection..."))
         connection_name = ""
         topics: List[str] = []
         if len(connection_files) == 0:
@@ -849,7 +850,7 @@ async def datasource_create(
                         folder=project.folder,
                     )
                 if datasource_type != "kafka":
-                    click.echo(FeedbackManager.info(message=f"/connections/{connection_name}.connection"))
+                    click.echo(FeedbackManager.info_file_created(file=f"connections/{connection_name}.connection"))
                     click.echo(FeedbackManager.success(message="✓ .connection created!"))
                 connection_files = get_connection_files()
             else:
@@ -857,9 +858,29 @@ async def datasource_create(
                 return
 
         if not connection_file:
-            connection_file = connection_files[0]
-            connection_path = Path(connection_file)
-            connection = connection_path.stem
+            if len(connection_files) > 1:
+                click.echo(
+                    FeedbackManager.highlight(
+                        message=f"? Multiple {datasource_types[datasource_type][0]} connections found. Please select one:"
+                    )
+                )
+                connection_index = -1
+                while connection_index == -1:
+                    for index, conn_file in enumerate(connection_files):
+                        conn_path = Path(conn_file)
+                        click.echo(f"  [{index + 1}] {conn_path.stem}")
+                    connection_index = click.prompt("\nSelect option", default=1)
+                    try:
+                        connection_file = connection_files[int(connection_index) - 1]
+                        connection_path = Path(connection_file)
+                        connection = connection_path.stem
+                    except Exception:
+                        connection_index = -1
+            else:
+                connection_file = connection_files[0]
+                connection_path = Path(connection_file)
+                connection = connection_path.stem
+                click.echo(FeedbackManager.info(message=f"Using connection: {connection}"))
 
     click.echo(FeedbackManager.gray(message="\n» Creating .datasource file..."))
 

{tinybird-0.0.1.dev230.dist-info → tinybird-0.0.1.dev232.dist-info}/METADATA

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

{tinybird-0.0.1.dev230.dist-info → tinybird-0.0.1.dev232.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=74d7saA-6Srgl_5lAe0EHS41KrA0HpZjHHjG-kLMPA8,247
+tinybird/tb/__cli__.py,sha256=0UBC5RBj5wMDEwCGOSS0ufzt8GHPKZxxes7QxuvmHeo,247
 tinybird/tb/check_pypi.py,sha256=i3l2L8IajeB7sreikR7oPlYJki9MtS3c_M4crnmbByc,760
 tinybird/tb/cli.py,sha256=0xYk2Ip4vb3nNFbxfTdG3VoIgdRvUKVbUVU_mviErPA,1107
 tinybird/tb/client.py,sha256=FKj61vY9STPW03kfVcxYuY1_csI-kP-mc1ERQfqJtg8,56505
@@ -27,10 +27,10 @@ tinybird/tb/modules/cicd.py,sha256=H22MaMsqAeDv-OmeHFOlUOAmfNzEzTFaEV9TQ0g4ehs,7
 tinybird/tb/modules/cli.py,sha256=I1G7q3-hjYZuwq1GQoZd5_2i8btl33s6ms8I0yF5XDc,16332
 tinybird/tb/modules/common.py,sha256=F6oaoFZ3aBxEMjiDKYhthsEIUqSFPkcdlMJ7h7A49Ac,83114
 tinybird/tb/modules/config.py,sha256=VnzYVUo4q1RBEEMMce4_OCrKp4erhgkRPHElydVlKj0,11488
-tinybird/tb/modules/connection.py,sha256=KHZgu7sK1C4UBgwkKAMeNdKB0RrTls0LG3YoaUfeAfw,17648
+tinybird/tb/modules/connection.py,sha256=2W4fh7C5pKJryFnC2NUUIktF4xBs1ILqmHnzuEp3ftQ,17970
 tinybird/tb/modules/copy.py,sha256=zHN1d5NA-MFsgbk2kKJq2P9qA8dNOnIsIa60QpVnSwc,4458
 tinybird/tb/modules/create.py,sha256=56_x6nwj_oAr72X7AvdxXAuXCDKGJp-w1dP6E2XdxqU,23376
-tinybird/tb/modules/datasource.py,sha256=UiQXDkSEmQIPj4NcUD_I4bAJe9IGVmlVW07yGgSu6kY,40941
+tinybird/tb/modules/datasource.py,sha256=9ncWkDVJ3dMc15q2GME40gHZf8REymOIjNn76m4CUNo,42028
 tinybird/tb/modules/deployment.py,sha256=ByXIgEvwxB49pJEKKj0EJIfORWyflCYr04k8961nBkA,28391
 tinybird/tb/modules/deprecations.py,sha256=rrszC1f_JJeJ8mUxGoCxckQTJFBCR8wREf4XXXN-PRc,4507
 tinybird/tb/modules/dev_server.py,sha256=57FCKuWpErwYUYgHspYDkLWEm9F4pbvVOtMrFXX1fVU,10129
@@ -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.dev230.dist-info/METADATA,sha256=v4gUOshqYC8ekjqDD-6Vii3ijifIHdWAVrDJCPi5Br0,1682
-tinybird-0.0.1.dev230.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev230.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev230.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev230.dist-info/RECORD,,
+tinybird-0.0.1.dev232.dist-info/METADATA,sha256=tYkjFtgiHT1P3mBLaOYdj53Z8RrUr_Bn5kp_x6u7mWA,1682
+tinybird-0.0.1.dev232.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev232.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev232.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev232.dist-info/RECORD,,