`AwaitExpr`/`AsyncFunction` - Represents an await expression (common in async programming).

[xmnlab]

from gpt:

1. Update ASTKind Enum in astx/base.py

Add the following lines to your ASTKind enumeration to include the new AwaitExprKind and AsyncFunctionDefKind:

@public
class ASTKind(Enum):
    # ... existing kinds ...

    AwaitExprKind = -813  # Add this line
    AsyncFunctionDefKind = -814  # Add this line

---

2. Define AwaitExpr Class

Add the following class definition to represent the AwaitExpr, likely in astx/expressions.py:

@public
class AwaitExpr(Expr):
    """AST class for await expressions."""

    value: Expr

    def __init__(
        self,
        value: Expr,
        loc: SourceLocation = NO_SOURCE_LOCATION,
        parent: Optional[ASTNodes] = None,
    ) -> None:
        super().__init__(loc=loc, parent=parent)
        self.value = value
        self.kind = ASTKind.AwaitExprKind

Notes:

• Inheritance: AwaitExpr inherits from Expr since it's used where expressions are expected.
• Attributes:
  • value: The expression representing the asynchronous operation to await.
• Usage: Represents await expressions in asynchronous code (e.g., await some_async_function()).

---

3. Define AsyncFunctionDef Class

Add the following class definition to represent an asynchronous function definition, possibly in astx/callables.py:

@public
class AsyncFunctionDef(FunctionDef):
    """AST class for asynchronous function definitions."""

    def __init__(
        self,
        name: str,
        args: Arguments,
        body: Block,
        returns: Optional[Expr] = None,
        decorators: Optional[List[Expr]] = None,
        loc: SourceLocation = NO_SOURCE_LOCATION,
        parent: Optional[ASTNodes] = None,
    ) -> None:
        super().__init__(name, args, body, returns, decorators, loc, parent)
        self.kind = ASTKind.AsyncFunctionDefKind

Notes:

• Inheritance: AsyncFunctionDef inherits from FunctionDef and sets its kind to ASTKind.AsyncFunctionDefKind.
• Attributes: Uses the same attributes as FunctionDef, representing the function's name, arguments, body, return type, and decorators.
• Usage: Represents asynchronous function definitions (e.g., async def foo(): ...).

---

Additional Considerations

• Integration:
  • Ensure that wherever functions are handled in your AST processing, asynchronous functions (AsyncFunctionDef) are appropriately distinguished from regular functions (FunctionDef).
  • Update any visitor or transformer classes to handle the new AsyncFunctionDefKind and AwaitExprKind.
• Imports:
  • Make sure to import necessary classes and types, such as Expr, FunctionDef, Arguments, Block, ASTNodes, SourceLocation, and Optional from the appropriate modules.
• AST Structure:
  • The AwaitExpr can be used within the body of an AsyncFunctionDef to represent awaiting asynchronous operations.
  • The AsyncFunctionDef class allows you to model functions that can use await expressions.

[xmnlab]

PS: in astx, the name of the class is Function, instead of FunctionDef. maybe it would make sense to rename it to FunctionDef in the future