navi/backend/services/navi_geo/app.py

"""navi-geo Flask application factory + gunicorn entry.

Gunicorn entry:
    gunicorn 'services.navi_geo.app:create_app()' --bind 127.0.0.1:8426 --workers 2
"""
import time

from flask import Flask

from shared.git_sha import git_short_sha

from . import geo_route, admin
from . import geocode as geocode_mod
from . import netsyms
from . import address_book


def create_app():
    app = Flask(__name__)

    app.config['VERSION'] = git_short_sha()
    app.config['METRICS'] = {
        'start_time': time.time(),
        'request_count': 0,
        'last_error_at': None,
    }

    # Fresh per-worker (and per-test) state, so each picks up current env.
    geo_route.reset_cache()
    netsyms.reset_conn()
    address_book.reset_cache()
    geocode_mod.setup_trace_logger()   # re-read NAVI_GEO_RERANK_TRACE_LOG

    @app.before_request
    def _count_request():
        app.config['METRICS']['request_count'] += 1

    @app.after_request
    def _track_errors(response):
        if response.status_code >= 500:
            app.config['METRICS']['last_error_at'] = time.strftime(
                '%Y-%m-%dT%H:%M:%SZ', time.gmtime()
            )
        return response

    app.register_blueprint(geo_route.bp)
    app.register_blueprint(admin.bp)
    return app