def index_after_commit(sender, changes):
"""Index a record in ES after it was committed to the DB.
This cannot happen in an ``after_record_commit`` receiver from Invenio-Records
because, despite the name, at that point we are not yet sure whether the record
has been really committed to the DB.
"""
indexer = RecordIndexer()
for model_instance, change in changes:
if isinstance(model_instance, RecordMetadata):
if change in ('insert', 'update') and not model_instance.json.get("deleted"):
if hasattr(model_instance, '_enhanced_record'):
record = model_instance._enhanced_record
else:
record = model_instance.json
indexer.index(InspireRecord(record, model_instance))
else:
try:
indexer.delete(InspireRecord(
model_instance.json, model_instance))
except NotFoundError:
# Record not found in ES
LOGGER.debug('Record %s not found in ES',
model_instance.json.get("id"))
pass
pid_type = get_pid_type_from_schema(model_instance.json['$schema'])
pid_value = model_instance.json['control_number']
db_version = model_instance.version_id
index_modified_citations_from_record.delay(pid_type, pid_value, db_version)
def receive_after_model_commit(sender, changes):
"""Perform actions after models committed to database."""
indexer = RecordIndexer()
for model_instance, change in changes:
if isinstance(model_instance, RecordMetadata):
if change in ('insert', 'update'):
indexer.index(InspireRecord(model_instance.json, model_instance))
else:
indexer.delete(InspireRecord(model_instance.json, model_instance))
def receive_after_model_commit(sender, changes):
"""Perform actions after models committed to database."""
indexer = RecordIndexer()
for model_instance, change in changes:
if isinstance(model_instance, RecordMetadata):
if change in ('insert', 'update'):
indexer.index(
InspireRecord(model_instance.json, model_instance))
else:
indexer.delete(
InspireRecord(model_instance.json, model_instance))
def _delete_record_from_everywhere(pid_type, record_control_number):
record = get_db_record(pid_type, record_control_number)
ri = RecordIndexer()
ri.delete(record)
record.delete(force=True)
pid = PersistentIdentifier.get(pid_type, record_control_number)
PersistentIdentifier.delete(pid)
object_uuid = pid.object_uuid
PersistentIdentifier.query.filter(
object_uuid == PersistentIdentifier.object_uuid).delete()
db.session.commit()
def index_after_commit(sender, changes):
"""Index a record in ES after it was committed to the DB.
This cannot happen in an ``after_record_commit`` receiver from Invenio-Records
because, despite the name, at that point we are not yet sure whether the record
has been really committed to the DB.
"""
indexer = RecordIndexer()
for model_instance, change in changes:
if isinstance(model_instance, RecordMetadata):
if change in ('insert', 'update'):
indexer.index(Record(model_instance.json, model_instance))
else:
indexer.delete(Record(model_instance.json, model_instance))
def index_after_commit(sender, changes):
"""Index a record in ES after it was committed to the DB.
This cannot happen in an ``after_record_commit`` receiver from Invenio-Records
because, despite the name, at that point we are not yet sure whether the record
has been really committed to the DB.
"""
indexer = RecordIndexer()
for model_instance, change in changes:
if isinstance(model_instance, RecordMetadata):
if change in ('insert', 'update'):
indexer.index(Record(model_instance.json, model_instance))
else:
indexer.delete(Record(model_instance.json, model_instance))
def index_after_commit(sender, changes):
"""Index records automatically after each modification."""
indexer = RecordIndexer()
for model_instance, change in changes:
if isinstance(model_instance, RecordMetadata):
if change in ('insert', 'update') and model_instance.json:
indexer.index(Record(model_instance.json, model_instance))
else:
try:
indexer.delete(Record(model_instance.json, model_instance))
except NotFoundError:
# Record not found in ES
current_app.logger.warning(
'Record with id "%s" not found in ElasticSearch' %
model_instance.json.get('control_number'))
