diff options
-rw-r--r-- | res/res/settings.py | 1 | ||||
-rw-r--r-- | res/res/swagger/management/__init__.py | 13 | ||||
-rw-r--r-- | res/res/swagger/management/commands/__init__.py | 13 | ||||
-rw-r--r-- | res/res/swagger/management/commands/export_swagger.py | 36 |
4 files changed, 63 insertions, 0 deletions
diff --git a/res/res/settings.py b/res/res/settings.py index ceda0e4..98ec46a 100644 --- a/res/res/settings.py +++ b/res/res/settings.py @@ -48,6 +48,7 @@ INSTALLED_APPS = [ 'rest_framework', 'res.pub.database', 'res.samples', + 'res.swagger', 'res.resources', 'drf_yasg', ] diff --git a/res/res/swagger/management/__init__.py b/res/res/swagger/management/__init__.py new file mode 100644 index 0000000..342c2a8 --- /dev/null +++ b/res/res/swagger/management/__init__.py @@ -0,0 +1,13 @@ +# Copyright 2018 ZTE Corporation. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/res/res/swagger/management/commands/__init__.py b/res/res/swagger/management/commands/__init__.py new file mode 100644 index 0000000..342c2a8 --- /dev/null +++ b/res/res/swagger/management/commands/__init__.py @@ -0,0 +1,13 @@ +# Copyright 2018 ZTE Corporation. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/res/res/swagger/management/commands/export_swagger.py b/res/res/swagger/management/commands/export_swagger.py new file mode 100644 index 0000000..8b5274a --- /dev/null +++ b/res/res/swagger/management/commands/export_swagger.py @@ -0,0 +1,36 @@ +# Copyright 2018 ZTE Corporation. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +import json + +from django.core.management.base import BaseCommand +from django.test import Client + + +class Command(BaseCommand): + def add_arguments(self, parser): + parser.add_argument( + '-f', + '--name', + action='store', + dest='name', + default='swagger.json', + help='name of swagger file.', + ) + + def handle(self, *args, **options): + self.client = Client() + response = self.client.get("/api/vnfres/v1/swagger.json") + with open(options['name'], 'w') as swagger_file: + swagger_file.write(json.dumps(response.data)) + print "swagger api is written to %s" % options['name'] |