tinyfan 0.1.0__tar.gz → 0.1.2__tar.gz

{tinyfan-0.1.0 → tinyfan-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinyfan
-Version: 0.1.0
+Version: 0.1.2
 Summary: Tiny Argo Workflows python devkit
 Project-URL: Documentation, https://github.com/eunchuldev/tinyfan#readme
 Project-URL: Issues, https://github.com/eunchuldev/tinyfan/issues

{tinyfan-0.1.0 → tinyfan-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tinyfan"
-version = "0.1.0"
+version = "0.1.2"
 description = 'Tiny Argo Workflows python devkit'
 readme = "README.md"
 requires-python = ">=3.12"

{tinyfan-0.1.0 → tinyfan-0.1.2}/src/tinyfan/codegen.py

@@ -15,6 +15,7 @@ from .argo_typing import ScriptTemplate
 from .utils.embed import embed
 from .utils.merge import deepmerge
 from .config import CLI_CONFIG_STATE, Config, ConfigValue, ConfigMapKeyRef, ConfigMapRef, SecretKeyRef, SecretRef
+from .flowrundata import FlowRunData
 
 VALID_CRON_REGEXP = r"^(?:(?:(?:(?:\d+,)+\d+|(?:\d+(?:\/|-|#)\d+)|\d+L?|\*(?:\/\d+)?|L(?:-\d+)?|\?|[A-Z]{3}(?:-[A-Z]{3})?) ?){5,7})|(@hourly|@daily|@midnight|@weekly|@monthly|@yearly|@annually)$"
 VALID_DEPENDS_REGEXP = r"^(?!.*\.[A-Za-z0-9_]+\.[A-Za-z0-9_]+)(?!.*\.(?!Succeeded|Failed|Errored|Skipped|Omitted|Daemoned)\w+)(?!.*(?:&&|\|\|)\s*(?:&&|\|\|))(?!.*!!)[A-Za-z0-9_&|!().\s]+$"
@@ -136,6 +137,7 @@ class AssetTree:
                 set(self.flow.assets.keys())
                 | set((self.flow.resources or {}).keys())
                 | set((self.flow.configs or {}).keys())
+                | set(inspect.get_annotations(FlowRunData).keys())
             )
             unknown_names = [n for n in func_param_names if n not in known_names]
             if len(unknown_names) > 0:

{tinyfan-0.1.0 → tinyfan-0.1.2}/src/tinyfan/config.py

@@ -12,19 +12,27 @@ class CliConfigState:
 CLI_CONFIG_STATE = CliConfigState()
 
 
-def cli_arg(name: str, from_env: str | None = None):
+def cli_arg(name: str, from_env: str | None = None, default: str | None = None) -> str:
     if CLI_CONFIG_STATE.codegen:
         if name not in CLI_CONFIG_STATE.cli_args:
             if from_env is not None:
-                val = os.environ.get(from_env)
+                val = os.environ.get(from_env, default)
                 if val is None:
                     raise Exception(f"neither of cli arg `{name}` nor env var `{from_env}` not setted.")
                 return val
-            raise Exception(f"cli arg `{name}` is not setted.")
+            elif default is not None:
+                return default
+            else:
+                raise Exception(f"cli arg `{name}` is not setted.")
         else:
             return CLI_CONFIG_STATE.cli_args[name]
     else:
-        return CLI_CONFIG_STATE.cli_args.get(name)
+        if default is not None:
+            return CLI_CONFIG_STATE.cli_args.get(name, default)
+        elif name not in CLI_CONFIG_STATE.cli_args:
+            raise Exception(f"cli arg `{name}` is not setted.")
+        else:
+            return CLI_CONFIG_STATE.cli_args[name]
 
 
 class ConfigValue(ABC):

{tinyfan-0.1.0 → tinyfan-0.1.2}/tests/test_codegen.py

@@ -13,7 +13,7 @@ def sample_flow() -> Generator:
     flow: Flow = Flow("test")
 
     @asset(flow=flow, schedule="@daily")
-    def asset1():  # type: ignore
+    def asset1(ds: str):  # type: ignore
         print("asset1 is executed")
         return "hello"
 

{tinyfan-0.1.0 → tinyfan-0.1.2}/uv.lock

@@ -774,7 +774,7 @@ wheels = [
 
 [[package]]
 name = "tinyfan"
-version = "0.1.0"
+version = "0.1.2"
 source = { editable = "." }
 dependencies = [
     { name = "croniter" },