moose-lib 0.6.176__py3-none-any.whl → 0.6.188__py3-none-any.whl

moose_lib/secrets.py

@@ -20,11 +20,13 @@ MOOSE_RUNTIME_ENV_PREFIX = "__MOOSE_RUNTIME_ENV__:"
 
 
 def get(env_var_name: str) -> str:
-    """Marks a value to be resolved from an environment variable at runtime.
+    """Gets a value from an environment variable, with behavior depending on context.
 
-    When you use this function, the value is not read immediately. Instead,
-    a special marker is created that the Moose CLI will resolve when it
-    processes your infrastructure configuration.
+    When IS_LOADING_INFRA_MAP=true (infrastructure loading):
+      Returns a marker string that Moose CLI will resolve later
+
+    When IS_LOADING_INFRA_MAP is unset (function/workflow runtime):
+      Returns the actual value from the environment variable
 
     This is useful for:
     - Credentials that should never be embedded in Docker images
@@ -36,10 +38,11 @@ def get(env_var_name: str) -> str:
         env_var_name: Name of the environment variable to resolve
 
     Returns:
-        A marker string that Moose CLI will resolve at runtime
+        Either a marker string or the actual environment variable value
 
     Raises:
         ValueError: If the environment variable name is empty
+        KeyError: If the environment variable is not set (runtime mode only)
 
     Example:
         >>> # Instead of this (evaluated at build time):
@@ -49,9 +52,26 @@ def get(env_var_name: str) -> str:
         >>> # Use this (evaluated at runtime):
         >>> aws_key = moose_runtime_env.get("AWS_ACCESS_KEY_ID")
     """
+    import os
+    
     if not env_var_name or not env_var_name.strip():
         raise ValueError("Environment variable name cannot be empty")
-    return f"{MOOSE_RUNTIME_ENV_PREFIX}{env_var_name}"
+    
+    # Check if we're loading infrastructure map
+    is_loading_infra_map = os.environ.get("IS_LOADING_INFRA_MAP") == "true"
+    
+    if is_loading_infra_map:
+        # Return marker string for later resolution by Moose CLI
+        return f"{MOOSE_RUNTIME_ENV_PREFIX}{env_var_name}"
+    else:
+        # Return actual value from environment for runtime execution
+        value = os.environ.get(env_var_name)
+        if value is None:
+            raise KeyError(
+                f"Environment variable '{env_var_name}' is not set. "
+                f"This is required for runtime execution of functions/workflows."
+            )
+        return value
 
 
 class MooseRuntimeEnv:

{moose_lib-0.6.176.dist-info → moose_lib-0.6.188.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: moose_lib
-Version: 0.6.176
+Version: 0.6.188
 Home-page: https://www.fiveonefour.com/moose
 Author: Fiveonefour Labs Inc.
 Author-email: support@fiveonefour.com

{moose_lib-0.6.176.dist-info → moose_lib-0.6.188.dist-info}/RECORD

@@ -7,7 +7,7 @@ moose_lib/internal.py,sha256=7T6PQyU2QLquSSPujO37e9t0tC_QYXpIK2CQom69li4,30453
 moose_lib/main.py,sha256=JWsgza52xEh25AyF61cO1ItJ8VXJHHz8j-4HG445Whg,20380
 moose_lib/query_builder.py,sha256=-L5p2dArBx3SBA-WZPkcCJPemKXnqJw60NHy-wn5wa4,6619
 moose_lib/query_param.py,sha256=kxcR09BMIsEg4o2qetjKrVu1YFRaLfMEzwzyGsKUpvA,6474
-moose_lib/secrets.py,sha256=upFjLorfAsiyL3F3UIl2STUFGjC6Pu0-Qe1FGs5S09c,3480
+moose_lib/secrets.py,sha256=Ykt_iJZrozMRXAirC_NclDerSJVE5WmGTSyyHrfsTFE,4249
 moose_lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 moose_lib/clients/redis_client.py,sha256=BDYjJ582V-rW92qVQ4neZ9Pu7JtDNt8x8jBWApt1XUg,11895
 moose_lib/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -42,10 +42,10 @@ tests/test_olap_table_versioning.py,sha256=ffuJsO5TYKBd2aWV0GWDmosCwL4j518Y_MqYJ
 tests/test_query_builder.py,sha256=O3imdFSaqU13kbK1jSQaHbBgynhVmJaApT8DlRqYwJU,1116
 tests/test_redis_client.py,sha256=d9_MLYsJ4ecVil_jPB2gW3Q5aWnavxmmjZg2uYI3LVo,3256
 tests/test_s3queue_config.py,sha256=A-17Lir2kvYoUA0qEaNczOkLKhHPzyp3glka7YqnPws,14490
-tests/test_secrets.py,sha256=g5ZjtOyT7vTz_sRUB71Vd6s715u0gMc4C-_8m_59Mtg,8347
+tests/test_secrets.py,sha256=WFvmVsxwYsl5bD0SiOUR9ZELMBZ3CgY0NR4h7s9Oaww,8707
 tests/test_simple_aggregate.py,sha256=yyFSYHUXskA1aTcwC-KQ9XmgOikeQ8wKXzntsUBIC6w,4055
 tests/test_web_app.py,sha256=iL86sg0NS2k6nP8jIGKZvrtg0BjvaqNTRp7-4T1TTck,6557
-moose_lib-0.6.176.dist-info/METADATA,sha256=Ox8432aYAvICgEuZcQC5LPeKrN-mVf3dXJztDUXXpNU,827
-moose_lib-0.6.176.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-moose_lib-0.6.176.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
-moose_lib-0.6.176.dist-info/RECORD,,
+moose_lib-0.6.188.dist-info/METADATA,sha256=pA-Ch_--rOTI4JeT55sonZEv1wMLE2sOQVVAMXGGym0,827
+moose_lib-0.6.188.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+moose_lib-0.6.188.dist-info/top_level.txt,sha256=XEns2-4aCmGp2XjJAeEH9TAUcGONLnSLy6ycT9FSJh8,16
+moose_lib-0.6.188.dist-info/RECORD,,

tests/test_secrets.py

@@ -5,10 +5,20 @@ which allows users to defer secret resolution until runtime rather than
 embedding secrets at build time.
 """
 
+import os
 import pytest
 from moose_lib.secrets import moose_runtime_env, get, MOOSE_RUNTIME_ENV_PREFIX
 
 
+@pytest.fixture(scope="module", autouse=True)
+def set_infra_map_loading_for_secrets_tests():
+    """Set IS_LOADING_INFRA_MAP=true for secrets tests so moose_runtime_env.get() returns markers."""
+    os.environ["IS_LOADING_INFRA_MAP"] = "true"
+    yield
+    # Clean up after all tests in this module
+    os.environ.pop("IS_LOADING_INFRA_MAP", None)
+
+
 class TestMooseRuntimeEnvGet:
     """Tests for the moose_runtime_env.get() method."""