forked from openapi-generators/openapi-python-client
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconfig.py
More file actions
35 lines (24 loc) · 1.01 KB
/
config.py
File metadata and controls
35 lines (24 loc) · 1.01 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
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]
def load_config(self) -> None:
""" Loads config from provided Path """
if self.class_overrides is not None:
from .parser import reference
for class_name, class_data in self.class_overrides.items():
reference.class_overrides[class_name] = reference.Reference(**dict(class_data))
from openapi_python_client import Project
Project.project_name_override = self.project_name_override
Project.package_name_override = self.package_name_override
@staticmethod
def load_from_path(path: Path) -> None:
config_data = yaml.safe_load(path.read_text())
Config(**config_data).load_config()