lkr-dev-cli 0.0.30__py3-none-any.whl → 0.0.31__py3-none-any.whl

lkr/tools/classes.py

@@ -1,6 +1,5 @@
 from typing import Literal, Optional, Self, cast
 
-from fastapi import Request
 from looker_sdk.sdk.api40.methods import Looker40SDK
 from looker_sdk.sdk.api40.models import (
     UserAttributeGroupValue,
@@ -46,10 +45,12 @@ class UserAttributeUpdater(BaseModel):
                 )
         return self
 
-    def get_request_authorization_for_value(self, request: Request):
-        authorization_token = request.headers.get("Authorization")
+    def get_request_authorization_for_value(self, headers: list[tuple[str, str]]):
+        authorization_token = next(
+            (header for header in headers if header[0] == "Authorization"), None
+        )
         if authorization_token:
-            self.value = authorization_token
+            self.value = authorization_token[1]
         else:
             logger.error("No authorization token found")
 

lkr/tools/main.py

@@ -22,7 +22,7 @@ def user_attribute_updater(
     @api.post("/identity_token")
     def identity_token(request: Request, body: UserAttributeUpdater):
         try:
-            body.get_request_authorization_for_value(request)
+            body.get_request_authorization_for_value(request.headers.items())
             body.update_user_attribute_value()
             raw_urls = os.getenv("LOOKER_WHITELISTED_BASE_URLS", "")
             whitelisted_base_urls = (

{lkr_dev_cli-0.0.30.dist-info → lkr_dev_cli-0.0.31.dist-info}/METADATA

@@ -1,23 +1,32 @@
 Metadata-Version: 2.4
 Name: lkr-dev-cli
-Version: 0.0.30
+Version: 0.0.31
 Summary: lkr: a command line interface for looker
 Author: bwebs
 License-Expression: MIT
 License-File: LICENSE
 Requires-Python: >=3.12
 Requires-Dist: cryptography>=42.0.0
-Requires-Dist: duckdb>=1.2.2
-Requires-Dist: fastapi>=0.115.12
 Requires-Dist: looker-sdk>=25.4.0
-Requires-Dist: mcp[cli]>=1.9.2
 Requires-Dist: pydantic>=2.11.4
 Requires-Dist: pydash>=8.0.5
 Requires-Dist: questionary>=2.1.0
 Requires-Dist: requests>=2.31.0
-Requires-Dist: selenium>=4.32.0
 Requires-Dist: structlog>=25.3.0
 Requires-Dist: typer>=0.15.2
+Provides-Extra: all
+Requires-Dist: duckdb>=1.2.2; extra == 'all'
+Requires-Dist: fastapi>=0.115.12; extra == 'all'
+Requires-Dist: mcp[cli]>=1.9.2; extra == 'all'
+Requires-Dist: selenium>=4.32.0; extra == 'all'
+Provides-Extra: embed-observability
+Requires-Dist: fastapi>=0.115.12; extra == 'embed-observability'
+Requires-Dist: selenium>=4.32.0; extra == 'embed-observability'
+Provides-Extra: mcp
+Requires-Dist: duckdb>=1.2.2; extra == 'mcp'
+Requires-Dist: mcp[cli]>=1.9.2; extra == 'mcp'
+Provides-Extra: user-attribute-updater
+Requires-Dist: fastapi>=0.115.12; extra == 'user-attribute-updater'
 Description-Content-Type: text/markdown
 
 # lkr cli
@@ -214,10 +223,13 @@ This can also be used to stress test your Looker environment as it serves an API
 ## User Attribute Updater (OIDC Token)
 
 1. Create a new cloud run using the `lkr-cli` public docker image `us-central1-docker.pkg.dev/lkr-dev-production/lkr-cli/cli:latest`
-2. Put in the environment variables LOOKERSDK_CLIENT_ID, LOOKERSDK_CLIENT_SECRET, LOOKERSDK_BASE_URL, LOOKER_WHITELISTED_BASE_URLS. The `LOOKER_WHITELISTED_BASE_URLS` would be the same url as the `LOOKERSDK_BASE_URL` if you are only using this for a single Looker instance. For more advanced use cases, you can set the `LOOKER_WHITELISTED_BASE_URLS` to a comma separated list of urls. The body of the request also accepts a `base_url`, `client_id`, and `client_secret` key that will override these settings.
-3. Deploy the cloud run
-4. Retrieve the URL of the cloud run
-5. Create the user attribute 
+2. Put in the environment variables LOOKERSDK_CLIENT_ID, LOOKERSDK_CLIENT_SECRET, LOOKERSDK_BASE_URL, LOOKER_WHITELISTED_BASE_URLS. The `LOOKER_WHITELISTED_BASE_URLS` would be the same url as the `LOOKERSDK_BASE_URL` if you are only using this for a single Looker instance. For more advanced use cases, you can set the `LOOKER_WHITELISTED_BASE_URLS` to a comma separated list of urls. The body of the request also accepts a `base_url`, `client_id`, and `client_secret` key that will override these settings. See example [`gcloud` command](#example-gcloud-command)
+3. For the command and arguments use:
+   - command: `lkr`
+   - args: `tools` `user-attribute-updater`
+4. Deploy the cloud run
+5. Retrieve the URL of the cloud run
+6. Create the user attribute 
    - Name: cloud_run_access_token 
    - Data Type: String
    - User Access: None
@@ -250,6 +262,23 @@ This can also be used to stress test your Looker environment as it serves an API
 9. Check the logs of the cloud run to see if there was a 200 response
 
 
+### Example `gcloud` command
+```bash
+export REGION=<your region>
+export PROJECT=<your project id>
+
+gcloud run deploy lkr-access-token-updater \
+  --image us-central1-docker.pkg.dev/lkr-dev-production/lkr-cli/cli:latest \
+  --command lkr \
+  --args tools,user-attribute-updater \
+  --platform managed \
+  --region $REGION \
+  --project $PROJECT \
+  --cpu 1 \
+  --memory 2Gi \
+  --set-env-vars LOOKERSDK_CLIENT_ID=<your client id>,LOOKERSDK_CLIENT_SECRET=<your client secret>,LOOKERSDK_BASE_URL=<your instance url>,LOOKER_WHITELISTED_BASE_URLS=<your instance url>
+  ```
+
 ## UserAttributeUpdater `lkr-dev-cli`
 
 Exported from the `lkr-dev-cli` package is the `UserAttributeUpdater` pydantic class. This class has all the necessary logic to update a user attribute value. 
@@ -301,12 +330,12 @@ from lkr import UserAttributeUpdater
 def request_authorization(request: Request):
     body = await request.json()
     updater = UserAttributeUpdater.model_validate(body)
-    updater.get_request_authorization_for_value(request)
+    updater.get_request_authorization_for_value(request.headers.items())
     updater.update_user_attribute_value()
 
 @app.post("/as_body")
 def as_body(request: Request, body: UserAttributeUpdater):
-    body.get_request_authorization_for_value(request)
+    body.get_request_authorization_for_value(request.headers.items())
     body.update_user_attribute_value()
 
 @app.post("/assigning_value")
@@ -317,4 +346,64 @@ def assigning_value(request: Request):
     )
     updater.value = request.headers.get("my_custom_header")
     updater.update_user_attribute_value()
+
+@app.delete("/:user_attribute_name/:email")
+def delete_user_attribute(user_attribute_name: str, email: str):
+    updater = UserAttributeUpdater(
+      user_attribute=user_attribute_name,
+      update_type="user",
+      email=email,
+    )
+    updater.delete_user_attribute_value()
+
+## Optional Dependencies
+
+The `lkr` CLI supports optional dependencies that enable additional functionality. You can install these individually or all at once.
+
+### Available Extras
+
+- **`mcp`**: Enables the MCP (Model Context Protocol) server functionality
+  - Includes: `mcp[cli]>=1.9.2`, `duckdb>=1.2.2`
+- **`embed-observability`**: Enables the observability embed monitoring features
+  - Includes: `fastapi>=0.115.12`, `selenium>=4.32.0`
+- **`user-attribute-updater`**: Enables the user attribute updater functionality
+  - Includes: `fastapi>=0.115.12`
+
+### Installing Optional Dependencies
+
+**Install all optional dependencies:**
+```bash
+uv sync --extra all
+```
+
+**Install specific extras:**
+```bash
+# Install MCP functionality
+uv sync --extra mcp
+
+# Install observability features
+uv sync --extra embed-observability
+
+# Install user attribute updater
+uv sync --extra user-attribute-updater
+
+# Install multiple extras
+uv sync --extra mcp --extra embed-observability
+```
+
+**Using pip:**
+```bash
+# Install all optional dependencies
+pip install lkr-dev-cli[all]
+
+# Install specific extras
+pip install lkr-dev-cli[mcp,embed-observability,user-attribute-updater]
 ```
+
+### What Each Extra Enables
+
+- **`mcp`**: Use the MCP server with tools like Cursor for enhanced IDE integration
+- **`embed-observability`**: Run the observability embed server for monitoring Looker dashboard performance
+- **`user-attribute-updater`**: Deploy the user attribute updater service for OIDC token management
+
+All extras are designed to work together seamlessly, and installing `all` is equivalent to installing all individual extras.

{lkr_dev_cli-0.0.30.dist-info → lkr_dev_cli-0.0.31.dist-info}/RECORD

@@ -16,10 +16,10 @@ lkr/observability/classes.py,sha256=LgGuUnY-J1csPrlAKnw4PPOqOfbvaOx2cxENlQgJYcE,
 lkr/observability/embed_container.html,sha256=IcDG-QVsYYNGQGrkDrx9OMZ2Pmo4C8oAjRHddFQ7Tlw,2939
 lkr/observability/main.py,sha256=XbejIdqhNNUMqHVezb5EnLaJ32dO9-Bt0o5d8lc0kyw,9544
 lkr/observability/utils.py,sha256=UpaBrp_ufaXLoz4p3xG3K6lHKBpP9wBhvP8rDmeGoWg,2148
-lkr/tools/classes.py,sha256=tKwHSr5X8SRL-fnBolqV5xe5A4_B8bEeh9UU7glq4wo,7459
-lkr/tools/main.py,sha256=W9Q0Qq8rZNmhpAdOvJLm1iWeHBPAIBsIAxod6noskU8,2667
-lkr_dev_cli-0.0.30.dist-info/METADATA,sha256=3StSuRBhzSNF-9nlKhsoSvvkXkGmX5Ynl1AFv0Or-Ss,15407
-lkr_dev_cli-0.0.30.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lkr_dev_cli-0.0.30.dist-info/entry_points.txt,sha256=nn2sFMGDpwUVE61ZUpbDPnQZkW7Gc08nV-tyLGo8q34,37
-lkr_dev_cli-0.0.30.dist-info/licenses/LICENSE,sha256=hKnCOORW1JRE_M2vStz8dblS5u1iR-2VpqS9xagKNa0,1063
-lkr_dev_cli-0.0.30.dist-info/RECORD,,
+lkr/tools/classes.py,sha256=ZyRRCQjjwV4WVWGmKlTfXiLiOGUf67XgrboYhOLuLts,7508
+lkr/tools/main.py,sha256=u9O5JgGVFscav9wFcrHd5ZPUUue_KpeK0QukYKqzros,2683
+lkr_dev_cli-0.0.31.dist-info/METADATA,sha256=8QimDm78NO_wvLUt3XJ_SlwdMkCY3pk5Jt8KY1FuGxo,18527
+lkr_dev_cli-0.0.31.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lkr_dev_cli-0.0.31.dist-info/entry_points.txt,sha256=nn2sFMGDpwUVE61ZUpbDPnQZkW7Gc08nV-tyLGo8q34,37
+lkr_dev_cli-0.0.31.dist-info/licenses/LICENSE,sha256=hKnCOORW1JRE_M2vStz8dblS5u1iR-2VpqS9xagKNa0,1063
+lkr_dev_cli-0.0.31.dist-info/RECORD,,