PyPI - tinybird - Versions diffs - 0.0.1.dev55__tar.gz → 0.0.1.dev56__tar.gz - Mend

@@ -21,11 +21,12 @@ class AuthHandler(http.server.SimpleHTTPRequestHandler):
         self.send_response(200)
         self.send_header("Content-type", "text/html")
         self.end_headers()
-        self.wfile.write(b"""
+        self.wfile.write(
+            """
         <html>
         <head>
             <style>
-                body {
+                body {{
                     font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
                     background: #f5f5f5;
                     display: flex;
@@ -33,7 +34,7 @@ class AuthHandler(http.server.SimpleHTTPRequestHandler):
                     justify-content: center;
                     height: 100vh;
                     margin: 0;
-                }
+                }}
             </style>
         </head>
         <body>
@@ -41,14 +42,16 @@ class AuthHandler(http.server.SimpleHTTPRequestHandler):
                 const searchParams = new URLSearchParams(window.location.search);
                 const code = searchParams.get('code');
                 const workspace = searchParams.get('workspace');
-                fetch('/?code=' + code, {method: 'POST'})
-                    .then(() => {
-                        window.location.href = "https://cloud.tinybird.co/cli-login?workspace=" + workspace;
-                    });
+                const host = "{auth_host}";
+                fetch('/?code=' + code, {{method: 'POST'}})
+                    .then(() => {{
+                        window.location.href = host + "/cli-login?workspace=" + workspace;
+                    }});
             </script>
         </body>
         </html>
-        """)
+        """.format(auth_host=self.server.auth_host).encode()
+        )
     def do_POST(self):
         parsed_path = urllib.parse.urlparse(self.path)
@@ -75,13 +78,14 @@ AUTH_SERVER_PORT = 49160
 class AuthServer(socketserver.TCPServer):
     allow_reuse_address = True
-    def __init__(self, server_address, RequestHandlerClass, auth_callback):
+    def __init__(self, server_address, RequestHandlerClass, auth_callback, auth_host):
         self.auth_callback = auth_callback
+        self.auth_host = auth_host
         super().__init__(server_address, RequestHandlerClass)
-def start_server(auth_callback):
-    with AuthServer(("", AUTH_SERVER_PORT), AuthHandler, auth_callback) as httpd:
+def start_server(auth_callback, auth_host):
+    with AuthServer(("", AUTH_SERVER_PORT), AuthHandler, auth_callback, auth_host) as httpd:
         httpd.timeout = 30
         start_time = time.time()
         while time.time() - start_time < 60:  # Run for a maximum of 60 seconds
@@ -109,6 +113,9 @@ async def login(host: str, auth_host: str, workspace: str):
     auth_event = threading.Event()
     auth_code: list[str] = []  # Using a list to store the code, as it's mutable
+    if auth_host == "https://cloud.tinybird.co" and "wadus" in host:
+        auth_host = "https://cloud-wadus.tinybird.co"
     def auth_callback(code):
         auth_code.append(code)
         auth_event.set()
@@ -116,7 +123,7 @@ async def login(host: str, auth_host: str, workspace: str):
     click.echo(FeedbackManager.highlight(message="» Opening browser for authentication..."))
     # Start the local server in a separate thread
-    server_thread = threading.Thread(target=start_server, args=(auth_callback,))
+    server_thread = threading.Thread(target=start_server, args=(auth_callback, auth_host))
     server_thread.daemon = True
     server_thread.start()

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

@@ -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.dev55'
-__revision__ = 'cc01364'
+__version__ = '0.0.1.dev56'
+__revision__ = 'e22d9a3'

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

tinybird 0.0.1.dev55tar.gz → 0.0.1.dev56tar.gz

Potentially problematic release.

tinybird 0.0.1.dev55__tar.gz → 0.0.1.dev56__tar.gz

Potentially problematic release.

tinybird 0.0.1.dev55tar.gz → 0.0.1.dev56tar.gz