UncountablePythonSDK 0.0.40__py3-none-any.whl → 0.0.41__py3-none-any.whl

{UncountablePythonSDK-0.0.40.dist-info → UncountablePythonSDK-0.0.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.40
+Version: 0.0.41
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{UncountablePythonSDK-0.0.40.dist-info → UncountablePythonSDK-0.0.41.dist-info}/RECORD

@@ -22,13 +22,14 @@ pkgs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pkgs/argument_parser/__init__.py,sha256=CsQ6QoPKSLLRVl-z6URAmPkiUL9ZPZoV4rJHgy_-RjA,385
 pkgs/argument_parser/_is_enum.py,sha256=Gw6jJa8nBwYGqXwwCZbSnWL8Rvr5alkg5lSVAqXtOZM,257
 pkgs/argument_parser/_is_namedtuple.py,sha256=Rjc1bKanIPPogl3qG5JPBxglG1TqWYOo1nxxhBASQWY,265
-pkgs/argument_parser/argument_parser.py,sha256=pcU4IhgFkea-I6PhHUex43BoVaQvnoRV-Mw3qOqCdBQ,17274
+pkgs/argument_parser/argument_parser.py,sha256=wNGidf80hO0PrPq2zC6NPQXm3Ia_qbrhlsiCXtGIETY,17272
 pkgs/argument_parser/case_convert.py,sha256=NuJLJUJRbyVb6_Slen4uqaStEHbcOS1d-hBBfDrrw-c,605
 pkgs/serialization/__init__.py,sha256=LifasRW0a50A3qRFmo2bf3FQ6TXhZWOTz2-CVTgPjcQ,753
 pkgs/serialization/missing_sentry.py,sha256=aM_9KxbCk9dVvXvcOKgkIQBqFWvLhv8QlIUCiuFEXMo,806
 pkgs/serialization/opaque_key.py,sha256=FIfXEE0DA1U8R_taFbQ1RCoTSgehrPjP06-qvo-GeNQ,177
 pkgs/serialization/serial_class.py,sha256=r0hrQdIbJA_X0W0_jKEVrxi_JzVRT9qHCjsUgGu3cCI,5290
 pkgs/serialization/serial_union.py,sha256=z8Ptj4bVHyb1ROfg0UPTwZ6Ef6iXLr0YJfAH5o_PU9A,2601
+pkgs/serialization/yaml.py,sha256=t-31FS0pL1YbhfspK88iQEGa1WDn60Fw16CpoLSClJM,1453
 pkgs/serialization_util/__init__.py,sha256=MVKqHTUl2YnWZAFG9xCxu1SgmkQ5xPofrAGlYg6h7rI,330
 pkgs/serialization_util/_get_type_for_serialization.py,sha256=dW5_W9MFd6wgWfW5qlWork-GBb-QFLtiOZkjk2Zqn2M,1177
 pkgs/serialization_util/convert_to_snakecase.py,sha256=H2BAo5ZdcCDN77RpLb-uP0s7-FQ5Ukwnsd3VYc1vD0M,583
@@ -38,14 +39,14 @@ pkgs/strenum_compat/strenum_compat.py,sha256=uOUAgpYTjHs1MX8dG81jRlyTkt3KNbkV_25
 pkgs/type_spec/__init__.py,sha256=h5DmJTca4QVV10sZR1x0-MlkZfuGYDfapR3zHvXfzto,19
 pkgs/type_spec/__main__.py,sha256=5bJaX9Y_-FavP0qwzhk-z-V97UY7uaezJTa1zhO_HHQ,1048
 pkgs/type_spec/builder.py,sha256=xQcY2HcQTI2FSOMycgx3yD23_Oz3_LfWdyW65pDaHoc,46667
-pkgs/type_spec/config.py,sha256=INfEiDcUsZFUKasHprsE6i33siPB0RnfmTKOsWcGnQ8,5043
+pkgs/type_spec/config.py,sha256=IQyo2Vj11uNt7_d6jQxvominAOU-oPB8ldEmuGzJLpU,4644
 pkgs/type_spec/emit_io_ts.py,sha256=Ghd8XYqyNYldHQDepwa9GLfHXcoi48ztBw84K28ETic,5707
-pkgs/type_spec/emit_open_api.py,sha256=NVVXAvjPHC_MPqch_SMY6klj3kPnckOa7fJsb-ZsFTs,24371
+pkgs/type_spec/emit_open_api.py,sha256=8BTZpiQlD628WoFsjDVzRdmM0ZBfMP19QSJOOcdZoLQ,24397
 pkgs/type_spec/emit_open_api_util.py,sha256=9tWLMT6NTeCa2caO8DdWo6aYoi0b4AEFiAXNaaQKJIs,2373
 pkgs/type_spec/emit_python.py,sha256=djF_nUs0v3c5NI8BJjOj0gbLtf_PgbMgk5CCgvh2aPg,46469
 pkgs/type_spec/emit_typescript.py,sha256=cdr5h8N70PuwORcvhURUujzwH9r1LVwJB8V2EoipGkw,17917
 pkgs/type_spec/emit_typescript_util.py,sha256=sR7ys3Ilnh6SQiXJbfYk4pxfOu0bDjbUFTEYEW-ud6c,863
-pkgs/type_spec/load_types.py,sha256=xEHwdB_miR3vNs161Oy1luafE0VC-yk9-utAyCJmbEo,3629
+pkgs/type_spec/load_types.py,sha256=BOLyndtxPqqhUqZAh-lIbN5IZBaW_m-bdYpKGsbPyXM,3654
 pkgs/type_spec/open_api_util.py,sha256=IGh-_snGPST_P_8FdYtO8MTEa9PUxRW6Rzg9X9EgQik,7114
 pkgs/type_spec/test.py,sha256=4ueujBq-pEgnX3Z69HyPmD-bullFXmpixcpVzfOkhP4,489
 pkgs/type_spec/util.py,sha256=6m6MPfY-SwjyZf2FWQKclswWB5o7gcdd-3tdpViPYOQ,4844
@@ -57,7 +58,7 @@ pkgs/type_spec/parts/base.ts.prepart,sha256=2FJJvpg2olCcavxj0nbYWdwKl6KeScour2Jj
 pkgs/type_spec/type_info/__main__.py,sha256=pmVjVqXyVh8vKTNCTFgz80Sg74C5BKToP3E6GS-X_So,857
 pkgs/type_spec/type_info/emit_type_info.py,sha256=7FtMm_WOhxoT8Hy2DDorexIccwCNVZ9mJTBURD8l9Rk,13282
 pkgs/type_spec/value_spec/__init__.py,sha256=Z-grlcZtxAfEXhPHsK0nD7PFLGsv4eqvunaPN7_TA84,83
-pkgs/type_spec/value_spec/__main__.py,sha256=-9L5pXYx02plnTetqNknaUZPieLRtzbyWdZDT6B-cWA,8294
+pkgs/type_spec/value_spec/__main__.py,sha256=6bzP85p_Cm4bPp5tXz8D_4p64wMn5SKsXC7SqSZquYc,8318
 pkgs/type_spec/value_spec/convert_type.py,sha256=SAYyEV6orQJJbkXSE4hhtOQJ2vKUXJCKPeYPrB8G9oA,2272
 pkgs/type_spec/value_spec/emit_python.py,sha256=rjg6LIGYdaagrZ19XpDfW_Z7LPNwCMDceBje5dsMFbw,6959
 pkgs/type_spec/value_spec/types.py,sha256=a2zxbbCRWepY1l8OtjeCDKgBKFPFHVgV99oP6pTtaro,441
@@ -70,9 +71,9 @@ uncountable/core/file_upload.py,sha256=t4cutIFB5rNN2qVCx3H0HotSjCSJSxis2_QBarDuX
 uncountable/core/types.py,sha256=s2CjqYJpsmbC7xMwxxT7kJ_V9bwokrjjWVVjpMcQpKI,333
 uncountable/integration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/integration/construct_client.py,sha256=N3K-wck7teqvHTrOh0N2bGF3u6kyuwQvgo7gsKLihQ0,1077
-uncountable/integration/cron.py,sha256=DNnMkeLYi70U-3wrcDAX382FgEGq16KL6xy-6OBv5Us,938
-uncountable/integration/entrypoint.py,sha256=MNXl9xv0N03V3_0vplckWlUio2CDFJ0jsjM8uvY1dTE,1404
-uncountable/integration/job.py,sha256=u4Vhm9l5KUvex7I7R_ysBHjmm1BBB1KvnjRvZvFkOig,807
+uncountable/integration/cron.py,sha256=JIBgHUi4BqUULEPwYCZlDsvdJXLtl-EgCt7RVIdVmSE,995
+uncountable/integration/entrypoint.py,sha256=zpYd1mzbc7uZ68IyXIkODqKPvzsUZ_3Lkj2H2bhKor0,1377
+uncountable/integration/job.py,sha256=Cy10LUfy2HaGcVz0uH7I00WeyyFYKYgEd-rE_OqVkUc,819
 uncountable/integration/server.py,sha256=5Wh80VyfgUUHA5NiFGn9LjS4KJ9zJz3QbNTLLKhFAso,3342
 uncountable/integration/types.py,sha256=n9idu2_qHOA5CQdE6NK8HS6aZ8ugTZKTfTTnm4sy8tI,2066
 uncountable/integration/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -216,7 +217,7 @@ uncountable/types/api/recipes/unarchive_recipes.py,sha256=W_tOZgz_FEVQcI3y1tnxJz
 uncountable/types/api/recipes/unlock_recipes.py,sha256=BORmGgYjQfBVRnqimVXc-RcvG3QJkvexesTGuKdgoOw,1094
 uncountable/types/api/triggers/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/triggers/run_trigger.py,sha256=nV81yBsxmBi5u4hJyO8RhOd6zG0BK0y1aN33zLDbJmM,884
-UncountablePythonSDK-0.0.40.dist-info/METADATA,sha256=shiGvKxG1hChoJWA-x-1_EFJOOYRGWZb77vshDSgc6U,1577
-UncountablePythonSDK-0.0.40.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-UncountablePythonSDK-0.0.40.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-UncountablePythonSDK-0.0.40.dist-info/RECORD,,
+UncountablePythonSDK-0.0.41.dist-info/METADATA,sha256=Y8go69WL8lqmsIZjvbolQvZSZKv2aHelEUa0Ghi2ZI0,1577
+UncountablePythonSDK-0.0.41.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+UncountablePythonSDK-0.0.41.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+UncountablePythonSDK-0.0.41.dist-info/RECORD,,

pkgs/argument_parser/argument_parser.py

@@ -8,13 +8,13 @@ from decimal import Decimal
 from importlib import resources
 
 import dateutil.parser
-import yaml
 
 from pkgs.serialization import (
     MissingSentryType,
     OpaqueKey,
     get_serial_class_data,
     get_serial_union_data,
+    yaml,
 )
 
 from ._is_enum import is_string_enum_class

pkgs/serialization/yaml.py

@@ -0,0 +1,54 @@
+from decimal import Decimal
+from typing import TYPE_CHECKING, Any
+
+import yaml
+
+if TYPE_CHECKING:
+    from _typeshed import SupportsRead as SupportsReadT
+    from _typeshed import SupportsWrite as SupportsWriteT
+
+    SupportsRead = SupportsReadT[Any]
+    SupportsWrite = SupportsWriteT[Any]
+else:
+    SupportsRead = object
+    SupportsWrite = object
+
+
+def _decimal_constructor(loader, node):  # type:ignore
+    value = loader.construct_scalar(node)
+    return Decimal(value)
+
+
+# A semi-acceptable patch to force a number to be parsed as a decimal, since pyyaml
+# parses them as lossy floats otherwise. Though a bit ugly, at least this way we have
+# support for decimal constants
+yaml.SafeLoader.add_constructor("!decimal", _decimal_constructor)
+
+
+class YAMLError(BaseException):
+    pass
+
+
+def dumps(obj: Any, sort_keys: bool = False) -> str:
+    return yaml.dump(obj, sort_keys=sort_keys)
+
+
+def dump(obj: Any, f: SupportsWrite, sort_keys: bool = False) -> None:
+    yaml.dump(obj, f, sort_keys=sort_keys)
+
+
+def safe_load(src: str | bytes | SupportsRead) -> Any:
+    try:
+        return yaml.safe_load(src)
+    except yaml.YAMLError as e:
+        raise YAMLError() from e
+
+
+def safe_dump(
+    obj: Any, sort_keys: bool = False, indent: int | None = None, width: int | None = None
+) -> str:
+    return yaml.safe_dump(obj, sort_keys=sort_keys, indent=indent, width=width)
+
+
+def c_load(f: SupportsRead) -> Any:
+    return yaml.load(f, Loader=yaml.CLoader)

pkgs/type_spec/config.py

@@ -1,10 +1,9 @@
 import os
 from collections.abc import Callable, Mapping
 from dataclasses import dataclass
-from decimal import Decimal
 from typing import Self, TypeVar
 
-import yaml
+from pkgs.serialization import yaml
 
 ConfigValueType = str | None | Mapping[str, str | None] | list[str]
 
@@ -103,17 +102,6 @@ def _parse_language(config_class: type[_T], raw_value: ConfigValueType) -> _T:
     return config_class(**raw_value)
 
 
-def _decimal_constructor(loader, node):  # type:ignore
-    value = loader.construct_scalar(node)
-    return Decimal(value)
-
-
-# A semi-acceptable patch to force a number to be parsed as a decimal, since pyyaml
-# parses them as lossy floats otherwise. Though a bit ugly, at least this way we have
-# support for decimal constants
-yaml.SafeLoader.add_constructor("!decimal", _decimal_constructor)
-
-
 def parse_yaml_config(config_file: str) -> Config:
     with open(config_file, encoding="utf-8") as input:
         raw_config: dict[str, ConfigValueType] = yaml.safe_load(input)

pkgs/type_spec/emit_open_api.py

@@ -9,7 +9,7 @@ import json
 import re
 from typing import Collection, cast
 
-import yaml
+from pkgs.serialization import yaml
 
 from . import builder, util
 from .builder import EndpointGuideKey, RootGuideKey
@@ -169,7 +169,7 @@ def _serialize_global_context(ctx: EmitOpenAPIGlobalContext) -> str:
         oa_paths[path.path] = {"$ref": path.ref}
     oa_root["paths"] = oa_paths
 
-    return yaml.dump(oa_root, sort_keys=False)
+    return yaml.dumps(oa_root, sort_keys=False)
 
 
 def _is_empty_object_type(typ: OpenAPIType) -> bool:
@@ -413,7 +413,7 @@ def _emit_namespace(
 
     path = f"{config.types_output}/common/{'/'.join(namespace.path)}.yaml"
     oa_namespace = {"components": oa_components}
-    _rewrite_with_notice(path, yaml.dump(oa_namespace, sort_keys=False))
+    _rewrite_with_notice(path, yaml.dumps(oa_namespace, sort_keys=False))
 
 
 def _emit_type(

pkgs/type_spec/load_types.py

@@ -3,9 +3,10 @@ from collections.abc import Callable
 from io import StringIO
 from typing import Optional
 
-import yaml
 from shelljob import fs
 
+from pkgs.serialization import yaml
+
 from .builder import SpecBuilder
 from .config import Config
 

pkgs/type_spec/value_spec/__main__.py

@@ -20,9 +20,9 @@ import sys
 from typing import TypeVar, cast
 
 import regex as re
-import yaml
 
 from main.base.types import base_t, value_spec_t
+from pkgs.serialization import yaml
 
 from ..util import parse_type_str, rewrite_file
 from .convert_type import convert_to_value_spec_type

uncountable/integration/cron.py

@@ -27,4 +27,6 @@ def cron_job_executor(**kwargs: dict) -> None:
         args_passed.definition.executor, args_passed.profile_metadata
     )
 
+    print(f"running job {args_passed.definition.name}")
+
     job.run(args=args)

uncountable/integration/entrypoint.py

@@ -9,10 +9,9 @@ from uncountable.integration.types import ProfileDefinition
 profile_parser = CachedParser(ProfileDefinition)
 
 
-def main() -> None:
+def main(blocking: bool) -> None:
     profiles_module = os.environ["UNC_PROFILES_MODULE"]
     with IntegrationServer(create_db_engine()) as server:
-        # TODO: Loop through all job spec yaml files and call server.add_job
         profiles = [
             entry
             for entry in resources.files(profiles_module).iterdir()
@@ -26,7 +25,7 @@ def main() -> None:
                     resource="profile.yaml",
                 )
             except FileNotFoundError as e:
-                print("WARN: profile.yaml not found", e)
+                print(f"WARN: profile.yaml not found for {profile_name}", e)
                 continue
             server.register_profile(
                 profile_name=profile_name,
@@ -35,8 +34,8 @@ def main() -> None:
                 jobs=profile.jobs,
             )
 
-        server.serve_forever()
+        if blocking:
+            server.serve_forever()
 
 
-if __name__ == "__main__":
-    main()
+main(__name__ == "__main__")

uncountable/integration/job.py

@@ -37,6 +37,6 @@ class CronJob(Job):
     def run(self, args: CronJobArguments) -> JobResult: ...
 
 
-def register_job(cls: Job) -> Job:
+def register_job(cls: type[Job]) -> type[Job]:
     cls._unc_job_registered = True
     return cls