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