run.py

"""
The starter module for RCS.  Currently it contains most of the functional code
for RCS and this should eventually end up in separate modules or packages.
"""
from __future__ import division, print_function, unicode_literals

import json, jsonschema, config, os, sys, logging, flask
from services import db, v1, v2, utils

from logging.handlers import RotatingFileHandler
from flask import Flask
from flask.ext.restful import request

# FIXME clean this up
app = Flask(__name__)

app.config.from_object(config)
if os.environ.get('RCS_CONFIG'):
    app.config.from_envvar('RCS_CONFIG')
handler = RotatingFileHandler(app.config['LOG_FILE'],
                              maxBytes=app.config.get('LOG_ROTATE_BYTES', 200000),
                              backupCount=app.config.get('LOG_BACKUPS', 5))
handler.setFormatter(logging.Formatter(
    '%(asctime)s %(levelname)s: %(message)s '
    '[in %(pathname)s:%(lineno)d]'
))

loggers = [app.logger, logging.getLogger('regparse.sigcheck')]
for l in loggers:
    l.setLevel(app.config['LOG_LEVEL'])
    l.addHandler(handler)

flask.got_request_exception.connect(utils.log_exception, app)
if 'ACCESS_LOG' in app.config:
    acc_log = logging.getLogger('rcsaccesslog')
    acc_log.setLevel(logging.DEBUG)
    acc_handler = RotatingFileHandler(app.config['ACCESS_LOG'],
                                      maxBytes=app.config.get('LOG_ROTATE_BYTES', 200000),
                                      backupCount=app.config.get('LOG_BACKUPS', 5))
    acc_handler.setFormatter(logging.Formatter('%(asctime)s %(levelname)s: %(message)s '))
    acc_log.addHandler(acc_handler)

    def log_request(sender):
        acc_log.info('{ip} {method} {path} {agent}'.format(method=request.method,
                                                           path=request.path,
                                                           ip=request.remote_addr,
                                                           agent=request.user_agent.string))
        acc_log.debug(request.data)
    flask.request_started.connect(log_request, app)

    def log_response(sender, response):
        acc_log.info('{code} {text}'.format(code=response.status_code, text=response.status))
    flask.request_finished.connect(log_response, app)

schema_path = app.config['REG_SCHEMA']
if not os.path.exists(schema_path):
    schema_path = os.path.join(sys.prefix, schema_path)


@app.before_request
def before_request():
    flask.g.get_validator = lambda: jsonschema.validators.Draft4Validator(json.load(open(schema_path)))
    flask.g.proxies = {'http': app.config['HTTP_PROXY'], 'https': app.config['HTTP_PROXY']}


if app.config.get('DEBUG_ENDPOINTS'):
    @app.after_request
    def after_request(response):
        response.headers.add('RCS-Debug-Enabled', 'True')
        return response

if app.config.get('PROD'):
    if app.config.get('DEBUG_ENDPOINTS') or not app.config.get('SIG_CHECK'):
        raise Exception('PROD enabled while DEBUG_ENDPOINTS is True or SIG_CHECK is False.  I hope that was a typo.')

db.init_auth_db(app.config['DB_CONN'], app.config['AUTH_DB'])
db.init_doc_db(app.config['DB_CONN'], app.config['STORAGE_DB'])
# client[app.config['DB_NAME']].authenticate( app.config['DB_USER'], app.config['DB_PASS'] )

global_prefix = app.config.get('URL_PREFIX', '')
api_v1_bp = v1.make_blueprint()
app.register_blueprint(api_v1_bp, url_prefix=global_prefix + '/v1')

api_v2_bp = v2.make_blueprint(app)
app.register_blueprint(api_v2_bp, url_prefix=global_prefix + '/v2')

if __name__ == '__main__':
    for l in loggers:
        l.info('logger started')
    host = '127.0.0.1'
    if '--listen-all' in sys.argv[1:]:
        host = '0.0.0.0'
    app.run(debug=True, host=host)