mirror of
https://github.com/Telecominfraproject/oopt-gnpy-api.git
synced 2025-10-31 01:47:49 +00:00
75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
'''
|
|
gnpy.tools.rest_example
|
|
=======================
|
|
|
|
GNPy as a rest API example
|
|
'''
|
|
|
|
import logging
|
|
from logging.handlers import RotatingFileHandler
|
|
|
|
import werkzeug
|
|
from flask_injector import FlaskInjector
|
|
|
|
from gnpyapi.core import app
|
|
from gnpyapi.core.exception.equipment_error import EquipmentError
|
|
from gnpyapi.core.exception.exception_handler import bad_request_handler, common_error_handler
|
|
from gnpyapi.core.exception.path_computation_error import PathComputationError
|
|
from gnpyapi.core.exception.topology_error import TopologyError
|
|
import argparse
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _init_logger():
|
|
handler = RotatingFileHandler('api.log', maxBytes=1024 * 1024, backupCount=5, encoding='utf-8')
|
|
ch = logging.StreamHandler()
|
|
logging.basicConfig(level=logging.INFO, handlers=[handler, ch],
|
|
format="%(asctime)s %(levelname)s %(name)s(%(lineno)s) [%(threadName)s - %(thread)d] - %("
|
|
"message)s")
|
|
|
|
|
|
def _init_app():
|
|
app.register_error_handler(KeyError, bad_request_handler)
|
|
app.register_error_handler(TypeError, bad_request_handler)
|
|
app.register_error_handler(ValueError, bad_request_handler)
|
|
# app.register_error_handler(exceptions.ConfigurationError, bad_request_handler)
|
|
# app.register_error_handler(exceptions.DisjunctionError, bad_request_handler)
|
|
# app.register_error_handler(exceptions.EquipmentConfigError, bad_request_handler)
|
|
# app.register_error_handler(exceptions.NetworkTopologyError, bad_request_handler)
|
|
# app.register_error_handler(exceptions.ServiceError, bad_request_handler)
|
|
# app.register_error_handler(exceptions.SpectrumError, bad_request_handler)
|
|
# app.register_error_handler(exceptions.ParametersError, bad_request_handler)
|
|
app.register_error_handler(AssertionError, bad_request_handler)
|
|
# app.register_error_handler(InternalServerError, common_error_handler)
|
|
app.register_error_handler(TopologyError, bad_request_handler)
|
|
app.register_error_handler(EquipmentError, bad_request_handler)
|
|
|
|
app.register_error_handler(PathComputationError, bad_request_handler)
|
|
for error_code in werkzeug.exceptions.default_exceptions:
|
|
app.register_error_handler(error_code, common_error_handler)
|
|
|
|
|
|
def main(http: bool = False):
|
|
_init_logger()
|
|
_init_app()
|
|
FlaskInjector(app=app)
|
|
|
|
if http:
|
|
app.run(host='0.0.0.0', port=8080)
|
|
else:
|
|
app.run(host='0.0.0.0', port=8080, ssl_context='adhoc')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description="Rest API example")
|
|
|
|
parser.add_argument("--http", action="store_true", help="run server with http instead of https")
|
|
|
|
args = parser.parse_args()
|
|
|
|
main(http=args.http)
|