lkr-dev-cli 0.0.26__py3-none-any.whl → 0.0.27__py3-none-any.whl

lkr/tools/classes.py

@@ -7,7 +7,7 @@ from looker_sdk.sdk.api40.models import (
     WriteUserAttribute,
     WriteUserAttributeWithValue,
 )
-from pydantic import BaseModel, model_validator
+from pydantic import BaseModel, Field, model_validator
 
 from lkr.auth_service import init_api_key_sdk
 from lkr.classes import LookerApiKey
@@ -15,10 +15,10 @@ from lkr.logger import logger
 
 
 class UserAttributeUpdater(BaseModel):
-    client_id: Optional[str] = None
-    client_secret: Optional[str] = None
-    base_url: Optional[str] = None
-    value: Optional[str] = None
+    client_id: Optional[str] = Field(default=None, exclude=True)
+    client_secret: Optional[str] = Field(default=None, exclude=True)
+    base_url: Optional[str] = Field(default=None, exclude=False)
+    value: Optional[str] = Field(default=None, exclude=True)
     user_attribute: Optional[str]
     user_attribute_id: Optional[str] = None
     update_type: Literal["group", "default", "user"]

lkr/tools/main.py

@@ -1,7 +1,10 @@
+import os
+
 import typer
 import uvicorn
 from fastapi import FastAPI, Request
 
+from lkr.logger import structured_logger as logger
 from lkr.tools.classes import AttributeUpdaterResponse, UserAttributeUpdater
 
 __all__ = ["group"]
@@ -22,6 +25,21 @@ def user_attribute_updater(
         try:
             body.get_request_authorization_for_value(request)
             body.update_user_attribute_value()
+            raw_urls = os.getenv("LOOKER_WHITELISTED_BASE_URLS", "")
+            whitelisted_base_urls = (
+                [url.strip() for url in raw_urls.split(",") if url.strip()]
+                if raw_urls
+                else []
+            )
+            logger.debug(
+                "identify_token_user_attribute",
+                **body.model_dump(),
+                whitelisted_base_urls=whitelisted_base_urls,
+            )
+
+            if body.base_url not in whitelisted_base_urls:
+                raise Exception(f"Base URL {body.base_url} not whitelisted")
+
             return AttributeUpdaterResponse(
                 success=True, message="User attribute updated"
             )
@@ -32,6 +50,10 @@ def user_attribute_updater(
     def delete_user_attribute_value(request: Request, body: UserAttributeUpdater):
         try:
             body.delete_user_attribute_value()
+            logger.debug(
+                "user_attribute_delete",
+                **body.model_dump(),
+            )
             return AttributeUpdaterResponse(
                 success=True, message="User attribute value deleted"
             )
@@ -42,6 +64,10 @@ def user_attribute_updater(
     def update_user_attribute_value(request: Request, body: UserAttributeUpdater):
         try:
             body.update_user_attribute_value()
+            logger.debug(
+                "user_attribute_update",
+                **body.model_dump(),
+            )
             return AttributeUpdaterResponse(
                 success=True, message="User attribute value updated"
             )

{lkr_dev_cli-0.0.26.dist-info → lkr_dev_cli-0.0.27.dist-info}/METADATA

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

{lkr_dev_cli-0.0.26.dist-info → lkr_dev_cli-0.0.27.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=tXOFLdahjEHlpr0dgR3miwDeb98Qw1X7bUA-26Hzhkg,7635
-lkr/tools/main.py,sha256=zHBZWjCp_pYw_dPKQg_rLZTHLbnKYAoPPFBuHSqtEc0,1751
-lkr_dev_cli-0.0.26.dist-info/METADATA,sha256=fTj56Vl0lcKtV9gruaG0haOjqJYmvXzBeCsN2JpRLaY,10953
-lkr_dev_cli-0.0.26.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lkr_dev_cli-0.0.26.dist-info/entry_points.txt,sha256=nn2sFMGDpwUVE61ZUpbDPnQZkW7Gc08nV-tyLGo8q34,37
-lkr_dev_cli-0.0.26.dist-info/licenses/LICENSE,sha256=hKnCOORW1JRE_M2vStz8dblS5u1iR-2VpqS9xagKNa0,1063
-lkr_dev_cli-0.0.26.dist-info/RECORD,,
+lkr/tools/classes.py,sha256=7QJwm0Ywm8QIqkvgCWnpRI12C-DxLUznXj_BNF_gifI,7759
+lkr/tools/main.py,sha256=nJyCtRtYAw2sL0AqQBwk7VOhlqb9veVMypkA4oTJmXI,2623
+lkr_dev_cli-0.0.27.dist-info/METADATA,sha256=Xt8gXEvd0ufytFL5FubA700iS9-RTaKW9OcN7IxweAc,10953
+lkr_dev_cli-0.0.27.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lkr_dev_cli-0.0.27.dist-info/entry_points.txt,sha256=nn2sFMGDpwUVE61ZUpbDPnQZkW7Gc08nV-tyLGo8q34,37
+lkr_dev_cli-0.0.27.dist-info/licenses/LICENSE,sha256=hKnCOORW1JRE_M2vStz8dblS5u1iR-2VpqS9xagKNa0,1063
+lkr_dev_cli-0.0.27.dist-info/RECORD,,