Skip to content

inspectors#

URLReferencedModelFieldInspector#

Bases: ReferencingSerializerInspector

Generate swagger for URLReferencedModelField from its child serializer.

Source code in src/apps/common/inspectors.py 
22
23
24
25
26
27
28
29
30
class URLReferencedModelFieldInspector(ReferencingSerializerInspector):
    """Generate swagger for URLReferencedModelField from its child serializer."""

    def field_to_swagger_object(self, field, swagger_object_type, use_references, **kwargs):
        if isinstance(field, URLReferencedModelField):
            return super().field_to_swagger_object(
                field.child, swagger_object_type, use_references, **kwargs
            )
        return NotHandled

SwaggerDescriptionInspector#

Bases: FieldInspector

Inspector that reads description from Serializer.Meta.swagger_description.

Source code in src/apps/common/inspectors.py 
33
34
35
36
37
38
39
40
41
42
43
44
class SwaggerDescriptionInspector(FieldInspector):
    """Inspector that reads description from Serializer.Meta.swagger_description."""

    def process_result(self, result, method_name, obj, **kwargs):
        if isinstance(result, openapi.Schema.OR_REF):
            # traverse any references and alter the Schema object in place
            schema = openapi.resolve_ref(result, self.components)
            if meta := getattr(obj, "Meta", None):
                if description := getattr(meta, "swagger_description", None):
                    schema["description"] = description

        return result

ExtendedSwaggerAutoSchema#

Bases: SwaggerAutoSchema

Class that generates the swagger documentation for views.

Source code in src/apps/common/inspectors.py 
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
class ExtendedSwaggerAutoSchema(SwaggerAutoSchema):
    """Class that generates the swagger documentation for views."""

    def get_tags(self, operation_keys=None):
        """Return tag for operation, used for grouping endpoints in Swagger."""
        operation_keys = operation_keys or self.operation_keys

        tags = self.overrides.get("tags")
        if not tags:
            if operation_keys[0] == "v3":
                # Group e.g. /v3/datasets in "datasets" instead of having everything in "v3"
                tags = [operation_keys[1]]
            else:
                tags = [operation_keys[0]]

        return tags

    def get_query_parameters(self) -> List[openapi.Parameter]:
        """Return the query parameters accepted by this view.

        Extended to allow listing query parameter serializers
        in `ViewSet.get_query_serializer_classes` method.
        """
        parameters = super().get_query_parameters()

        getter = getattr(self.view, "get_query_serializer_classes", lambda: [])
        for serializer in getter():
            serializer = force_serializer_instance(serializer)
            serializer_parameters = self.serializer_to_parameters(serializer, in_=openapi.IN_QUERY)

            old_params = set(param_list_to_odict(parameters))
            new_params = set(param_list_to_odict(serializer_parameters))

            if conflicts := old_params & new_params:
                raise SwaggerGenerationError(
                    f"your query serializer {serializer.__class__} contains fields that conflict with the "
                    f"filter_backend or paginator_class or another query serializer "
                    f"on the view - {self.method} {self.path} {conflicts=}"
                )
            parameters += serializer_parameters
        return parameters

get_query_parameters() #

Return the query parameters accepted by this view.

Extended to allow listing query parameter serializers in ViewSet.get_query_serializer_classes method.

Source code in src/apps/common/inspectors.py 
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
def get_query_parameters(self) -> List[openapi.Parameter]:
    """Return the query parameters accepted by this view.

    Extended to allow listing query parameter serializers
    in `ViewSet.get_query_serializer_classes` method.
    """
    parameters = super().get_query_parameters()

    getter = getattr(self.view, "get_query_serializer_classes", lambda: [])
    for serializer in getter():
        serializer = force_serializer_instance(serializer)
        serializer_parameters = self.serializer_to_parameters(serializer, in_=openapi.IN_QUERY)

        old_params = set(param_list_to_odict(parameters))
        new_params = set(param_list_to_odict(serializer_parameters))

        if conflicts := old_params & new_params:
            raise SwaggerGenerationError(
                f"your query serializer {serializer.__class__} contains fields that conflict with the "
                f"filter_backend or paginator_class or another query serializer "
                f"on the view - {self.method} {self.path} {conflicts=}"
            )
        parameters += serializer_parameters
    return parameters

get_tags(operation_keys=None) #

Return tag for operation, used for grouping endpoints in Swagger.

Source code in src/apps/common/inspectors.py 
50
51
52
53
54
55
56
57
58
59
60
61
62
def get_tags(self, operation_keys=None):
    """Return tag for operation, used for grouping endpoints in Swagger."""
    operation_keys = operation_keys or self.operation_keys

    tags = self.overrides.get("tags")
    if not tags:
        if operation_keys[0] == "v3":
            # Group e.g. /v3/datasets in "datasets" instead of having everything in "v3"
            tags = [operation_keys[1]]
        else:
            tags = [operation_keys[0]]

    return tags