lkr-dev-cli 0.0.47__tar.gz → 0.0.51__tar.gz

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/PKG-INFO

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

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/codemode.md

@@ -25,6 +25,9 @@ uvx lkr-dev-cli code-mode sandbox --code="return me()"
 
 # Execute Python script file
 uvx lkr-dev-cli code-mode sandbox --file=./path/to/script.py
+
+# Execute Python script file with injected variables (--var / -v)
+uvx lkr-dev-cli code-mode sandbox --file=./path/to/script.py --var project=my_project --var connection=my_connection
 ```
 
 ### 2. Starting the Server

{lkr_dev_cli-0.0.47/lkr_dev_cli → lkr_dev_cli-0.0.51/lkr}/__init__.py

@@ -1,5 +1,5 @@
-from lkr.tools.classes import UserAttributeUpdater
-from lkr.extended_sdk_methods import ExtendedLooker40SDK
 from lkr.auth_service import init_sdk
+from lkr.extended_sdk_methods import ExtendedLooker40SDK
+from lkr.tools.classes import UserAttributeUpdater
 
-__all__ = ["UserAttributeUpdater", "ExtendedLooker40SDK", "init_sdk"]
+__all__ = ["ExtendedLooker40SDK", "UserAttributeUpdater", "init_sdk"]

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/auth/main.py

@@ -1,5 +1,6 @@
+import builtins
 import urllib.parse
-from typing import Annotated, List, Union
+from typing import Annotated
 
 import typer
 from looker_sdk.rtl.auth_token import AccessToken, AuthToken
@@ -82,7 +83,7 @@ def login(
             logger.error(f"Instance '{instance_name}' not found")
             raise typer.Exit(1)
     else:
-        options: List[questionary.Choice] = []
+        options: builtins.list[questionary.Choice] = []
         max_name_length = 0
         for name, url, current, up in all_instances:
             max_name_length = max(max_name_length, len(name))
@@ -139,7 +140,7 @@ def login(
                 "Instance name must be set before adding auth."
             )
 
-            def auth_callback(token: Union[AuthToken, AccessToken]):
+            def auth_callback(token: AuthToken | AccessToken):
                 auth.add_auth(instance_name, origin, token, use_production)
 
             oauth = OAuth2PKCE(
@@ -166,9 +167,9 @@ def login(
                     else:
                         logger.error("Failed to exchange authorization code for tokens")
                         raise typer.Exit(1)
-                except Exception as e:
+                except Exception as e:  # noqa: BLE001
                     logger.error(
-                        f"Failed to exchange authorization code for tokens: {str(e)}"
+                        f"Failed to exchange authorization code for tokens: {e!s}"
                     )
                     raise typer.Exit(1)
             else:
@@ -249,7 +250,7 @@ def whoami(ctx: typer.Context):
         if "invalid_grant" in str(e) or "token expired" in str(e):
             logger.error("Your Looker OAuth session has expired. Please run 'lkr auth login' to re-authenticate.")
             raise typer.Exit(1)
-        raise e
+        raise
 
 
 @group.command()

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/auth/oauth.py

@@ -8,7 +8,7 @@ import threading
 import time
 import urllib.parse
 import webbrowser
-from typing import Optional, TypedDict, cast
+from typing import TypedDict, cast
 from urllib.parse import parse_qs
 
 from lkr.custom_types import NewTokenCallback
@@ -24,19 +24,21 @@ def kill_process_on_port(port: int, retries: int = 5, delay: float = 1) -> None:
         sock.bind(("localhost", port))
         sock.close()
         return  # Port is free, no need to kill anything
-    except socket.error:
+    except OSError:
         # Port is in use, try to kill the process
         if os.name == "posix":  # macOS/Linux
             subprocess.run(
                 f"lsof -ti tcp:{port} | xargs kill -9",
                 shell=True,
                 capture_output=True,
+                check=False,
             )
         elif os.name == "nt":  # Windows
             subprocess.run(
                 f'for /f "tokens=5" %a in (\'netstat -aon ^| find ":{port}"\') do taskkill /F /PID %a',
                 shell=True,
                 capture_output=True,
+                check=False,
             )
         # After killing, wait for the port to be free
         for _ in range(retries):
@@ -46,7 +48,7 @@ def kill_process_on_port(port: int, retries: int = 5, delay: float = 1) -> None:
                 sock.bind(("localhost", port))
                 sock.close()
                 return
-            except socket.error:
+            except OSError:
                 time.sleep(delay)
         raise RuntimeError(f"Port {port} is still in use after killing process.")
 
@@ -91,9 +93,7 @@ class OAuthCallbackHandler(http.server.BaseHTTPRequestHandler):
                 f"Looker OAuth returned error: {error} - {error_description}"
             )
             self.wfile.write(
-                f"OAuth Authentication Failed: {error} - {error_description}. You can close this window.".encode(
-                    "utf-8"
-                )
+                f"OAuth Authentication Failed: {error} - {error_description}. You can close this window.".encode()
             )
             threading.Thread(target=server.shutdown).start()
         else:
@@ -107,7 +107,6 @@ class OAuthCallbackHandler(http.server.BaseHTTPRequestHandler):
 
     def log_message(self, format, *args):
         """Suppress logging of requests"""
-        pass
 
 
 class OAuthCallbackServer(socketserver.TCPServer):
@@ -119,8 +118,8 @@ class OAuthCallbackServer(socketserver.TCPServer):
 
 
 class LoginResponse(TypedDict):
-    auth_code: Optional[str]
-    code_verifier: Optional[str]
+    auth_code: str | None
+    code_verifier: str | None
 
 
 class OAuth2PKCE:
@@ -132,7 +131,7 @@ class OAuth2PKCE:
     ):
         from lkr.auth_service import DbOAuthSession
 
-        self.auth_code: Optional[str] = None
+        self.auth_code: str | None = None
         self.state = secrets.token_urlsafe(16)
         self.new_token_callback: NewTokenCallback = new_token_callback
         self.auth_session: DbOAuthSession | None = None
@@ -181,9 +180,7 @@ class OAuth2PKCE:
                 else:
                     raise
         if not server_created:
-            import logging
-
-            logging.error(
+            logger.error(
                 f"Failed to bind to port {self.port} after waiting: {last_exception}"
             )
             raise RuntimeError(

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/auth_service.py

@@ -2,8 +2,8 @@ import json
 import os
 import sqlite3
 import types
-from datetime import datetime, timedelta, timezone
-from typing import List, Self, Tuple, Union
+from datetime import UTC, datetime, timedelta
+from typing import Self, Union
 
 import requests
 import typer
@@ -13,21 +13,21 @@ from looker_sdk.rtl.auth_session import AuthSession, CryptoHash, OAuthSession
 from looker_sdk.rtl.auth_token import AccessToken, AuthToken
 from looker_sdk.rtl.requests_transport import RequestsTransport
 from looker_sdk.rtl.transport import LOOKER_API_ID, HttpMethod
-from lkr.extended_sdk_methods import ExtendedLooker40SDK as Looker40SDK
 from pydantic import BaseModel, Field, computed_field
 from pydash import get
 
 from lkr.classes import LkrCtxObj, LookerApiKey
 from lkr.constants import LOOKER_API_VERSION, OAUTH_CLIENT_ID
 from lkr.custom_types import NewTokenCallback
+from lkr.extended_sdk_methods import ExtendedLooker40SDK as Looker40SDK
 from lkr.logger import logger
 
 __all__ = [
-    "get_auth",
     "ApiKeyAuthSession",
     "DbOAuthSession",
-    "is_auth_expired",
+    "get_auth",
     "init_sdk",
+    "is_auth_expired",
 ]
 
 
@@ -138,7 +138,7 @@ class DbOAuthSession(OAuthSession):
         if not self.use_production:
             try:
                 self._switch_to_dev_mode()
-            except Exception as e:
+            except Exception as e:  # noqa: BLE001
                 logger.error(f"Failed to switch to development mode: {e}")
                 raise typer.Exit(1)
         self.new_token_callback(self.token)
@@ -223,9 +223,7 @@ def monkey_patch_prepare_request(session: requests.Session):
             x.headers.pop(LOOKER_API_ID)
         return x
 
-    setattr(
-        session, "prepare_request", types.MethodType(prepare_request, session)
-    )
+    session.prepare_request = types.MethodType(prepare_request, session)  # type: ignore
 
 
 class MonkeyPatchTransport(RequestsTransport):
@@ -269,7 +267,7 @@ class CurrentAuth(BaseModel):
     refresh_token: str
     refresh_expires_at: str = Field(
         default_factory=lambda: (
-            datetime.now(timezone.utc) + timedelta(days=30)
+            datetime.now(UTC) + timedelta(days=30)
         ).isoformat()
     )
     token_type: str
@@ -283,23 +281,23 @@ class CurrentAuth(BaseModel):
         if not self.refresh_expires_at:
             return False
         return datetime.fromisoformat(self.refresh_expires_at).replace(
-            tzinfo=timezone.utc
-        ) > (datetime.now(timezone.utc) + timedelta(hours=24))
+            tzinfo=UTC
+        ) > (datetime.now(UTC) + timedelta(hours=24))
 
     @computed_field
     @property
     def expires_at(self) -> str:
         return (
-            datetime.now(timezone.utc) + timedelta(seconds=self.expires_in)
+            datetime.now(UTC) + timedelta(seconds=self.expires_in)
         ).isoformat()
 
-    def __add__(self, other: Union[AccessToken, AuthToken]) -> Self:
+    def __add__(self, other: AccessToken | AuthToken) -> Self:
         self.access_token = other.access_token or ""
         self.refresh_token = other.refresh_token or ""
         self.token_type = other.token_type or ""
         self.expires_in = other.expires_in or 0
         self.refresh_expires_at = (
-            datetime.now(timezone.utc) + timedelta(days=30)
+            datetime.now(UTC) + timedelta(days=30)
         ).isoformat()
         return self
 
@@ -307,7 +305,7 @@ class CurrentAuth(BaseModel):
     def from_access_token(
         cls,
         *,
-        access_token: Union[AccessToken, AuthToken],
+        access_token: AccessToken | AuthToken,
         instance_name: str,
         base_url: str,
         use_production: bool,
@@ -327,8 +325,8 @@ class CurrentAuth(BaseModel):
     def from_db_row(cls, row: sqlite3.Row) -> "CurrentAuth":
         expires_at_dt = datetime.fromisoformat(row["expires_at"])
         if expires_at_dt.tzinfo is None:
-            expires_at_dt = expires_at_dt.replace(tzinfo=timezone.utc)
-        now_utc = datetime.now(timezone.utc)
+            expires_at_dt = expires_at_dt.replace(tzinfo=UTC)
+        now_utc = datetime.now(UTC)
         expires_in = int((expires_at_dt - now_utc).total_seconds())
         return cls(
             instance_name=row["instance_name"],
@@ -364,22 +362,29 @@ class CurrentAuth(BaseModel):
         self,
         connection: sqlite3.Connection,
         *,
-        new_token: Union[AccessToken, AuthToken] | None = None,
+        new_token: AccessToken | AuthToken | None = None,
         commit: bool = True,
     ):
         expires_at = (
-            datetime.now(timezone.utc)
+            datetime.now(UTC)
             + timedelta(seconds=(new_token.expires_in or 0) if new_token else 0)
         ).isoformat()
         refresh_expires_at = (
             datetime.fromisoformat(expires_at) + timedelta(days=30)
         ).isoformat()
         if self.from_db and new_token:
+            if new_token.refresh_token:
+                self.refresh_token = new_token.refresh_token
+                self.refresh_expires_at = refresh_expires_at
+            self.access_token = new_token.access_token or self.access_token
+            self.token_type = new_token.token_type or self.token_type
             connection.execute(
-                "UPDATE auth SET access_token = ?, token_type = ?, expires_at = ? WHERE instance_name = ?",
+                "UPDATE auth SET access_token = ?, refresh_token = ?, refresh_expires_at = ?, token_type = ?, expires_at = ? WHERE instance_name = ?",
                 (
-                    new_token.access_token,
-                    new_token.token_type,
+                    self.access_token,
+                    self.refresh_token,
+                    self.refresh_expires_at,
+                    self.token_type,
                     expires_at,
                     self.instance_name,
                 ),
@@ -443,7 +448,7 @@ class SqlLiteAuth:
         self,
         instance_name: str,
         base_url: str,
-        access_token: Union[AuthToken, AccessToken],
+        access_token: AuthToken | AccessToken,
         use_production: bool,
     ):
         self.conn.execute("UPDATE auth SET current_instance = 0")
@@ -498,7 +503,7 @@ class SqlLiteAuth:
                 else:
                     raise InvalidRefreshTokenError(current_auth.instance_name)
 
-            def refresh_current_token(token: Union[AccessToken, AuthToken]):
+            def refresh_current_token(token: AccessToken | AuthToken):
                 current_auth.set_token(self.conn, new_token=token, commit=True)
 
             sdk = init_oauth_sdk(
@@ -513,13 +518,12 @@ class SqlLiteAuth:
                 try:
                     sdk.auth.authenticate({})
                 except Exception as e:
-                    if is_auth_expired(e):
-                        if sys.stdin.isatty():
-                            self._cli_confirm_refresh_token(current_auth, quiet=False)
-                            return self.get_current_sdk(
-                                prompt_refresh_invalid_token=False
-                            )
-                    raise e
+                    if is_auth_expired(e) and sys.stdin.isatty():
+                        self._cli_confirm_refresh_token(current_auth, quiet=False)
+                        return self.get_current_sdk(
+                            prompt_refresh_invalid_token=False
+                        )
+                    raise
 
             return sdk
 
@@ -531,7 +535,7 @@ class SqlLiteAuth:
         self.conn.execute("DELETE FROM auth WHERE instance_name = ?", (instance_name,))
         self.conn.commit()
 
-    def list_auth(self) -> List[Tuple[str, str, bool, bool]]:
+    def list_auth(self) -> list[tuple[str, str, bool, bool]]:
         cursor = self.conn.execute(
             "SELECT instance_name, base_url, current_instance, use_production FROM auth ORDER BY instance_name ASC"
         )
@@ -562,7 +566,7 @@ class SqlLiteAuth:
         )
         if confirmed:
 
-            def add_auth(token: Union[AccessToken, AuthToken]):
+            def add_auth(token: AccessToken | AuthToken):
                 current_auth + token
                 current_auth.update_refresh_expires_at(self.conn, commit=False)
                 current_auth.set_token(self.conn, commit=True, new_token=token)
@@ -600,7 +604,7 @@ class ApiKeyAuth:
         self,
         instance_name: str,
         base_url: str,
-        access_token: Union[AuthToken, AccessToken],
+        access_token: AuthToken | AccessToken,
         use_production: bool,
     ):
         raise NotImplementedError("ApiKeyAuth does not support adding auth")
@@ -608,7 +612,7 @@ class ApiKeyAuth:
     def delete_auth(self, instance_name: str):
         raise NotImplementedError("ApiKeyAuth does not support deleting auth")
 
-    def list_auth(self) -> List[Tuple[str, str, bool, bool]]:
+    def list_auth(self) -> list[tuple[str, str, bool, bool]]:
         raise NotImplementedError("ApiKeyAuth does not support listing auth")
 
     def set_current_instance(self, instance_name: str):

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/classes.py

@@ -21,7 +21,7 @@ class LookerApiKey(BaseModel):
                 base_url=os.environ.get("LOOKERSDK_BASE_URL"),  # type: ignore
                 verify_ssl=verify_ssl,
             )
-        except Exception:
+        except Exception:  # noqa: BLE001
             return None
 
 
@@ -38,6 +38,6 @@ class LkrCtxObj(BaseModel):
         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)
+        super().__init__(*args, api_key=api_key, **kwargs)
         if not self.api_key:
             self.api_key = LookerApiKey.from_env()

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/codemode/download_swagger.py

@@ -1,7 +1,8 @@
-import os
-import requests
 import json
+import os
 import sys
+
+import requests
 from dotenv import load_dotenv
 
 

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/codemode/help.py

@@ -1,7 +1,8 @@
-import re
 import inspect
 import json
 import os
+import re
+
 from lkr.codemode.constant import EXCLUDED_FUNCS
 
 _operation_map = None
@@ -22,8 +23,8 @@ def _get_operation_map():
         with open(swagger_path, 'r') as f:
             swagger = json.load(f)
             paths = swagger.get('paths', {})
-            for path, path_item in paths.items():
-                for method, op in path_item.items():
+            for path_item in paths.values():
+                for op in path_item.values():
                     if isinstance(op, dict) and 'operationId' in op:
                         op_id = op['operationId']
                         summary = op.get('summary', '')
@@ -95,7 +96,7 @@ def _get_matches(query: str, external_funcs: dict, sdk) -> list:
                 params = list(sig.parameters.keys())
                 if any(pattern.search(p) for p in params):
                     hit_in_params = True
-            except Exception:
+            except Exception:  # noqa: BLE001, S110
                 pass
                 
         # Search in return type fields (Outputs)
@@ -111,7 +112,7 @@ def _get_matches(query: str, external_funcs: dict, sdk) -> list:
                     if matching_fields:
                         hit_in_output = True
                         output_fields = matching_fields
-            except Exception:
+            except Exception:  # noqa: BLE001, S110
                 pass
                 
         if hit_in_name or matching_lines or hit_in_params or hit_in_output:
@@ -133,11 +134,11 @@ def _get_type_matches(query: str) -> list:
         pattern = re.compile(re.escape(query), re.IGNORECASE)
 
     try:
-        from lkr.codemode.type import _get_swagger_data, _get_ext_definitions
+        from lkr.codemode.type import _get_ext_definitions, _get_swagger_data
         swagger = _get_swagger_data()
         definitions = dict(swagger.get('definitions', {}))
         definitions.update(_get_ext_definitions())
-    except Exception:
+    except Exception:  # noqa: BLE001
         return []
 
     matches = []
@@ -217,7 +218,7 @@ def search_with_lookups(query: str, external_funcs: dict, sdk) -> list:
         try:
             from lkr.codemode.type import lookup_type
             results.append(lookup_type(m['name']))
-        except Exception:
+        except Exception:  # noqa: BLE001, S110
             pass
     return results
 
@@ -230,7 +231,7 @@ def lookup_function(name: str, external_funcs: dict, sdk) -> str:
             res = lookup_type(name)
             if not res.startswith("Type '"):
                 return res
-        except Exception:
+        except Exception:  # noqa: BLE001, S110
             pass
         return f"Function or type '{name}' not found."
         
@@ -279,5 +280,5 @@ Inputs:
 Outputs:
 {outputs_str}
 """
-    except Exception as e:
+    except Exception as e:  # noqa: BLE001
         return f"Function: {name}\n\nDocstring:\n{doc}\n\n(Could not inspect signature: {e})"

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/codemode/main.py

@@ -1,22 +1,23 @@
 import ast
 import inspect
 import json
+import keyword
 import os
 import re
 import sys
 import tempfile
 from contextlib import contextmanager
 
-import typer
 import pydantic_monty
+import typer
 from mcp.server.fastmcp import FastMCP
 
 from lkr.auth_service import get_auth, is_auth_expired
 from lkr.classes import LkrCtxObj
+from lkr.codemode.constant import EXCLUDED_FUNCS
 from lkr.codemode.examples import EXAMPLES
-from lkr.codemode.help import search_help, lookup_function, search_with_lookups
+from lkr.codemode.help import lookup_function, search_help, search_with_lookups
 from lkr.codemode.readme import get_readme
-from lkr.codemode.constant import EXCLUDED_FUNCS
 from lkr.codemode.type import lookup_type
 from lkr.logger import logger
 
@@ -87,7 +88,7 @@ def to_primitive(obj):
                     return _to_primitive(vars(o))
                 except TypeError:
                     return str(o)
-                except Exception:
+                except Exception:  # noqa: BLE001
                     return str(o)
         finally:
             seen.remove(obj_id)
@@ -113,7 +114,6 @@ def run_python_code(code: str, dev_mode: bool = False) -> str:
     - Dev Mode: Set `dev_mode=True` to ensure you are in development mode before running code.
     """
     try:
-        global ctx_lkr
         ctx = (
             LkrCtxObj(
                 force_oauth=ctx_lkr.force_oauth,
@@ -172,7 +172,7 @@ def run_python_code(code: str, dev_mode: bool = False) -> str:
             tree = SDKAttributeRewriter().visit(tree)
             ast.fix_missing_locations(tree)
             code = ast.unparse(tree)
-        except Exception:
+        except Exception:  # noqa: BLE001
             # Fallback to regex if parsing fails
             code = re.sub(r"\bsdk\.([a-zA-Z_][a-zA-Z0-9_]*)\b", r"\1", code)
 
@@ -195,7 +195,7 @@ def run_python_code(code: str, dev_mode: bool = False) -> str:
                     output = json.dumps(result, indent=2, default=str)
             else:
                 output = ""
-        except Exception:
+        except Exception:  # noqa: BLE001
             output = repr(result)
             
         if printed_output:
@@ -204,11 +204,11 @@ def run_python_code(code: str, dev_mode: bool = False) -> str:
                 "result": result
             }, indent=2, default=str)
         return output
-    except Exception as e:
+    except Exception as e:  # noqa: BLE001
         logger.error(f"Error executing Monty: {e}")
         if is_auth_expired(e):
             return "Error: Your Looker OAuth session has expired. Please run 'lkr auth login' to re-authenticate."
-        return f"Error: {str(e)}"
+        return f"Error: {e!s}"
 
 
 @group.command(name="sandbox")
@@ -223,6 +223,9 @@ def sandbox(
     dev_mode: bool = typer.Option(
         False, "--dev-mode", help="Run in dev mode"
     ),
+    var: list[str] | None = typer.Option(  # noqa: B008
+        None, "--var", "-v", help="Inject variable as key=value pair (e.g. -v project=my_project)"
+    ),
 ):
     if not code and not file:
         logger.error("Must specify either --code or --file")
@@ -236,11 +239,27 @@ def sandbox(
         try:
             with open(file, "r", encoding="utf-8") as f:
                 code_to_run = f.read()
-        except Exception as e:
+        except Exception as e:  # noqa: BLE001
             logger.error(f"Failed to read file {file}: {e}")
             raise typer.Exit(1)
-    else:
+    elif code:
         code_to_run = code
+    else:
+        raise typer.Exit(1)
+
+    if var:
+        var_defs = []
+        for v_str in var:
+            if "=" not in v_str:
+                logger.error(f"Invalid variable format: '{v_str}'. Must be key=value.")
+                raise typer.Exit(1)
+            k, v = v_str.split("=", 1)
+            k = k.strip()
+            if not k.isidentifier() or keyword.iskeyword(k):
+                logger.error(f"Invalid variable name: '{k}'. Must be a valid Python identifier and not a keyword.")
+                raise typer.Exit(1)
+            var_defs.append(f"{k} = {json.dumps(v)}")
+        code_to_run = "\n".join(var_defs) + "\n" + code_to_run
 
     global ctx_lkr
     ctx_lkr = (

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/codemode/readme.py

@@ -13,6 +13,7 @@ RULES & HINTS:
 6. **Efficiency**: Always use the `fields` parameter (e.g., `all_dashboards(fields="id,title")`) when listing many objects to prevent timeouts.
 7. **Nested Folders**: Use `folder_children(id)` to get sub-folders.
 8. **Search with Lookups**: If `help('pattern')` returns too many results and you want to see full details for all of them at once, use `search_with_lookups('pattern')`.
+9. **Injected CLI Variables**: When using `sandbox` with `--var key=value` flags, those variables are directly available in your code as top-level string constants.
 
 EXAMPLES:
 

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/codemode/type.py

@@ -1,7 +1,9 @@
 import inspect
 import json
 import os
+
 from pydantic import BaseModel
+
 import lkr.extended_sdk_methods.classes as ext_classes
 
 _swagger_data = None
@@ -23,7 +25,7 @@ def _get_swagger_data():
     try:
         with open(swagger_path, 'r') as f:
             _swagger_data = json.load(f)
-    except Exception:
+    except Exception:  # noqa: BLE001
         _swagger_data = {}
         
     return _swagger_data
@@ -63,7 +65,7 @@ def _get_ext_definitions() -> dict:
                         prop_type = t.__name__
                     elif t is bool:
                         prop_type = "boolean"
-                    elif hasattr(t, '__origin__') and getattr(t, '__origin__') in (list, set):
+                    elif hasattr(t, '__origin__') and t.__origin__ in (list, set):
                         is_array = True
                         sub_args = getattr(t, '__args__', ())
                         for sub_t in sub_args:

{lkr_dev_cli-0.0.47 → lkr_dev_cli-0.0.51}/lkr/custom_types.py

@@ -1,9 +1,9 @@
+from collections.abc import Callable
 from enum import Enum
-from typing import Callable, Union
 
 from looker_sdk.rtl.auth_token import AccessToken, AuthToken
 
-NewTokenCallback = Callable[[Union[AuthToken, AccessToken]], None]
+NewTokenCallback = Callable[[AuthToken | AccessToken], None]
 
 class LogLevel(str, Enum):
     DEBUG = "DEBUG"