Skip to content

Instantly share code, notes, and snippets.

What would you like to do?
A python decorator that adds query parameters to django-rest-framework swagger docstring.
# -*- coding: utf-8 -*-
import collections
import yaml
from rest_framework import fields
Convert rest_framework.fields classes to Swagger data types according to
Return 'string' by default.
field_to_yaml_type = collections.defaultdict(
lambda: 'string',
fields.IntegerField: 'integer',
fields.FloatField: 'number',
fields.BooleanField: 'boolean',
fields.NullBooleanField: 'boolean',
def add_query_parameters(serializer_class):
def decorator(method):
def wrapper(*args, **kwargs):
return method(*args, **kwargs)
from rest_framework_swagger.introspectors import IntrospectorHelper, BaseViewIntrospector
i = BaseViewIntrospector(None, None, None, None)
parser = i.get_yaml_parser()
plain_text_doc = IntrospectorHelper.strip_yaml_from_docstring(method.__doc__)
yaml_object = parser.load_obj_from_docstring(method.__doc__) or {}
if 'parameters' in yaml_object:
if not isinstance(yaml_object['parameters'], list):
raise ValueError('paramters in YAML docstring should be a list instance')
yaml_object['parameters'] = list()
serializer = serializer_class()
for field_name, field in serializer.fields.items():
parameter = {
'name': field_name,
'description': field.help_text,
'type': field_to_yaml_type[type(field)],
'required': field.required,
'paramType': 'query',
wrapper.__doc__ = '\n'.join([plain_text_doc,
yaml.dump(yaml_object, default_flow_style=False)])
return wrapper
return decorator
class ListBansSerializer(serializers.Serializer):
limit = serializers.IntegerField(default=10, help_text='query limit')
offset = serializers.IntegerField(default=0, help_text='query offset')
class ListBans(BaseBanView):
def get(self, request):
List all profile bans
response_serializer: backend_serializers.BanSerializer
serializer = moderator_serializers.ListBansSerializer(data=request.query_params)
if serializer.is_valid(raise_exception=True):
# query profile bans
data = []
return APIResponse(status=status.HTTP_200_OK, data=data)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment