from pathlib import Path
from typing import Dict, Optional
import yaml
from pydantic import BaseModel
class ClassOverride(BaseModel):
class_name: str
module_name: str
class Config(BaseModel):
class_overrides: Optional[Dict[str, ClassOverride]]
project_name_override: Optional[str]
package_name_override: Optional[str]
package_version_override: Optional[str]
field_prefix: Optional[str]
def load_config(self) -> None:
""" Sets globals based on Config """
from openapi_python_client import Project
from . import utils
from .parser import reference
if self.class_overrides is not None:
for class_name, class_data in self.class_overrides.items():
reference.class_overrides[class_name] = reference.Reference(**dict(class_data))
Project.project_name_override = self.project_name_override
Project.package_name_override = self.package_name_override
Project.package_version_override = self.package_version_override
if self.field_prefix is not None:
utils.FIELD_PREFIX = self.field_prefix
@staticmethod
def load_from_path(path: Path) -> None:
""" Creates a Config from provided JSON or YAML file and sets a bunch of globals from it """
config_data = yaml.safe_load(path.read_text())
Config(**config_data).load_config()