def check_database(request):
"""Check that the database is available for a simple query."""
data = _check_timed(lambda: ping_session(request.db_session))
if data["up"]:
current_heads = [
row[0] for row in request.db_session.execute(
"select version_num from alembic_version")
]
alembic_version = ",".join(sorted(current_heads))
else:
alembic_version = "unknown"
data["alembic_version"] = alembic_version
return data
def main(
ping_connections=False,
_db=None,
_geoip_db=None,
_http_session=None,
_raven_client=None,
_redis_client=None,
_position_searcher=None,
_region_searcher=None,
):
"""
Configure the web app stored in :data:`ichnaea.webapp.app._APP`.
Does connection, logging and view config setup. Attaches some
additional functionality to the :class:`pyramid.registry.Registry`
instance.
At startup ping all outbound connections like the database
once, to ensure they are actually up and responding.
The parameters starting with an underscore are test-only hooks
to provide pre-configured connection objects.
:param ping_connections: If True, ping and test outside connections.
:type ping_connections: bool
:returns: A configured WSGI app, the result of calling
:meth:`pyramid.config.Configurator.make_wsgi_app`.
"""
configure_logging()
config = Configurator()
check_config()
# add support for pt templates
config.include("pyramid_chameleon")
# add a config setting to skip logging for some views
config.registry.skip_logging = set()
configure_api(config)
configure_content(config)
configure_monitor(config)
# configure outside connections
registry = config.registry
registry.db = configure_db("ro", _db=_db)
registry.raven_client = raven_client = configure_raven(
transport="gevent", tags={"app": "webapp"}, _client=_raven_client
)
registry.redis_client = redis_client = configure_redis(_client=_redis_client)
configure_stats()
registry.http_session = configure_http_session(_session=_http_session)
registry.geoip_db = geoip_db = configure_geoip(
raven_client=raven_client, _client=_geoip_db
)
# Needs to be the exact same as the *_incoming entries in taskapp.config.
registry.data_queues = data_queues = {
"update_incoming": DataQueue(
"update_incoming", redis_client, "report", batch=100, compress=True
)
}
for name, func, default in (
("position_searcher", configure_position_searcher, _position_searcher),
("region_searcher", configure_region_searcher, _region_searcher),
):
searcher = func(
geoip_db=geoip_db,
raven_client=raven_client,
redis_client=redis_client,
data_queues=data_queues,
_searcher=default,
)
setattr(registry, name, searcher)
config.add_tween("ichnaea.db.db_tween_factory", under=EXCVIEW)
config.add_tween("ichnaea.log.log_tween_factory", under=EXCVIEW)
config.add_request_method(db_session, property=True)
# freeze skip logging set
config.registry.skip_logging = frozenset(config.registry.skip_logging)
# Should we try to initialize and establish the outbound connections?
if ping_connections:
with db_worker_session(registry.db, commit=False) as session:
ping_session(session)
registry.redis_client.ping()
return config.make_wsgi_app()