dycw-utilities 0.136.5__tar.gz → 0.136.7__tar.gz

{dycw_utilities-0.136.5 → dycw_utilities-0.136.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.136.5
+Version: 0.136.7
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.136.5 → dycw_utilities-0.136.7}/pyproject.toml

@@ -51,7 +51,7 @@ luigi = ["luigi >= 3.6.0, < 3.7"]
 math-test = ["numpy"]
 memory-profiler = ["memory-profiler >= 0.61.0, < 0.62"]
 more-itertools = ["more-itertools >= 10.7.0, < 10.8"]
-numpy = ["numpy >= 2.3.0, < 2.4"]
+numpy = ["numpy >= 2.3.1, < 2.4"]
 operator = ["polars-lts-cpu"]
 optuna = ["optuna >= 4.4.0, < 4.5"]
 orjson = ["orjson >= 3.10.18, < 3.11"]
@@ -80,7 +80,7 @@ python-dotenv = ["python-dotenv >= 1.1.0, < 1.2"]
 redis = ["redis >= 6.2.0, < 6.3", "orjson"]
 redis-test = ["pytest-rerunfailures"]
 reprlib-test = ["rich"]
-scipy = ["scipy >= 1.15.3, < 1.16"]
+scipy = ["scipy >= 1.16.0, < 1.17"]
 sklearn = ["scikit-learn >= 1.7.0, < 1.8"]
 slack-sdk = ["slack-sdk >= 3.35.0, < 3.36"]
 slack-sdk-test = ["aiohttp"]
@@ -105,7 +105,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.136.5"
+version = "0.136.7"
 
 [project.optional-dependencies]
 logging = [
@@ -133,7 +133,7 @@ test = [
 # bump-my-version
 [tool.bumpversion]
 allow_dirty = true
-current_version = "0.136.5"
+current_version = "0.136.7"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.136.5 → dycw_utilities-0.136.7}/src/tests/test_libcst.py

@@ -22,6 +22,7 @@ from pytest import raises
 from tests.conftest import SKIPIF_CI
 from utilities.iterables import one
 from utilities.libcst import (
+    GenerateImportFromError,
     _ParseImportAliasError,
     _ParseImportEmptyModuleError,
     generate_f_string,
@@ -63,8 +64,10 @@ class TestGenerateImportFrom:
         case=sampled_from([
             ("foo", "bar", None, "from foo import bar"),
             ("foo", "bar", "bar2", "from foo import bar as bar2"),
+            ("foo", "*", None, "from foo import *"),
             ("foo.bar", "baz", None, "from foo.bar import baz"),
             ("foo.bar", "baz", "baz2", "from foo.bar import baz as baz2"),
+            ("foo.bar", "*", None, "from foo.bar import *"),
         ])
     )
     def test_main(self, *, case: tuple[str, str, str | None, str]) -> None:
@@ -73,6 +76,13 @@ class TestGenerateImportFrom:
         result = Module([SimpleStatementLine([imp])]).code.strip("\n")
         assert result == expected
 
+    def test_error(self) -> None:
+        with raises(
+            GenerateImportFromError,
+            match=r"Invalid import: 'from foo import \* as bar'",
+        ):
+            _ = generate_import_from("foo", "*", asname="bar")
+
 
 class TestParseImport:
     def test_import_one_name(self) -> None:

{dycw_utilities-0.136.5 → dycw_utilities-0.136.7}/src/tests/test_sqlalchemy.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
+from enum import Enum, StrEnum, auto
 from itertools import chain
 from typing import TYPE_CHECKING, Any, Literal, cast, overload, override
 from uuid import uuid4
@@ -69,6 +70,8 @@ from utilities.sqlalchemy import (
     create_async_engine,
     ensure_tables_created,
     ensure_tables_dropped,
+    enum_name,
+    enum_values,
     get_chunk_size,
     get_column_names,
     get_columns,
@@ -278,6 +281,25 @@ class TestEnsureTablesDropped:
                 _ = await conn.execute(sel)
 
 
+class TestEnumName:
+    def test_main(self) -> None:
+        class Example(Enum): ...
+
+        result = enum_name(Example)
+        assert result == "example_enum"
+
+
+class TestEnumValues:
+    def test_main(self) -> None:
+        class Example(StrEnum):
+            true = auto()
+            false = auto()
+
+        result = enum_values(Example)
+        expected = ["true", "false"]
+        assert result == expected
+
+
 class TestGetChunkSize:
     @given(chunk_size_frac=floats(0.0, 1.0), scaling=int32s(min_value=0))
     @settings(

{dycw_utilities-0.136.5 → dycw_utilities-0.136.7}/src/utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.136.5"
+__version__ = "0.136.7"

{dycw_utilities-0.136.5 → dycw_utilities-0.136.7}/src/utilities/libcst.py

@@ -43,10 +43,30 @@ def generate_import_from(
     module: str, name: str, /, *, asname: str | None = None
 ) -> ImportFrom:
     """Generate an `ImportFrom` object."""
-    alias = ImportAlias(
-        name=Name(name), asname=AsName(Name(asname)) if asname else None
-    )
-    return ImportFrom(module=split_dotted_str(module), names=[alias])
+    match name, asname:
+        case "*", None:
+            names = ImportStar()
+        case "*", str():
+            raise GenerateImportFromError(module=module, asname=asname)
+        case _, None:
+            alias = ImportAlias(name=Name(name))
+            names = [alias]
+        case _, str():
+            alias = ImportAlias(name=Name(name), asname=AsName(Name(asname)))
+            names = [alias]
+        case _ as never:
+            assert_never(never)
+    return ImportFrom(module=split_dotted_str(module), names=names)
+
+
+@dataclass(kw_only=True, slots=True)
+class GenerateImportFromError(Exception):
+    module: str
+    asname: str | None = None
+
+    @override
+    def __str__(self) -> str:
+        return f"Invalid import: 'from {self.module} import * as {self.asname}'"
 
 
 ##
@@ -168,6 +188,7 @@ def render_module(source: str | Module, /) -> str:
 
 
 __all__ = [
+    "GenerateImportFromError",
     "ParseImportError",
     "generate_f_string",
     "generate_import",

{dycw_utilities-0.136.5 → dycw_utilities-0.136.7}/src/utilities/sqlalchemy.py

@@ -86,6 +86,8 @@ from utilities.types import MaybeIterable, MaybeType, StrMapping, TupleOrStrMapp
 from utilities.whenever import SECOND
 
 if TYPE_CHECKING:
+    from enum import Enum, StrEnum
+
     from whenever import TimeDelta
 
 type _EngineOrConnectionOrAsync = Engine | Connection | AsyncEngine | AsyncConnection
@@ -284,6 +286,22 @@ async def ensure_tables_dropped(
 ##
 
 
+def enum_name(enum: type[Enum], /) -> str:
+    """Get the name of an Enum."""
+    return f"{snake_case(get_class_name(enum))}_enum"
+
+
+##
+
+
+def enum_values(enum: type[StrEnum], /) -> list[str]:
+    """Get the values of a StrEnum."""
+    return [e.value for e in enum]
+
+
+##
+
+
 def get_chunk_size(
     engine_or_conn: _EngineOrConnectionOrAsync,
     /,
@@ -1171,6 +1189,8 @@ __all__ = [
     "create_async_engine",
     "ensure_tables_created",
     "ensure_tables_dropped",
+    "enum_name",
+    "enum_values",
     "get_chunk_size",
     "get_column_names",
     "get_columns",