Set URL and information in local swagger.json

This commit is contained in:
Joshua Boniface 2019-12-23 21:09:58 -05:00
parent 60ffe534a1
commit 398fb2dfe9
1 changed files with 13 additions and 3 deletions

View File

@ -40,6 +40,8 @@ from celery import Celery
import api_lib.pvcapi_helper as api_helper import api_lib.pvcapi_helper as api_helper
import api_lib.pvcapi_provisioner as api_provisioner import api_lib.pvcapi_provisioner as api_provisioner
API_VERSION = 1.0
# Parse the configuration file # Parse the configuration file
try: try:
pvc_config_file = os.environ['PVC_CONFIG_FILE'] pvc_config_file = os.environ['PVC_CONFIG_FILE']
@ -197,15 +199,23 @@ class API_Root(Resource):
message: message:
type: string type: string
description: A text message description: A text message
example: "PVC API version 1" example: "PVC API version 1.0"
""" """
return {"message":"PVC API version 1"} return { "message": "PVC API version {}".format(API_VERSION) }
api.add_resource(API_Root, '/') api.add_resource(API_Root, '/')
# /doc # /doc
class API_Doc(Resource): class API_Doc(Resource):
def get(self): def get(self):
return swagger(app) if config['ssl_enabled']:
scheme = 'https'
else:
scheme = 'http'
swagger_data = swagger(pvc_api.app)
swagger_data['info']['version'] = API_VERSION
swagger_data['info']['title'] = "PVC Client and Provisioner API"
swagger_data['info']['url'] = "{}://{}:{}".format(scheme, config['listen_address'], config['listen_port'])
return swagger_data
api.add_resource(API_Doc, '/doc') api.add_resource(API_Doc, '/doc')
# /login # /login