tic/server/tic_server.py

   1 import collections
   2 from flask import Flask
   3 from flask import Response
   4 from flask.globals import request
   5 from flask.helpers import url_for
   6 import json
   7 import os
   8 import logging
   9 from tic import command
  10 from tic.utils import error
  11 from flask_cors import CORS
  12
  13 app = Flask(__name__)
  14 CORS(app)
  15
  16 @app.route('/')
  17 def index():
  18     return '''
  19         <title>TIC-Core web server</title>
  20         <h1> TIC-Core web server is working<h1>
  21     '''
  22
  23 @app.route('/analysis', methods=['POST'])
  24 def analysis():
  25     try:
  26         logger = logging.getLogger(__name__)
  27         logger.info('%s - %s %s : data=%s' % (request.remote_addr, request.method, request.path, request.data))
  28         repo_info = json.loads(request.data)
  29         view_data = command.analyze(repo_info.get('repos'), repo_info.get('recipes'))
  30         resp = makeresponse(view_data, None)
  31     except error.TICError as err:
  32         logger.error(err)
  33         resp = makeresponse(str(err), err)
  34     except ValueError as ve:
  35         logger.error(ve)
  36         resp = makeresponse(str(ve), ve)
  37     except Exception as ex:
  38         logger.error(ex)
  39         resp = makeresponse(str(ex), ex)
  40
  41     return resp
  42
  43 @app.route('/exports', methods=['POST'])
  44 def exports():
  45     try:
  46         logger = logging.getLogger(__name__)
  47         logger.info('%s - %s %s : data=%s' % (request.remote_addr, request.method, request.path, request.data))
  48
  49     except error.TICError as err:
  50         logger.error(err)
  51         resp = makeresponse(str(err), err)
  52     except ValueError as ve:
  53         logger.error(ve)
  54         resp = makeresponse(str(ve), ve)
  55     except Exception as ex:
  56         logger.error(ex)
  57         resp = makeresponse(str(ex), ex)
  58
  59     return resp
  60
  61
  62 def start(port_num=59001):
  63     # cryptographic random generator
  64     app.secret_key = os.urandom(24)
  65
  66     with app.test_request_context():
  67         print(url_for('index'))
  68         print(url_for('analysis'))
  69     app.run(host='0.0.0.0', port=port_num)
  70
  71
  72 def makeresponse(data, err):
  73     status = 200
  74     if err:
  75         if isinstance(err, error.TICError) or isinstance(err, ValueError):
  76             status = 400 # Bad Request
  77         elif isinstance(err, Exception):
  78             status = 500 # Internal Server Error
  79         res_body = json.dumps(ResultInfo('false', None, data)._asdict())
  80     else:
  81         res_body = json.dumps(ResultInfo('true', data, None)._asdict())
  82
  83     resp = Response(status=status,
  84                     mimetype='application/json',
  85                     response=res_body)
  86     return resp
  87
  88
  89 ResultInfoType = collections.namedtuple('ResultInfo', 'result, data, message')
  90 def ResultInfo(result='false', data=None, message=None):
  91     return ResultInfoType(result, data, message)
  92
  93 if __name__ == '__main__':
  94     start(59003)