feat(schema.py): add ClassCodeDetails and CallableCodeDetails data classes to store details about classes and callables respectively

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-26 07:46:31 -03:00
commit c4a2c1c80a

View file

@ -0,0 +1,29 @@
from pydantic import BaseModel, Field
from typing import Optional
class ClassCodeDetails(BaseModel):
"""
A dataclass for storing details about a class.
"""
name: str
doc: Optional[str]
bases: list
attributes: list
methods: list
init: Optional[dict] = Field(default_factory=dict)
class CallableCodeDetails(BaseModel):
"""
A dataclass for storing details about a callable.
"""
name: str
doc: Optional[str]
args: list
body: list
return_type: Optional[str]