Skip to content

custom_visitor

fastagency.api.openapi.security_schema_visitor.custom_visitor #

custom_visitor(
    parser: OpenAPIParser, model_path: Path
) -> dict[str, object]
Source code in fastagency/api/openapi/security_schema_visitor.py 
 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
def custom_visitor(parser: OpenAPIParser, model_path: Path) -> dict[str, object]:
    if "securitySchemes" not in parser.raw_obj["components"]:
        return {}
    security_schemes = parser.raw_obj["components"]["securitySchemes"]
    server_url = parser.raw_obj["servers"][0]["url"]

    security_classes = []
    security_parameters = {}
    for k, v in security_schemes.items():
        v["server_url"] = server_url
        security_class = BaseSecurity.get_security_class(
            type=v["type"], schema_parameters=v
        )

        security_classes.append(security_class.__name__)

        security_parameters[k] = security_class.get_security_parameters(
            schema_parameters=v
        )

    return {
        "security_schemes": security_schemes,
        "security_classes": security_classes,
        "security_parameters": security_parameters,
    }