UncountablePythonSDK 0.0.40__tar.gz → 0.0.41__tar.gz

{uncountablepythonsdk-0.0.40/UncountablePythonSDK.egg-info → uncountablepythonsdk-0.0.41}/PKG-INFO

@@ -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 → uncountablepythonsdk-0.0.41/UncountablePythonSDK.egg-info}/PKG-INFO

@@ -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 → uncountablepythonsdk-0.0.41}/UncountablePythonSDK.egg-info/SOURCES.txt

@@ -39,6 +39,7 @@ pkgs/serialization/missing_sentry.py
 pkgs/serialization/opaque_key.py
 pkgs/serialization/serial_class.py
 pkgs/serialization/serial_union.py
+pkgs/serialization/yaml.py
 pkgs/serialization_util/__init__.py
 pkgs/serialization_util/_get_type_for_serialization.py
 pkgs/serialization_util/convert_to_snakecase.py

{uncountablepythonsdk-0.0.40 → uncountablepythonsdk-0.0.41}/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

uncountablepythonsdk-0.0.41/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)

{uncountablepythonsdk-0.0.40 → uncountablepythonsdk-0.0.41}/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)

{uncountablepythonsdk-0.0.40 → uncountablepythonsdk-0.0.41}/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(

{uncountablepythonsdk-0.0.40 → uncountablepythonsdk-0.0.41}/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
 

{uncountablepythonsdk-0.0.40 → uncountablepythonsdk-0.0.41}/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

{uncountablepythonsdk-0.0.40 → uncountablepythonsdk-0.0.41}/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)

{uncountablepythonsdk-0.0.40 → uncountablepythonsdk-0.0.41}/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__")

{uncountablepythonsdk-0.0.40 → uncountablepythonsdk-0.0.41}/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