lkr-dev-cli 0.0.22__py3-none-any.whl → 0.0.23__py3-none-any.whl

lkr/auth/oauth.py

@@ -19,21 +19,23 @@ def kill_process_on_port(port: int, retries: int = 5, delay: float = 1) -> None:
         # Try to create a socket binding to check if port is in use
         sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-        sock.bind(('localhost', port))
+        sock.bind(("localhost", port))
         sock.close()
         return  # Port is free, no need to kill anything
     except socket.error:
         # Port is in use, try to kill the process
-        if os.name == 'posix':  # macOS/Linux
-            os.system(f'lsof -ti tcp:{port} | xargs kill -9 2>/dev/null')
-        elif os.name == 'nt':  # Windows
-            os.system(f'for /f "tokens=5" %a in (\'netstat -aon ^| find ":{port}"\') do taskkill /F /PID %a 2>nul')
+        if os.name == "posix":  # macOS/Linux
+            os.system(f"lsof -ti tcp:{port} | xargs kill -9 2>/dev/null")
+        elif os.name == "nt":  # Windows
+            os.system(
+                f'for /f "tokens=5" %a in (\'netstat -aon ^| find ":{port}"\') do taskkill /F /PID %a 2>nul'
+            )
         # After killing, wait for the port to be free
         for _ in range(retries):
             try:
                 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                 sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-                sock.bind(('localhost', port))
+                sock.bind(("localhost", port))
                 sock.close()
                 return
             except socket.error:
@@ -45,19 +47,21 @@ class OAuthCallbackHandler(http.server.BaseHTTPRequestHandler):
     def do_GET(self):
         """Handle the callback from OAuth authorization"""
         self.send_response(200)
-        self.send_header('Content-type', 'text/html')
+        self.send_header("Content-type", "text/html")
         self.end_headers()
-        
+
         # Parse the authorization code from query parameters
         query_components = parse_qs(urllib.parse.urlparse(self.path).query)
-        
+
         # Store the code in the server instance
-        if 'code' in query_components:
-            self.server.auth_code = query_components['code'][0] # type: ignore
-        
+        if "code" in query_components:
+            self.server.auth_code = query_components["code"][0]  # type: ignore
+
         # Display a success message to the user
-        self.wfile.write(b"Authentication successful! You can close this window.")
-        
+        self.wfile.write(
+            b"Successfully authenticated to Looker OAuth! You can close this window."
+        )
+
         # Shutdown the server
         threading.Thread(target=self.server.shutdown).start()
 
@@ -65,18 +69,22 @@ class OAuthCallbackHandler(http.server.BaseHTTPRequestHandler):
         """Suppress logging of requests"""
         pass
 
+
 class OAuthCallbackServer(socketserver.TCPServer):
     def __init__(self, server_address):
         super().__init__(server_address, OAuthCallbackHandler)
         self.auth_code: str | None = None
 
+
 class LoginResponse(TypedDict):
     auth_code: Optional[str]
     code_verifier: Optional[str]
 
+
 class OAuth2PKCE:
     def __init__(self, new_token_callback: NewTokenCallback, use_production: bool):
         from lkr.auth_service import DbOAuthSession
+
         self.auth_code: Optional[str] = None
         self.state = secrets.token_urlsafe(16)
         self.new_token_callback: NewTokenCallback = new_token_callback
@@ -85,7 +93,7 @@ class OAuth2PKCE:
         self.server: OAuthCallbackServer | None = None
         self.port: int = 8000
         self.use_production: bool = use_production
-        
+
     def cleanup(self):
         """Clean up the server and its thread."""
         if self.server:
@@ -100,51 +108,64 @@ class OAuth2PKCE:
         """
         Initiates the OAuth2 PKCE login flow by opening the browser with the authorization URL
         and starting a local server to catch the callback.
-        
+
         Returns:
             Optional[str]: The authorization code if successful, None otherwise
         """
         from lkr.auth_service import get_auth_session
-        
+
         # Kill any process using port 8000
         kill_process_on_port(self.port)
-        
+
         # Start the local server
-        self.server = OAuthCallbackServer(('localhost', self.port))
-        
+        self.server = OAuthCallbackServer(("localhost", self.port))
+
         # Start the server in a separate thread
         self.server_thread = threading.Thread(target=self.server.serve_forever)
         self.server_thread.daemon = True
         self.server_thread.start()
 
         # Construct and open the OAuth URL
-        self.auth_session = get_auth_session(base_url, self.new_token_callback, use_production=self.use_production)
-        oauth_url = self.auth_session.create_auth_code_request_url('cors_api', self.state)
-        
+        self.auth_session = get_auth_session(
+            base_url, self.new_token_callback, use_production=self.use_production
+        )
+        oauth_url = self.auth_session.create_auth_code_request_url(
+            "cors_api", self.state
+        )
+
         webbrowser.open(oauth_url)
-        
+
         # Wait for the callback
         self.server_thread.join()
-        
+
         # Get the authorization code
-        return LoginResponse(auth_code=self.server.auth_code, code_verifier=self.auth_session.code_verifier)
+        return LoginResponse(
+            auth_code=self.server.auth_code,
+            code_verifier=self.auth_session.code_verifier,
+        )
 
     def exchange_code_for_token(self):
         """
         Exchange the authorization code for access and refresh tokens.
-        
+
         Args:
             base_url: The base URL of the Looker instance
             client_id: The OAuth client ID
-            
+
         Returns:
             Dict containing access_token, refresh_token, token_type, and expires_in
         """
         if not self.auth_code:
-            raise ValueError("No authorization code available. Must call initiate_login first.")
+            raise ValueError(
+                "No authorization code available. Must call initiate_login first."
+            )
         if not self.auth_session:
-            raise ValueError("No auth session available. Must call initiate_login first.")
-        self.auth_session.redeem_auth_code(self.auth_code, self.auth_session.code_verifier)
+            raise ValueError(
+                "No auth session available. Must call initiate_login first."
+            )
+        self.auth_session.redeem_auth_code(
+            self.auth_code, self.auth_session.code_verifier
+        )
         self.cleanup()
         return self.auth_session.token
 

lkr/auth_service.py

@@ -20,7 +20,7 @@ from lkr.constants import LOOKER_API_VERSION, OAUTH_CLIENT_ID, OAUTH_REDIRECT_UR
 from lkr.logging import logger
 from lkr.types import NewTokenCallback
 
-__all__ = ["get_auth"]
+__all__ = ["get_auth", "ApiKeyAuthSession", "DbOAuthSession"]
 
 
 def get_auth(ctx: typer.Context | LkrCtxObj) -> Union["SqlLiteAuth", "ApiKeyAuth"]:
@@ -33,7 +33,7 @@ def get_auth(ctx: typer.Context | LkrCtxObj) -> Union["SqlLiteAuth", "ApiKeyAuth
             raise typer.Exit(1)
     if lkr_ctx.use_sdk == "api_key" and lkr_ctx.api_key:
         logger.info("Using API key authentication")
-        return ApiKeyAuth(lkr_ctx.api_key)
+        return ApiKeyAuth(lkr_ctx.api_key, use_production=lkr_ctx.use_production)
     else:
         return SqlLiteAuth(lkr_ctx)
 
@@ -70,6 +70,36 @@ class OAuthApiSettings(ApiSettings):
         )
 
 
+class ApiKeyAuthSession(AuthSession):
+    def __init__(self, *args, use_production: bool, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.use_production = use_production
+
+    def _login(self, *args, **kwargs):
+        super()._login(*args, **kwargs)
+        if not self.use_production:
+            self._switch_to_dev_mode()
+
+    def _switch_to_dev_mode(self):
+        logger.debug("Switching to dev mode")
+        config = self.settings.read_config()
+        if "base_url" in config:
+            url = f"{config['base_url']}/api/{LOOKER_API_VERSION}/session"
+            return self.transport.request(
+                method=HttpMethod.PATCH,
+                path=url,
+                body=json.dumps({"workspace_id": "dev"}).encode("utf-8"),
+                transport_options={
+                    "headers": {
+                        "Content-Type": "application/json",
+                        "Authorization": f"Bearer {self.token.access_token}",
+                    }
+                },
+            )
+        else:
+            raise ValueError("Base URL not found in settings")
+
+
 class DbOAuthSession(OAuthSession):
     def __init__(
         self,
@@ -137,18 +167,19 @@ def get_auth_session(
     return auth
 
 
-def init_api_key_sdk(api_key: LookerApiKey) -> Looker40SDK:
+def init_api_key_sdk(api_key: LookerApiKey, use_production: bool) -> Looker40SDK:
     from looker_sdk.rtl import serialize
 
     settings = ApiKeyApiSettings(api_key)
     settings.is_configured()
     transport = RequestsTransport.configure(settings)
     return Looker40SDK(
-        auth=AuthSession(
+        auth=ApiKeyAuthSession(
             settings,
             transport,
             serialize.deserialize40,  # type: ignore
             LOOKER_API_VERSION,
+            use_production=use_production,
         ),
         deserialize=serialize.deserialize40,  # type: ignore
         serialize=serialize.serialize40,  # type: ignore
@@ -395,9 +426,7 @@ class SqlLiteAuth:
             return current_auth.instance_name
         return None
 
-    def get_current_sdk(
-        self, prompt_refresh_invalid_token: bool = False
-    ) -> Looker40SDK:
+    def get_current_sdk(self, prompt_refresh_invalid_token: bool = True) -> Looker40SDK:
         current_auth = self._get_current_auth()
         if current_auth:
             if not current_auth.valid_refresh_token:
@@ -480,8 +509,9 @@ class SqlLiteAuth:
 
 
 class ApiKeyAuth:
-    def __init__(self, api_key: LookerApiKey):
+    def __init__(self, api_key: LookerApiKey, use_production: bool):
         self.api_key = api_key
+        self.use_production = use_production
 
     def __enter__(self):
         return self
@@ -518,7 +548,7 @@ class ApiKeyAuth:
         )
 
     def get_current_sdk(self, **kwargs) -> Looker40SDK:
-        return init_api_key_sdk(self.api_key)
+        return init_api_key_sdk(self.api_key, self.use_production)
 
     def get_current_instance(self) -> str | None:
         raise NotImplementedError(

lkr/classes.py

@@ -13,24 +13,25 @@ class LookerApiKey(BaseModel):
     def from_env(cls):
         try:
             return cls(
-                client_id=os.environ.get("LOOKERSDK_CLIENT_ID"), # type: ignore
-                client_secret=os.environ.get("LOOKERSDK_CLIENT_SECRET"), # type: ignore
-                base_url=os.environ.get("LOOKERSDK_BASE_URL"), # type: ignore
+                client_id=os.environ.get("LOOKERSDK_CLIENT_ID"),  # type: ignore
+                client_secret=os.environ.get("LOOKERSDK_CLIENT_SECRET"),  # type: ignore
+                base_url=os.environ.get("LOOKERSDK_BASE_URL"),  # type: ignore
             )
         except Exception:
             return None
 
-        
+
 class LkrCtxObj(BaseModel):
     api_key: LookerApiKey | None
     force_oauth: bool = False
-    
+    use_production: bool = True
+
     @property
     def use_sdk(self) -> Literal["oauth", "api_key"]:
         if self.force_oauth:
             return "oauth"
         return "api_key" if self.api_key else "oauth"
-    
+
     def __init__(self, api_key: LookerApiKey | None = None, *args, **kwargs):
         super().__init__(api_key=api_key, *args, **kwargs)
         if not self.api_key:

lkr/main.py

@@ -1,5 +1,5 @@
 import os
-from typing import Annotated
+from typing import Annotated, Optional
 
 import typer
 
@@ -30,6 +30,7 @@ def callback(
     log_level: Annotated[LogLevel | None, typer.Option(envvar="LOG_LEVEL")] = None,
     quiet: Annotated[bool, typer.Option("--quiet")] = False,
     force_oauth: Annotated[bool, typer.Option("--force-oauth")] = False,
+    dev: Annotated[Optional[bool], typer.Option("--dev")] = None,
 ):
     if client_id:
         os.environ["LOOKERSDK_CLIENT_ID"] = client_id
@@ -43,7 +44,16 @@ def callback(
     # Initialize ctx.obj as a dictionary if it's None
     if ctx.obj is None:
         ctx.obj = {}
-    ctx.obj["ctx_lkr"] = LkrCtxObj(force_oauth=force_oauth)
+
+    ctx_obj = LkrCtxObj(
+        force_oauth=force_oauth,
+        use_production=not dev if dev is not None else True,
+    )
+    ctx.obj["ctx_lkr"] = ctx_obj
+    # if the user passes --dev, but lkrCtxObj.use_sdk is oauth, then we need to log a warning saying we're ignoring the --dev flag
+    if dev and ctx_obj.use_sdk == "oauth":
+        logger.warning("Ignoring --dev flag because OAuth token tracks dev/prod mode.")
+
     if log_level:
         from lkr.logging import set_log_level
 

lkr/mcp/main.py

@@ -101,7 +101,7 @@ class SpectaclesRequest(BaseModel):
 
 
 def get_mcp_sdk(ctx: LkrCtxObj | typer.Context):
-    sdk = get_auth(ctx).get_current_sdk()
+    sdk = get_auth(ctx).get_current_sdk(prompt_refresh_invalid_token=False)
     sdk.auth.settings.agent_tag += "-mcp"
     return sdk
 

lkr/observability/embed_container.html

@@ -32,10 +32,12 @@
         const urlParams = new URLSearchParams(window.location.search);
         const iframeUrl = urlParams.get('iframe_url');
         const origin = new URL(iframeUrl).origin;
-        console.log(origin)
-        console.log(iframeUrl)
         const sessionId = urlParams.get('session_id');
+        const debug = urlParams.get('debug');
         // Set the iframe source
+        if (debug) {
+            console.log({debug, iframeUrl, origin, sessionId})
+        }
         document.getElementById('looker-iframe').src = iframeUrl;
 
         // Track which events we've received
@@ -55,7 +57,7 @@
                 return;
             }
             const {type, ...data} = JSON.parse(event.data)
-            if (!trackedEvents.has(type)) {
+            if (!trackedEvents.has(type) && !debug) {
                 return;
             }
             if (type) {
@@ -65,6 +67,9 @@
                     event_data: data,
                     timestamp: now.toISOString(),
                 };
+                if (debug) {
+                    console.log(eventData)
+                }
                 // Send event data to the server
                 fetch(`/log_event?session_id=${sessionId}`, {
                     method: 'POST',

lkr/observability/main.py

@@ -158,14 +158,30 @@ def health_check(
             params.model_dump(mode="json"), "health_check_start", session_id
         )
         chrome_options = Options()
+        chrome_options.set_capability("goog:loggingPrefs", {"browser": "ALL"})
         chrome_options.add_argument("--headless=new")
         chrome_options.add_argument("--no-sandbox")
+        chrome_options.add_argument("--disable-dev-shm-usage")
+        chrome_options.add_argument("--disable-gpu")
+        chrome_options.add_argument("--enable-logging")
+        chrome_options.add_argument("--v=1")
+
+        chrome_options.add_experimental_option(
+            "prefs",
+            {
+                "profile.default_content_settings.cookies": 1,
+                "profile.cookie_controls_mode": 0,
+            },
+        )
         driver = webdriver.Chrome(options=chrome_options)
         url = observability_ctx.sdk.create_sso_embed_url(
             body=params.to_embed_sso_params(
                 observability_ctx.origin, observability_ctx.base_url or ""
             )
         )
+        observability_ctx.log_event(
+            {"sso_url": url.url}, "create_sso_embed_url", session_id
+        )
 
         if not (url and url.url):
             raise HTTPException(status_code=500, detail="No URL found")
@@ -175,11 +191,16 @@ def health_check(
         else:
             quoted_url = quote(url.url, safe="")
             embed_url = f"{observability_ctx.origin}/?iframe_url={quoted_url}&session_id={session_id}"
-            if True:
-                driver.get(embed_url)
-                WebDriverWait(driver, observability_ctx.timeout).until(
-                    EC.presence_of_element_located((By.ID, "completion-indicator"))
-                )
+            driver.get(embed_url)
+            observability_ctx.log_event(
+                {"url": embed_url}, "chromium_driver_get", session_id
+            )
+            WebDriverWait(driver, observability_ctx.timeout).until(
+                EC.presence_of_element_located((By.ID, "completion-indicator"))
+            )
+            observability_ctx.log_event(
+                {"session_id": session_id}, "chromium_driver_get_complete", session_id
+            )
 
     except TimeoutException:
         observability_ctx.log_event(
@@ -209,6 +230,9 @@ def health_check(
             session_id,
         )
     finally:
+        observability_ctx.log_event(
+            {"session_id": session_id}, "health_check_complete", session_id
+        )
         if driver:
             driver.quit()
         if not redirect:

{lkr_dev_cli-0.0.22.dist-info → lkr_dev_cli-0.0.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lkr-dev-cli
-Version: 0.0.22
+Version: 0.0.23
 Summary: lkr: a command line interface for looker
 Author: bwebs
 License-Expression: MIT

lkr_dev_cli-0.0.23.dist-info/RECORD

@@ -0,0 +1,21 @@
+lkr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lkr/auth_service.py,sha256=gwKbNXa2v6oSc01h7Vlhrld12uQgrFd5NkbxYnxjGS0,19140
+lkr/classes.py,sha256=f2TJOXFta0s8LJLEXOqPdWPLg-EIIntUSDS6gDOon7M,1163
+lkr/constants.py,sha256=DdCfsV6q8wgs2iHpIQeb6oDP_2XejusEHyPvCbaM3yY,108
+lkr/exceptions.py,sha256=M_aR4YaCZtY4wyxhcoqJCVkxVu9z3Wwo5KgSDyOoEnI,210
+lkr/logging.py,sha256=bBGdxkVrUidRTJAvqGHyQ5KS6IJGYDCoool5tTB_JxM,1822
+lkr/main.py,sha256=R3WSyj1AklinWE7wntuV9JSO_a6SsPhE3iOJE1nUn30,2371
+lkr/types.py,sha256=feJ-W2U61PJTiotMLuZJqxrotA53er95kO1O30mooy4,323
+lkr/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lkr/auth/main.py,sha256=ebGp-NdgULodHaDrLKspnGSFf_sOkV2Vanil1NbOJKY,7382
+lkr/auth/oauth.py,sha256=THynLg8ejAPl4RJ7q4PugEsLq4FFB0wKCHhDeNO9NDo,6106
+lkr/mcp/main.py,sha256=91LQ8Ny6VTOBRRcDI58Z1c8na9TyfgJfZ8nPjmt1n6w,22887
+lkr/observability/classes.py,sha256=5kQnHxa3pgUartrbccuG0yMClfoFqYCXY8iCTJTZvgk,6025
+lkr/observability/embed_container.html,sha256=IcDG-QVsYYNGQGrkDrx9OMZ2Pmo4C8oAjRHddFQ7Tlw,2939
+lkr/observability/main.py,sha256=1KCmJQLb6ht2-l2TXlN8Y8CCfpIP_E5TcMTy2VMaboA,9545
+lkr/observability/utils.py,sha256=Hv3g60cI03cQwyEe8QV7bUMssE6pyrZKnDyl9rxm5b8,2915
+lkr_dev_cli-0.0.23.dist-info/METADATA,sha256=M-UniLnNT-vHWfPc5xC4h09X5WzmSmjkwh5BQZrfr2o,10663
+lkr_dev_cli-0.0.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lkr_dev_cli-0.0.23.dist-info/entry_points.txt,sha256=nn2sFMGDpwUVE61ZUpbDPnQZkW7Gc08nV-tyLGo8q34,37
+lkr_dev_cli-0.0.23.dist-info/licenses/LICENSE,sha256=hKnCOORW1JRE_M2vStz8dblS5u1iR-2VpqS9xagKNa0,1063
+lkr_dev_cli-0.0.23.dist-info/RECORD,,

lkr_dev_cli-0.0.22.dist-info/RECORD

@@ -1,21 +0,0 @@
-lkr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lkr/auth_service.py,sha256=_enIsYxpL1IILtBfNG4YAA_H9cqwnwcvrGMyfmzgH7U,17815
-lkr/classes.py,sha256=0NKttc5kAhW_omuxwnN9Jy6oQnRKgI7b2c7HZUfI0lc,1144
-lkr/constants.py,sha256=DdCfsV6q8wgs2iHpIQeb6oDP_2XejusEHyPvCbaM3yY,108
-lkr/exceptions.py,sha256=M_aR4YaCZtY4wyxhcoqJCVkxVu9z3Wwo5KgSDyOoEnI,210
-lkr/logging.py,sha256=bBGdxkVrUidRTJAvqGHyQ5KS6IJGYDCoool5tTB_JxM,1822
-lkr/main.py,sha256=5m036LAS4PTLIzpWW5ohxIs2FGYtVf-OwYnTmtrpnxU,1933
-lkr/types.py,sha256=feJ-W2U61PJTiotMLuZJqxrotA53er95kO1O30mooy4,323
-lkr/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lkr/auth/main.py,sha256=ebGp-NdgULodHaDrLKspnGSFf_sOkV2Vanil1NbOJKY,7382
-lkr/auth/oauth.py,sha256=-QCb7YGIqmuTiy6SZcUTbVZrDo7itOpUUOK1VuvfMqM,5987
-lkr/embed/observability/utils.py,sha256=Hv3g60cI03cQwyEe8QV7bUMssE6pyrZKnDyl9rxm5b8,2915
-lkr/mcp/main.py,sha256=39VO0_bdscrCW7BC3kdlgjfIWaNigpT38HZ74_Drm-A,22853
-lkr/observability/classes.py,sha256=5kQnHxa3pgUartrbccuG0yMClfoFqYCXY8iCTJTZvgk,6025
-lkr/observability/embed_container.html,sha256=BMzV_jfBhcussUExf0KccOmOTAO3UEV7SuZ4lznQ0no,2758
-lkr/observability/main.py,sha256=6xAcaW4ruxGZDiGHNTi0oIaQnI6HhKxsjCZ5jq_ldsE,8557
-lkr_dev_cli-0.0.22.dist-info/METADATA,sha256=Yp81tfskRQC4vKRCecxXGmmBK7eiT5Djf_fBzj9VFyo,10663
-lkr_dev_cli-0.0.22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lkr_dev_cli-0.0.22.dist-info/entry_points.txt,sha256=nn2sFMGDpwUVE61ZUpbDPnQZkW7Gc08nV-tyLGo8q34,37
-lkr_dev_cli-0.0.22.dist-info/licenses/LICENSE,sha256=hKnCOORW1JRE_M2vStz8dblS5u1iR-2VpqS9xagKNa0,1063
-lkr_dev_cli-0.0.22.dist-info/RECORD,,