From ccdca64bafb07057dd482cf04da7bf9e64dc1928 Mon Sep 17 00:00:00 2001 From: chenjiankun Date: Thu, 1 Dec 2016 01:51:16 +0000 Subject: Add swagger support for Rest API JIRA: YARDSTICK-439 Change-Id: I36ad0663455c51d635c4329f5cbb9da25d8042e1 Signed-off-by: chenjiankun --- api/views.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'api/views.py') diff --git a/api/views.py b/api/views.py index f899251dd..928d8e9eb 100644 --- a/api/views.py +++ b/api/views.py @@ -7,11 +7,14 @@ # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## import logging +import os from flask import request from flask_restful import Resource +from flasgger.utils import swag_from from api.utils import common as common_utils +from api.swagger import models from api.actions import test as test_action from api.actions import samples as samples_action from api.actions import result as result_action @@ -20,7 +23,14 @@ from api.actions import env as env_action logger = logging.getLogger(__name__) +TestCaseActionModel = models.TestCaseActionModel +TestCaseActionArgsModel = models.TestCaseActionArgsModel +TestCaseActionArgsOptsModel = models.TestCaseActionArgsOptsModel +TestCaseActionArgsOptsTaskArgModel = models.TestCaseActionArgsOptsTaskArgModel + + class Release(Resource): + @swag_from(os.getcwd() + '/swagger/docs/testcases.yaml') def post(self): action = common_utils.translate_to_str(request.json.get('action', '')) args = common_utils.translate_to_str(request.json.get('args', {})) @@ -44,7 +54,11 @@ class Samples(Resource): return common_utils.error_handler('Wrong action') +ResultModel = models.ResultModel + + class Results(Resource): + @swag_from(os.getcwd() + '/swagger/docs/results.yaml') def get(self): args = common_utils.translate_to_str(request.args) action = args.get('action', '') -- cgit 1.2.3-korg