summaryrefslogtreecommitdiffstats
path: root/ice-server/heat_test/app.py
blob: e982c6852006de6c3ac3800cd89be9ba1a2fa452 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#!flask/bin/python
import argparse
import logging
import requests

import connexion
from connexion.resolver import RestyResolver


def parse_args():
    """
    parse argument parameters
    """
    parser = argparse.ArgumentParser(description='start the heat validation rest server')
    parser.add_argument("--debug", help="increase output verbosity")
    parser.add_argument("-p", "--port", type=int, help="listen port (default 5000)", default=5000)
    args = parser.parse_args()
    if args.debug:
        logging.info("debug mode")
        debug = True
    else:
        debug = False
    if args.port:
        port = args.port
    else:
        port = 5000


def create_app():
    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
    app = connexion.App(__name__, specification_dir='swagger/')

    # load default config
    app.app.config.from_object("default_settings")
    app.app.config.from_envvar('ICE_SETTINGS', silent=True)
    app.add_api('ice_api.yaml', swagger_ui=app.app.config['DEBUG'], resolver=RestyResolver('ice_validator'))

    return app

def create_test_app():
    print("create_test_app")
    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
    app = connexion.App(__name__, specification_dir='swagger/')
    app.add_api('ice_api.yaml', swagger_ui=app.app.config['DEBUG'], resolver=RestyResolver('ice_validator'))
    app.app.testing = True
    return app


def start_app(app):
    logging.info("######################################")
    logging.info("starting server")
    logging.info("######################################")
    app.run(port=app.app.config['ICE_PORT'], debug=app.app.config['DEBUG'])

    # register service in MSB
    try:
        msb_url = app.app.config['MSB_URL']
        requests.get(msb_url)
        logging.info("registration to %s done", msb_url)
    except:
        logging.info("registration to %s failed", msb_url)


if __name__ == '__main__':
    start_app(create_app())