django-cfg 1.4.11__py3-none-any.whl → 1.4.13__py3-none-any.whl

django_cfg/modules/django_client/core/generator/python/generator.py

@@ -0,0 +1,182 @@
+"""
+Python Generator - Main orchestrator for Python client generation.
+
+Coordinates all sub-generators:
+- ModelsGenerator: Pydantic models and enums
+- OperationsGenerator: Operation methods (async/sync)
+- AsyncClientGenerator: Async client classes
+- SyncClientGenerator: Sync client classes
+- FilesGenerator: Auxiliary files (__init__.py, logger, schema)
+"""
+
+from __future__ import annotations
+
+import pathlib
+from jinja2 import Environment, FileSystemLoader, select_autoescape
+
+from ..base import BaseGenerator, GeneratedFile
+from ...ir import IROperationObject, IRSchemaObject
+
+from .models_generator import ModelsGenerator
+from .operations_generator import OperationsGenerator
+from .async_client_gen import AsyncClientGenerator
+from .sync_client_gen import SyncClientGenerator
+from .files_generator import FilesGenerator
+
+
+class PythonGenerator(BaseGenerator):
+    """
+    Python client generator.
+
+    Generates:
+    - models.py: Pydantic 2 models (User, UserRequest, PatchedUser)
+    - enums.py: Enum classes (StatusEnum, RoleEnum)
+    - client.py: AsyncClient with all operations
+    - sync_client.py: SyncClient with all operations
+    - __init__.py: Package exports
+    """
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        # Setup Jinja2 environment
+        templates_dir = pathlib.Path(__file__).parent / "templates"
+        self.jinja_env = Environment(
+            loader=FileSystemLoader(str(templates_dir)),
+            autoescape=select_autoescape(['html', 'xml']),
+            trim_blocks=True,
+            lstrip_blocks=True,
+        )
+
+        # Initialize sub-generators
+        self.models_gen = ModelsGenerator(self.jinja_env, self.context, self)
+        self.operations_gen = OperationsGenerator(self.jinja_env, self)
+        self.async_client_gen = AsyncClientGenerator(self.jinja_env, self.context, self, self.operations_gen)
+        self.sync_client_gen = SyncClientGenerator(self.jinja_env, self, self.operations_gen)
+        self.files_gen = FilesGenerator(self.jinja_env, self.context, self)
+
+    def generate(self) -> list[GeneratedFile]:
+        """Generate all Python client files."""
+        files = []
+
+        if self.client_structure == "namespaced":
+            # Generate per-app folders
+            ops_by_tag = self.group_operations_by_tag()
+
+            for tag, operations in sorted(ops_by_tag.items()):
+                # Generate app folder (models.py, client.py, sync_client.py, __init__.py)
+                files.extend(self._generate_app_folder(tag, operations))
+
+            # Generate shared enums.py (Variant 2: all enums in root)
+            all_schemas = self.context.schemas
+            all_enums = self._collect_enums_from_schemas(all_schemas)
+            if all_enums:
+                files.append(self.models_gen.generate_shared_enums_file(all_enums))
+
+            # Generate main async client.py
+            files.append(self.async_client_gen.generate_main_client_file(ops_by_tag))
+
+            # Generate main sync client.py
+            files.append(self.sync_client_gen.generate_sync_main_client_file(ops_by_tag))
+
+            # Generate main __init__.py
+            files.append(self.files_gen.generate_main_init_file())
+
+            # Generate logger.py with Rich
+            files.append(self.files_gen.generate_logger_file())
+
+            # Generate retry.py with tenacity
+            files.append(self.files_gen.generate_retry_file())
+
+            # Generate schema.py with OpenAPI schema
+            if self.openapi_schema:
+                files.append(self.files_gen.generate_schema_file(self.openapi_schema))
+        else:
+            # Flat structure (original logic)
+            files.append(self.models_gen.generate_models_file())
+
+            enum_schemas = self.get_enum_schemas()
+            if enum_schemas:
+                files.append(self.models_gen.generate_enums_file())
+
+            files.append(self.async_client_gen.generate_client_file())
+            files.append(self.files_gen.generate_init_file())
+
+            # Generate logger.py with Rich
+            files.append(self.files_gen.generate_logger_file())
+
+            # Generate retry.py with tenacity
+            files.append(self.files_gen.generate_retry_file())
+
+            # Generate schema.py with OpenAPI schema
+            if self.openapi_schema:
+                files.append(self.files_gen.generate_schema_file(self.openapi_schema))
+
+        # Generate package files if requested
+        if self.generate_package_files:
+            files.append(self.files_gen.generate_pyproject_toml_file(self.package_config))
+
+        return files
+
+    def _generate_app_folder(self, tag: str, operations: list[IROperationObject]) -> list[GeneratedFile]:
+        """Generate folder for a specific app (tag)."""
+        files = []
+
+        # Get schemas used by this app
+        app_schemas = self._get_schemas_for_operations(operations)
+
+        # Generate models.py for this app
+        files.append(self.models_gen.generate_app_models_file(tag, app_schemas, operations))
+
+        # Generate async client.py for this app
+        files.append(self.async_client_gen.generate_app_client_file(tag, operations))
+
+        # Generate sync client.py for this app
+        files.append(self.sync_client_gen.generate_app_sync_client_file(tag, operations))
+
+        # Generate __init__.py for this app
+        files.append(self.files_gen.generate_app_init_file(tag, operations))
+
+        return files
+
+    def _get_schemas_for_operations(self, operations: list[IROperationObject]) -> dict[str, IRSchemaObject]:
+        """Get all schemas used by given operations."""
+        schemas = {}
+
+        for operation in operations:
+            # Request body schemas
+            if operation.request_body and operation.request_body.schema_name:
+                schema_name = operation.request_body.schema_name
+                if schema_name in self.context.schemas:
+                    schemas[schema_name] = self.context.schemas[schema_name]
+
+            # Patch request body schemas
+            if operation.patch_request_body and operation.patch_request_body.schema_name:
+                schema_name = operation.patch_request_body.schema_name
+                if schema_name in self.context.schemas:
+                    schemas[schema_name] = self.context.schemas[schema_name]
+
+            # Response schemas
+            for status_code, response in operation.responses.items():
+                if response.schema_name:
+                    if response.schema_name in self.context.schemas:
+                        schemas[response.schema_name] = self.context.schemas[response.schema_name]
+
+        return schemas
+
+    # Backward compatibility - delegate to sub-generators
+    def generate_schema(self, schema: IRSchemaObject) -> str:
+        """Generate Pydantic model for schema (delegates to ModelsGenerator)."""
+        return self.models_gen.generate_schema(schema)
+
+    def generate_enum(self, schema: IRSchemaObject) -> str:
+        """Generate Enum class (delegates to ModelsGenerator)."""
+        return self.models_gen.generate_enum(schema)
+
+    def generate_operation(self, operation: IROperationObject, remove_tag_prefix: bool = False) -> str:
+        """Generate async method for operation (delegates to OperationsGenerator)."""
+        return self.operations_gen.generate_async_operation(operation, remove_tag_prefix)
+
+    def generate_sync_operation(self, operation: IROperationObject, remove_tag_prefix: bool = False) -> str:
+        """Generate sync method for operation (delegates to OperationsGenerator)."""
+        return self.operations_gen.generate_sync_operation(operation, remove_tag_prefix)

django_cfg/modules/django_client/core/generator/python/models_generator.py

@@ -0,0 +1,318 @@
+"""
+Models Generator - Generates Pydantic models and enums.
+
+Handles:
+- Pydantic 2 model classes (Request/Response/Patch)
+- Enum classes (StrEnum, IntEnum)
+- Field validation and constraints
+"""
+
+from __future__ import annotations
+
+from jinja2 import Environment
+
+from ..base import GeneratedFile
+from ...ir import IRSchemaObject
+
+
+class ModelsGenerator:
+    """Generates Pydantic models and enum classes."""
+
+    def __init__(self, jinja_env: Environment, context, base_generator):
+        """
+        Initialize models generator.
+
+        Args:
+            jinja_env: Jinja2 environment for templates
+            context: Generation context with schemas
+            base_generator: Reference to base generator for utility methods
+        """
+        self.jinja_env = jinja_env
+        self.context = context
+        self.base = base_generator
+
+    def generate_models_file(self) -> GeneratedFile:
+        """Generate models.py with all Pydantic models."""
+        # Generate all schemas
+        schema_codes = []
+
+        # Response models first
+        for name, schema in self.base.get_response_schemas().items():
+            schema_codes.append(self.generate_schema(schema))
+
+        # Request models
+        for name, schema in self.base.get_request_schemas().items():
+            schema_codes.append(self.generate_schema(schema))
+
+        # Patch models
+        for name, schema in self.base.get_patch_schemas().items():
+            schema_codes.append(self.generate_schema(schema))
+
+        template = self.jinja_env.get_template('models/models.py.jinja')
+        content = template.render(
+            has_enums=bool(self.base.get_enum_schemas()),
+            schemas=schema_codes
+        )
+
+        return GeneratedFile(
+            path="models.py",
+            content=content,
+            description="Pydantic 2 models (Request/Response/Patch)",
+        )
+
+    def generate_enums_file(self) -> GeneratedFile:
+        """Generate enums.py with all Enum classes (flat structure)."""
+        # Generate all enums
+        enum_codes = []
+        for name, schema in self.base.get_enum_schemas().items():
+            enum_codes.append(self.generate_enum(schema))
+
+        template = self.jinja_env.get_template('models/enums.py.jinja')
+        content = template.render(enums=enum_codes)
+
+        return GeneratedFile(
+            path="enums.py",
+            content=content,
+            description="Enum classes from x-enum-varnames",
+        )
+
+    def generate_shared_enums_file(self, enums: dict[str, IRSchemaObject]) -> GeneratedFile:
+        """Generate shared enums.py for namespaced structure (Variant 2)."""
+        # Generate all enums
+        enum_codes = []
+        for name, schema in enums.items():
+            enum_codes.append(self.generate_enum(schema))
+
+        template = self.jinja_env.get_template('models/enums.py.jinja')
+        content = template.render(enums=enum_codes)
+
+        return GeneratedFile(
+            path="enums.py",
+            content=content,
+            description="Shared enum classes from x-enum-varnames",
+        )
+
+    def generate_schema(self, schema: IRSchemaObject) -> str:
+        """Generate Pydantic model for schema."""
+        if schema.type != "object":
+            # For primitive types, skip (they'll be inlined)
+            return ""
+
+        # Class docstring
+        docstring_lines = []
+        if schema.description:
+            docstring_lines.extend(self.base.wrap_comment(schema.description, 76))
+
+        # Add metadata about model type
+        if schema.is_request_model:
+            docstring_lines.append("")
+            docstring_lines.append("Request model (no read-only fields).")
+        elif schema.is_patch_model:
+            docstring_lines.append("")
+            docstring_lines.append("PATCH model (all fields optional).")
+        elif schema.is_response_model:
+            docstring_lines.append("")
+            docstring_lines.append("Response model (includes read-only fields).")
+
+        docstring = "\n".join(docstring_lines) if docstring_lines else None
+
+        # Fields
+        field_lines = []
+        for prop_name, prop_schema in schema.properties.items():
+            field_lines.append(self._generate_field(prop_name, prop_schema, schema.required))
+
+        template = self.jinja_env.get_template('models/schema_class.py.jinja')
+        return template.render(
+            name=schema.name,
+            docstring=docstring,
+            fields=field_lines
+        )
+
+    def _generate_field(
+        self,
+        name: str,
+        schema: IRSchemaObject,
+        required_fields: list[str],
+    ) -> str:
+        """
+        Generate Pydantic field definition.
+
+        Examples:
+            id: int
+            username: str
+            email: str | None = None
+            age: int = Field(..., ge=0, le=150)
+            status: StatusEnum
+        """
+        # Check if this field is an enum
+        if schema.enum and schema.name:
+            # Use enum type from shared enums (sanitized to PascalCase)
+            python_type = self.base.sanitize_enum_name(schema.name)
+            if schema.nullable:
+                python_type = f"{python_type} | None"
+        # Check if this field is a reference to an enum (via $ref)
+        elif schema.ref and schema.ref in self.context.schemas:
+            ref_schema = self.context.schemas[schema.ref]
+            if ref_schema.enum:
+                # This is a reference to an enum component (sanitized to PascalCase)
+                python_type = self.base.sanitize_enum_name(schema.ref)
+                if schema.nullable:
+                    python_type = f"{python_type} | None"
+            else:
+                # Regular reference
+                python_type = schema.python_type
+        else:
+            # Get Python type
+            python_type = schema.python_type
+
+        # Check if required
+        is_required = name in required_fields
+
+        # Build Field() kwargs
+        field_kwargs = []
+
+        if schema.description:
+            field_kwargs.append(f"description={schema.description!r}")
+
+        # Validation constraints
+        if schema.min_length is not None:
+            field_kwargs.append(f"min_length={schema.min_length}")
+        if schema.max_length is not None:
+            field_kwargs.append(f"max_length={schema.max_length}")
+        if schema.pattern:
+            field_kwargs.append(f"pattern={schema.pattern!r}")
+        if schema.minimum is not None:
+            field_kwargs.append(f"ge={schema.minimum}")
+        if schema.maximum is not None:
+            field_kwargs.append(f"le={schema.maximum}")
+
+        # Example
+        if schema.example:
+            field_kwargs.append(f"examples=[{schema.example!r}]")
+
+        # Default value
+        if is_required:
+            if field_kwargs:
+                default = f"Field({', '.join(field_kwargs)})"
+            else:
+                default = "..."
+        else:
+            if field_kwargs:
+                default = f"Field(None, {', '.join(field_kwargs)})"
+            else:
+                default = "None"
+
+        return f"{name}: {python_type} = {default}"
+
+    def generate_enum(self, schema: IRSchemaObject) -> str:
+        """Generate Enum class from x-enum-varnames."""
+        # Determine enum base class
+        if schema.type == "integer":
+            base_class = "IntEnum"
+        else:
+            base_class = "StrEnum"
+
+        # Sanitize enum class name (convert to PascalCase)
+        # "OrderDetail.status" → "OrderDetailStatus"
+        # "Currency.currency_type" → "CurrencyCurrencyType"
+        enum_name = self.base.sanitize_enum_name(schema.name)
+
+        # Class docstring
+        docstring = None
+        if schema.description:
+            # Format enum description to split bullet points
+            docstring = self.base.format_enum_description(schema.description)
+
+        # Enum members
+        member_lines = []
+        for var_name, value in zip(schema.enum_var_names, schema.enum):
+            # Skip empty values (from blank=True in Django)
+            if not var_name or (isinstance(value, str) and value == ''):
+                continue
+
+            if isinstance(value, str):
+                member_lines.append(f'{var_name} = "{value}"')
+            else:
+                member_lines.append(f"{var_name} = {value}")
+
+        template = self.jinja_env.get_template('models/enum_class.py.jinja')
+        return template.render(
+            name=enum_name,
+            base_class=base_class,
+            docstring=docstring,
+            members=member_lines
+        )
+
+    def generate_app_models_file(
+        self,
+        tag: str,
+        schemas: dict[str, IRSchemaObject],
+        operations: list
+    ) -> GeneratedFile:
+        """Generate models.py for a specific app (namespaced structure)."""
+        schema_codes = []
+
+        # Collect enum names used in these schemas
+        enum_names = set()
+
+        # Track seen schema names to avoid duplicates
+        seen_schemas = set()
+
+        # Response models
+        for name, schema in schemas.items():
+            if schema.is_response_model and name not in seen_schemas:
+                schema_codes.append(self.generate_schema(schema))
+                seen_schemas.add(name)
+                # Collect enums from properties
+                if schema.properties:
+                    for prop in schema.properties.values():
+                        if prop.enum and prop.name:
+                            enum_names.add(prop.name.replace(".", ""))
+                        elif prop.ref and prop.ref in self.context.schemas:
+                            ref_schema = self.context.schemas[prop.ref]
+                            if ref_schema.enum:
+                                enum_names.add(prop.ref.replace(".", ""))
+
+        # Request models
+        for name, schema in schemas.items():
+            if schema.is_request_model and name not in seen_schemas:
+                schema_codes.append(self.generate_schema(schema))
+                seen_schemas.add(name)
+                # Collect enums from properties
+                if schema.properties:
+                    for prop in schema.properties.values():
+                        if prop.enum and prop.name:
+                            enum_names.add(prop.name.replace(".", ""))
+                        elif prop.ref and prop.ref in self.context.schemas:
+                            ref_schema = self.context.schemas[prop.ref]
+                            if ref_schema.enum:
+                                enum_names.add(prop.ref.replace(".", ""))
+
+        # Patch models
+        for name, schema in schemas.items():
+            if schema.is_patch_model and name not in seen_schemas:
+                schema_codes.append(self.generate_schema(schema))
+                seen_schemas.add(name)
+                # Collect enums from properties
+                if schema.properties:
+                    for prop in schema.properties.values():
+                        if prop.enum and prop.name:
+                            enum_names.add(prop.name.replace(".", ""))
+                        elif prop.ref and prop.ref in self.context.schemas:
+                            ref_schema = self.context.schemas[prop.ref]
+                            if ref_schema.enum:
+                                enum_names.add(prop.ref.replace(".", ""))
+
+        template = self.jinja_env.get_template('models/app_models.py.jinja')
+        content = template.render(
+            has_enums=len(enum_names) > 0,
+            enum_names=sorted(enum_names),  # Sort for consistent output
+            schemas=schema_codes
+        )
+
+        folder_name = self.base.tag_and_app_to_folder_name(tag, operations)
+        return GeneratedFile(
+            path=f"{folder_name}/models.py",
+            content=content,
+            description=f"Models for {tag}",
+        )