RIC-1059: dms_cli to use flask-restx
[ric-plt/appmgr.git] / xapp_orchestrater / dev / xapp_onboarder / xapp_onboarder / api / models / response_models.py
1 ################################################################################
2 #   Copyright (c) 2020 AT&T Intellectual Property.                             #
3 #                                                                              #
4 #   Licensed under the Apache License, Version 2.0 (the "License");            #
5 #   you may not use this file except in compliance with the License.           #
6 #   You may obtain a copy of the License at                                    #
7 #                                                                              #
8 #       http://www.apache.org/licenses/LICENSE-2.0                             #
9 #                                                                              #
10 #   Unless required by applicable law or agreed to in writing, software        #
11 #   distributed under the License is distributed on an "AS IS" BASIS,          #
12 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   #
13 #   See the License for the specific language governing permissions and        #
14 #   limitations under the License.                                             #
15 ################################################################################
16
17 from flask_restx import fields, marshal
18 from xapp_onboarder.api.api_reference import api
19
20 error_message_model = api.model('error_message', {
21     'error_source': fields.String(description='source of the error', required=True),
22     'error_message': fields.String(description='source of the error', required=True),
23     'status': fields.String(description='http response message', required=True),
24 })
25
26 status_message_model = api.model('status', {
27     'status': fields.String(description='status of the service', required=True)
28 })
29
30
31 class response():
32
33     def __init__(self, model, status_code, status = "" , error_source = "", error_message = ""):
34         self.model = model
35         self.status = status
36         self.status_code = status_code
37         self.error_source = error_source
38         self.error_message = error_message
39
40     def get_return(self):
41         return marshal(self, self.model), self.status_code