def check_templates(filename, elasticsearch_id):
stored = session.query(LogTemplate).order_by(
LogTemplate.id.desc()).first()
if not stored or stored.filename != filename or stored.elasticsearch_id != elasticsearch_id:
session.add(
LogTemplate(filename=filename,
elasticsearch_id=elasticsearch_id))
def synchronize_log_template(*, session: Session = NEW_SESSION) -> None:
"""Synchronize log template configs with table.
This checks if the last row fully matches the current config values, and
insert a new row if not.
"""
filename = conf.get("logging", "log_filename_template")
elasticsearch_id = conf.get("elasticsearch", "log_id_template")
# Before checking if the _current_ value exists, we need to check if the old config value we upgraded in
# place exists!
pre_upgrade_filename = conf.upgraded_values.get(("logging", "log_filename_template"), filename)
pre_upgrade_elasticsearch_id = conf.upgraded_values.get(
("elasticsearch", "log_id_template"), elasticsearch_id
)
if pre_upgrade_filename != filename or pre_upgrade_elasticsearch_id != elasticsearch_id:
# The previous non-upgraded value likely won't be the _latest_ value (as after we've recorded the
# recorded the upgraded value it will be second-to-newest), so we'll have to just search which is okay
# as this is a table with a tiny number of rows
row = (
session.query(LogTemplate.id)
.filter(
or_(
LogTemplate.filename == pre_upgrade_filename,
LogTemplate.elasticsearch_id == pre_upgrade_elasticsearch_id,
)
)
.order_by(LogTemplate.id.desc())
.first()
)
if not row:
session.add(
LogTemplate(filename=pre_upgrade_filename, elasticsearch_id=pre_upgrade_elasticsearch_id)
)
session.flush()
stored = session.query(LogTemplate).order_by(LogTemplate.id.desc()).first()
if not stored or stored.filename != filename or stored.elasticsearch_id != elasticsearch_id:
session.add(LogTemplate(filename=filename, elasticsearch_id=elasticsearch_id))
def synchronize_log_template(*, session: Session = NEW_SESSION) -> None:
"""Synchronize log template configs with table.
This checks if the last row fully matches the current config values, and
insert a new row if not.
"""
stored = session.query(LogTemplate).order_by(LogTemplate.id.desc()).first()
filename = conf.get("logging", "log_filename_template")
prefix = conf.get("logging", "task_log_prefix_template")
if stored and stored.filename == filename and stored.task_prefix == prefix:
return
session.merge(LogTemplate(filename=filename, task_prefix=prefix))