feat(typescript) [DRAFT] Add support for enums in Python v2 generator

[GrantRheingold]

Description

This PR adds support for generating enums in python using a typescript generator function.

Changes Made

• Creates Enum and EnumValue AST nodes for structured enum generation
• Implements proper Python enum class structure with inheritance from enum.Enum
• Adds visit method generation with type-safe callable parameters for pattern matching
• Creates a new EnumGenerator with enum-specific generator logic

Future Work

• As discussed, the imports should be prefixed by the package to avoid cross-module confusion between imported packages. So the final implementation should have typing.Union[...] instead of from typing import Union ... Union[...]
• Additionally, Enums are probably more naturally written as extensions of a Class, so this implementation is probably best left to extending that

Testing

• Manual testing completed

Tested by running

pnpm seed test --generator pydantic-v2 --log-level debug --fixture exhaustive --skip-scripts --exhaustive

and comparing the output to the generated output of the python-native generated file here.

from typing import TypeVar
from enum import Enum

class WeatherReport(Enum):
    Sunny = "SUNNY"
    Cloudy = "CLOUDY"
    Raining = "RAINING"
    Snowing = "SNOWING"

[graphite-app]

There appears to be a syntax error in the generated visit method. The parameter definitions incorrectly include the enum values as part of the parameter names (e.g., ApiError = "API_ERROR" instead of just ApiError). Similarly, the if statements contain malformed conditions that append the enum values after the enum member references.

The correct Python syntax should look like:

def visit(
    self,
    ApiError: typing.Callable[[], T_Result],
    AuthenticationError: typing.Callable[[], T_Result],
    InvalidRequestError: typing.Callable[[], T_Result]
) -> T_Result:
    if self is ErrorCategory.ApiError:
        return ApiError()
    if self is ErrorCategory.AuthenticationError:
        return AuthenticationError()
    if self is ErrorCategory.InvalidRequestError:
        return InvalidRequestError()

This issue appears to originate in the Enum.ts write method implementation that's generating the Python code.

Suggested change

	def visit(
	self,
	ApiError = "API_ERROR": typing.Callable[[], T_Result],
	AuthenticationError = "AUTHENTICATION_ERROR": typing.Callable[[], T_Result],
	InvalidRequestError = "INVALID_REQUEST_ERROR": typing.Callable[[], T_Result]
	) -> T_Result:
	if self is ErrorCategory.ApiError = "API_ERROR":
	return ApiError = "API_ERROR"()
	if self is ErrorCategory.AuthenticationError = "AUTHENTICATION_ERROR":
	return AuthenticationError = "AUTHENTICATION_ERROR"()
	if self is ErrorCategory.InvalidRequestError = "INVALID_REQUEST_ERROR":
	return InvalidRequestError = "INVALID_REQUEST_ERROR"()
	def visit(
	self,
	ApiError: typing.Callable[[], T_Result],
	AuthenticationError: typing.Callable[[], T_Result],
	InvalidRequestError: typing.Callable[[], T_Result]
	) -> T_Result:
	if self is ErrorCategory.ApiError:
	return ApiError()
	if self is ErrorCategory.AuthenticationError:
	return AuthenticationError()
	if self is ErrorCategory.InvalidRequestError:
	return InvalidRequestError()

Spotted by Diamond

Is this helpful? React 👍 or 👎 to let us know.