45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
"""
|
|
Generator - Stack-specific code generators for modelgen.
|
|
|
|
Supported generators:
|
|
- PydanticGenerator: Pydantic BaseModel classes
|
|
- DjangoGenerator: Django ORM models
|
|
- TypeScriptGenerator: TypeScript interfaces
|
|
- ProtobufGenerator: Protocol Buffer definitions
|
|
- PrismaGenerator: Prisma schema
|
|
- GrapheneGenerator: Graphene ObjectType/InputObjectType classes
|
|
"""
|
|
|
|
from typing import Dict, Type
|
|
|
|
from .base import BaseGenerator
|
|
from .django import DjangoGenerator
|
|
from .graphene import GrapheneGenerator
|
|
from .prisma import PrismaGenerator
|
|
from .protobuf import ProtobufGenerator
|
|
from .pydantic import PydanticGenerator
|
|
from .typescript import TypeScriptGenerator
|
|
|
|
# Registry of available generators
|
|
GENERATORS: Dict[str, Type[BaseGenerator]] = {
|
|
"pydantic": PydanticGenerator,
|
|
"django": DjangoGenerator,
|
|
"typescript": TypeScriptGenerator,
|
|
"ts": TypeScriptGenerator, # Alias
|
|
"protobuf": ProtobufGenerator,
|
|
"proto": ProtobufGenerator, # Alias
|
|
"prisma": PrismaGenerator,
|
|
"graphene": GrapheneGenerator,
|
|
}
|
|
|
|
__all__ = [
|
|
"BaseGenerator",
|
|
"PydanticGenerator",
|
|
"DjangoGenerator",
|
|
"GrapheneGenerator",
|
|
"TypeScriptGenerator",
|
|
"ProtobufGenerator",
|
|
"PrismaGenerator",
|
|
"GENERATORS",
|
|
]
|