openapi-python-client/openapi_python_client/config.py at main · ramnes/openapi-python-client · GitHub

Repository navigation

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()