16 lines
607 B
Python
16 lines
607 B
Python
from rest_framework.schemas import SchemaGenerator
|
|
|
|
from . import openapi
|
|
|
|
|
|
class OpenAPISchemaGenerator(SchemaGenerator):
|
|
def __init__(self, info, version, url=None, patterns=None, urlconf=None):
|
|
super(OpenAPISchemaGenerator, self).__init__(info.title, url, info.description, patterns, urlconf)
|
|
self.info = info
|
|
self.version = version
|
|
|
|
def get_schema(self, request=None, public=False):
|
|
document = super(OpenAPISchemaGenerator, self).get_schema(request, public)
|
|
swagger = openapi.Swagger.from_coreapi(document, self.info, self.version)
|
|
return swagger
|