def __init__(
self, input_name: str, input_text: str, model_path: Optional[str] = None
) -> None:
self.input_name: str = input_name
self.input_text: str = input_text
if model_path:
model_path_var.set(model_path)
self.open_api_model_parser: OpenAPIModelParser = OpenAPIModelParser(source='')
def __init__(
self, input_name: str, input_text: str, model_path: Optional[str] = None
) -> None:
self.input_name: str = input_name
self.input_text: str = input_text
self.openapi_model_parser: OpenAPIModelParser = OpenAPIModelParser(source='')
if model_path:
self.openapi_model_parser.current_source_path = '.models'
def __init__(
self,
input_name: str,
input_text: str,
openapi_model_parser: Optional[OpenAPIModelParser] = None,
) -> None:
self.input_name: str = input_name
self.input_text: str = input_text
self.openapi_model_parser: OpenAPIModelParser = openapi_model_parser or OpenAPIModelParser(
source='')
def generate_code(input_name: str, input_text: str, output_dir: Path,
template_dir: Optional[Path]) -> None:
if not output_dir.exists():
output_dir.mkdir(parents=True)
if not template_dir:
template_dir = BUILTIN_TEMPLATE_DIR
model_parser = OpenAPIModelParser(source=input_text, )
parser = OpenAPIParser(input_name,
input_text,
openapi_model_parser=model_parser)
parsed_object: ParsedObject = parser.parse()
environment: Environment = Environment(loader=FileSystemLoader(
template_dir if template_dir else f"{Path(__file__).parent}/template",
encoding="utf8",
), )
results: Dict[Path, str] = {}
code_formatter = CodeFormatter(PythonVersion.PY_38, Path().resolve())
for target in template_dir.rglob("*"):
relative_path = target.relative_to(template_dir)
result = environment.get_template(str(relative_path)).render(
operations=parsed_object.operations,
imports=parsed_object.imports,
info=parsed_object.info,
)
results[relative_path] = code_formatter.format_code(result)
timestamp = datetime.now(timezone.utc).replace(microsecond=0).isoformat()
header = f"""\
# generated by fastapi-codegen:
# filename: {Path(input_name).name}
# timestamp: {timestamp}"""
for path, code in results.items():
with output_dir.joinpath(path.with_suffix(".py")).open("wt") as file:
print(header, file=file)
print("", file=file)
print(code.rstrip(), file=file)
with chdir(output_dir):
results = model_parser.parse()
if not results:
return
elif isinstance(results, str):
output = output_dir / MODEL_PATH
modules = {output: (results, input_name)}
else:
raise Exception('Modular references are not supported in this version')
header = f'''\
# generated by fastapi-codegen:
# filename: {{filename}}'''
# if not disable_timestamp:
header += f'\n# timestamp: {timestamp}'
for path, body_and_filename in modules.items():
body, filename = body_and_filename
if path is None:
file = None
else:
if not path.parent.exists():
path.parent.mkdir(parents=True)
file = path.open('wt', encoding='utf8')
print(header.format(filename=filename), file=file)
if body:
print('', file=file)
print(body.rstrip(), file=file)
if file is not None:
file.close()