forked from hans/Nominatim
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
"""
|
|
Functions for updating a database from a replication source.
|
|
"""
|
|
import datetime
|
|
import logging
|
|
|
|
from osmium.replication.server import ReplicationServer
|
|
|
|
from ..db import status
|
|
|
|
LOG = logging.getLogger()
|
|
|
|
def init_replication(conn, base_url):
|
|
""" Set up replication for the server at the given base URL.
|
|
"""
|
|
LOG.info("Using replication source: %s", base_url)
|
|
date = status.compute_database_date(conn)
|
|
|
|
# margin of error to make sure we get all data
|
|
date -= datetime.timedelta(hours=3)
|
|
|
|
repl = ReplicationServer(base_url)
|
|
|
|
seq = repl.timestamp_to_sequence(date)
|
|
|
|
if seq is None:
|
|
LOG.fatal("Cannot reach the configured replication service '%s'.\n"
|
|
"Does the URL point to a directory containing OSM update data?",
|
|
base_url)
|
|
raise RuntimeError("Failed to reach replication service")
|
|
|
|
status.set_status(conn, date=date, seq=seq)
|
|
|
|
LOG.warning("Updates intialised at sequence %s (%s)", seq, date)
|
|
|
|
|
|
def check_for_updates(conn, base_url):
|
|
""" Check if new data is available from the replication service at the
|
|
given base URL.
|
|
"""
|
|
_, seq, _ = status.get_status(conn)
|
|
|
|
if seq is None:
|
|
LOG.error("Replication not set up. "
|
|
"Please run 'nominatim replication --init' first.")
|
|
return 254
|
|
|
|
state = ReplicationServer(base_url).get_state_info()
|
|
|
|
if state is None:
|
|
LOG.error("Cannot get state for URL %s.", base_url)
|
|
return 253
|
|
|
|
if state.sequence <= seq:
|
|
LOG.warning("Database is up to date.")
|
|
return 1
|
|
|
|
LOG.warning("New data available (%i => %i).", seq, state.sequence)
|
|
return 0
|