From db606f00058fc9e114d0f0c0f67ae564df235a84 Mon Sep 17 00:00:00 2001 From: fujinhua Date: Fri, 3 Aug 2018 17:29:09 +0800 Subject: Add swagger export command for ztevnfmadaptor Change-Id: I8a5a54133a3aa9eac32e5a0a9ee339370dab02ea Issue-ID: VFC-1011 Signed-off-by: fujinhua --- zte/vmanager/driver/settings.py | 1 + zte/vmanager/driver/swagger/management/__init__.py | 13 ++++++++ .../driver/swagger/management/commands/__init__.py | 13 ++++++++ .../swagger/management/commands/export_swagger.py | 36 ++++++++++++++++++++++ 4 files changed, 63 insertions(+) create mode 100644 zte/vmanager/driver/swagger/management/__init__.py create mode 100644 zte/vmanager/driver/swagger/management/commands/__init__.py create mode 100644 zte/vmanager/driver/swagger/management/commands/export_swagger.py diff --git a/zte/vmanager/driver/settings.py b/zte/vmanager/driver/settings.py index 4f7e21c8..9c268a2c 100644 --- a/zte/vmanager/driver/settings.py +++ b/zte/vmanager/driver/settings.py @@ -46,6 +46,7 @@ INSTALLED_APPS = [ 'rest_framework', 'driver.pub.database', 'driver.interfaces', + 'driver.swagger', 'drf_yasg', ] diff --git a/zte/vmanager/driver/swagger/management/__init__.py b/zte/vmanager/driver/swagger/management/__init__.py new file mode 100644 index 00000000..342c2a8c --- /dev/null +++ b/zte/vmanager/driver/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/zte/vmanager/driver/swagger/management/commands/__init__.py b/zte/vmanager/driver/swagger/management/commands/__init__.py new file mode 100644 index 00000000..342c2a8c --- /dev/null +++ b/zte/vmanager/driver/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/zte/vmanager/driver/swagger/management/commands/export_swagger.py b/zte/vmanager/driver/swagger/management/commands/export_swagger.py new file mode 100644 index 00000000..4971022f --- /dev/null +++ b/zte/vmanager/driver/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/ztevnfmdriver/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'] -- cgit 1.2.3-korg