app.py

import app_config
import app_utils
import datetime
import logging
import static

from app_utils import comma_filter, percent_filter, open_db, close_db, never_cache_preview
from flask import Flask, make_response, render_template
from flask_admin import Admin
from flask_admin.contrib.peewee import ModelView
from models import models
from render_utils import make_context, smarty_filter, urlencode_filter
from werkzeug.debug import DebuggedApplication

app = Flask(__name__)
app.debug = app_config.DEBUG
secrets = app_config.get_secrets()
app.secret_key = secrets.get('FLASK_SECRET_KEY')

app.add_template_filter(comma_filter, name='comma')
app.add_template_filter(percent_filter, name='percent')

try:
    file_handler = logging.FileHandler('%s/admin_app.log' % app_config.SERVER_LOG_PATH)
    file_handler.setLevel(logging.INFO)
    app.logger.addHandler(file_handler)
except IOError:
    print('Could not open %s/admin_app.log, skipping file-based logging' % app_config.SERVER_LOG_PATH)

app.logger.setLevel(logging.INFO)

app.register_blueprint(static.static, url_prefix='/%s' % app_config.PROJECT_SLUG)

app.add_template_filter(smarty_filter, name='smarty')
app.add_template_filter(urlencode_filter, name='urlencode')

admin = Admin(app, url='/%s/admin' % app_config.PROJECT_SLUG)
admin.add_view(ModelView(models.Result))
admin.add_view(ModelView(models.Call))

SLUG_TO_OFFICENAME = {
    'senate': 'U.S. Senate',
    'house': 'U.S. House',
    'president': 'President',
    'governor': 'Governor'
}

# Example application views
@app.route('/%s/calls/<office>/' % app_config.PROJECT_SLUG, methods=['GET'])
def calls_admin(office):
    officename = SLUG_TO_OFFICENAME[office]
    results = app_utils.filter_results(officename)
    grouped = app_utils.group_results_by_race(results, officename)

    context = make_context(asset_depth=1)
    context.update({
        'races': grouped
    })

    return make_response(render_template('calls.html', **context))

@app.route('/%s/calls/<office>/call-npr' % app_config.PROJECT_SLUG, methods=['POST'])
def call_npr(office):
    from flask import request

    result_id = request.form.get('result_id')

    result = models.Result.get(models.Result.id == result_id)
    call = result.call[0]
    if call.override_winner:
        call.override_winner = False
    else:
        call.override_winner = True

    call.save()

    race_id = result.raceid
    statepostal = result.statepostal
    officename = result.officename
    level = result.level
    reportingunitname = result.reportingunitname

    race_results = models.Result.select().where(
        models.Result.level == level,
        models.Result.raceid == race_id,
        models.Result.officename == officename,
        models.Result.statepostal == statepostal,
        models.Result.reportingunitname == reportingunitname
    )

    for race_result in race_results:
        race_call = race_result.call[0]
        if call.override_winner:
            race_call.accept_ap = False

        if race_call.call_id != call.call_id:
            race_call.override_winner = False

        race_call.save()

    return 'Success', 200

@app.route('/%s/calls/<office>/accept-ap' % app_config.PROJECT_SLUG, methods=['POST'])
def accept_ap(office):
    from flask import request

    officename = SLUG_TO_OFFICENAME[office]

    race_id = request.form.get('race_id')
    statepostal = request.form.get('statepostal')
    reportingunit = request.form.get('reportingunit')
    level = request.form.get('level')

    if level == 'district':
        results = models.Result.select().where(
            models.Result.level == 'district',
            models.Result.raceid == race_id,
            models.Result.officename == officename,
            models.Result.statepostal == statepostal,
            models.Result.reportingunitname == reportingunit
        )
    else:
        results = models.Result.select().where(
            (models.Result.level == 'state') | (models.Result.level == 'national'),
            models.Result.raceid == race_id,
            models.Result.officename == officename,
            models.Result.statepostal == statepostal,
        )

    for result in results:
        call = result.call[0]
        if call.accept_ap:
            call.accept_ap = False
        else:
            call.accept_ap = True
        call.save()

    return 'Success', 200


@app.route('/%s/test/' % app_config.PROJECT_SLUG, methods=['GET'])
def _test_app():
    """
    Test route for verifying the application is running.
    """
    app.logger.info('Test URL requested.')

    return make_response(datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'))

# Example of rendering index.html with admin_app
@app.route ('/%s/' % app_config.PROJECT_SLUG, methods=['GET'])
def index():
    """
    Example view rendering a simple page.
    """
    context = make_context(asset_depth=1)

    return make_response(render_template('index.html', **context))


app.before_request(open_db)
app.after_request(close_db)
app.after_request(never_cache_preview)

# Enable Werkzeug debug pages
if app_config.DEBUG:
    wsgi_app = DebuggedApplication(app, evalex=False)
else:
    wsgi_app = app