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

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

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

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

@@ -105,7 +105,7 @@ dependencies = [
 name = "dycw-utilities"
 readme = "README.md"
 requires-python = ">= 3.12"
-version = "0.136.5"
+version = "0.136.6"
 
 [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.6"
 
 [[tool.bumpversion.files]]
 filename = "src/utilities/__init__.py"

{dycw_utilities-0.136.5 → dycw_utilities-0.136.6}/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.6}/src/utilities/__init__.py

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

{dycw_utilities-0.136.5 → dycw_utilities-0.136.6}/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",