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

Repository navigation

""" A Reference is ultimately a Class which will be in models, usually defined in a body input or return type """

from dataclasses import dataclass

from typing import Dict

from .. import utils

class_overrides: Dict[str, "Reference"] = {}

@dataclass

class Reference:

""" A reference to a class which will be in models """

class_name: str

module_name: str

@staticmethod

def from_ref(ref: str) -> "Reference":

""" Get a Reference from the openapi #/schemas/blahblah string """

ref_value = ref.split("/")[-1]

class_name = utils.pascal_case(ref_value)

if class_name in class_overrides:

return class_overrides[class_name]

return Reference(class_name=class_name, module_name=utils.snake_case(class_name))