fluent-codegen 0.1.1__tar.gz → 0.3.0__tar.gz

{fluent_codegen-0.1.1/src/fluent_codegen.egg-info → fluent_codegen-0.3.0}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: fluent-codegen
-Version: 0.1.1
+Version: 0.3.0
 Summary: A Python library for generating Python code via AST construction.
 Author-email: Luke Plant <luke@lukeplant.me.uk>
 License-Expression: Apache-2.0
 Project-URL: Repository, https://github.com/spookylukey/fluent-codegen
 Keywords: codegen,code-generation,ast,python,metaprogramming
-Classifier: Development Status :: 3 - Alpha
+Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3 :: Only
@@ -76,7 +76,7 @@ fluent method-chaining for expressions:
    func, _ = module.create_function("fizzbuzz", args=["n"])
 
    # 2. A Name reference to the "n" parameter (Function *is* a Scope)
-   n = codegen.Name("n", func)
+   n = func.name("n")
 
    # 3. Build an if / elif / else chain
    if_stmt = func.body.create_if()
@@ -94,7 +94,7 @@ fluent method-chaining for expressions:
    branch.create_return(codegen.String("Buzz"))
 
    #    else: return str(n)
-   if_stmt.else_block.create_return(codegen.function_call("str", [n], {}, func))
+   if_stmt.else_block.create_return(module.scope.name("str").call([n]))
 
    # 4. Inspect the generated source
    print(module.as_python_source())

{fluent_codegen-0.1.1 → fluent_codegen-0.3.0}/README.rst

@@ -54,7 +54,7 @@ fluent method-chaining for expressions:
    func, _ = module.create_function("fizzbuzz", args=["n"])
 
    # 2. A Name reference to the "n" parameter (Function *is* a Scope)
-   n = codegen.Name("n", func)
+   n = func.name("n")
 
    # 3. Build an if / elif / else chain
    if_stmt = func.body.create_if()
@@ -72,7 +72,7 @@ fluent method-chaining for expressions:
    branch.create_return(codegen.String("Buzz"))
 
    #    else: return str(n)
-   if_stmt.else_block.create_return(codegen.function_call("str", [n], {}, func))
+   if_stmt.else_block.create_return(module.scope.name("str").call([n]))
 
    # 4. Inspect the generated source
    print(module.as_python_source())

{fluent_codegen-0.1.1 → fluent_codegen-0.3.0}/pyproject.toml

@@ -14,7 +14,7 @@ keywords = [
   "metaprogramming",
 ]
 classifiers = [
-  "Development Status :: 3 - Alpha",
+  "Development Status :: 4 - Beta",
   "Intended Audience :: Developers",
   "Operating System :: OS Independent",
   "Programming Language :: Python :: 3 :: Only",
@@ -24,7 +24,7 @@ classifiers = [
   "Topic :: Software Development :: Code Generators",
   "Topic :: Software Development :: Libraries :: Python Modules",
 ]
-version = "0.1.1"
+version = "0.3.0"
 dependencies = []
 
 [project.urls]
@@ -74,7 +74,7 @@ dev = [
     "pyright==1.1.406",
     "pytest>=7.4.4",
     "pytest-cov>=7.0.0",
-    "ruff>=0.4.0",
+    "ruff>=0.15.1",
     "tox>=4.36.0",
     "tox-uv>=1.29.0",
 ]

{fluent_codegen-0.1.1 → fluent_codegen-0.3.0}/src/fluent_codegen/ast_compat.py

@@ -11,6 +11,7 @@ from typing import TypedDict
 
 Add = ast.Add
 And = ast.And
+Assert = ast.Assert
 Assign = ast.Assign
 AnnAssign = ast.AnnAssign
 BoolOp = ast.BoolOp
@@ -77,6 +78,12 @@ alias = ast.alias
 # It's hard to get something sensible we can put for line/col numbers so we put arbitrary values.
 
 
+try:
+    _Unparser = ast._Unparser  # type: ignore
+except AttributeError:  # pragma: no cover
+    from _ast_unparse import Unparser as _Unparser  # pragma: no cover   # type: ignore
+
+
 class DefaultAstArgs(TypedDict):
     lineno: int
     col_offset: int
@@ -91,3 +98,31 @@ DEFAULT_AST_ARGS_ARGUMENTS: dict[str, object] = dict()
 
 def subscript_slice_object[T](value: T) -> T:
     return value
+
+
+class CommentNode(ast.stmt):
+    """Custom AST statement node representing a comment.
+
+    This is not a standard Python AST node. It is ignored by ``compile()``
+    (callers must strip it first, which ``as_ast()`` does automatically),
+    but is rendered by :func:`unparse_with_comments`.
+    """
+
+    _fields = ("text",)
+
+    def __init__(self, text: str, **kwargs: int) -> None:
+        self.text = text
+        super().__init__(**kwargs)  # type: ignore[reportCallIssue]
+
+
+class _CommentUnparser(_Unparser):  # type: ignore[reportAttributeAccessIssue]
+    """An unparser that knows how to render :class:`CommentNode`."""
+
+    def visit_CommentNode(self, node: CommentNode) -> None:
+        self.fill("# " + node.text)  # type: ignore[reportAttributeAccessIssue]
+
+
+def unparse_with_comments(node: ast.AST) -> str:
+    """Like :func:`ast.unparse`, but also renders :class:`CommentNode` nodes."""
+    unparser = _CommentUnparser()
+    return unparser.visit(node)  # type: ignore[reportUnknownMemberType,reportUnknownVariableType]