From 44e6c8438e6415d350815bfa7bfd8f3ced783176 Mon Sep 17 00:00:00 2001
From: Maximilian Moser <maximilian.moser@tuwien.ac.at>
Date: Wed, 28 Apr 2021 14:31:35 +0200
Subject: [PATCH] Persist the last_check_at timestamp after file verification

---
 invenio_utilities_tuw/cli/drafts.py  | 4 ++++
 invenio_utilities_tuw/cli/records.py | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/invenio_utilities_tuw/cli/drafts.py b/invenio_utilities_tuw/cli/drafts.py
index 937aaab..75e6b85 100644
--- a/invenio_utilities_tuw/cli/drafts.py
+++ b/invenio_utilities_tuw/cli/drafts.py
@@ -15,6 +15,7 @@ from os.path import basename, isdir, isfile, join
 
 import click
 from flask.cli import with_appcontext
+from invenio_db import db
 from invenio_files_rest.models import ObjectVersion
 
 from ..utils import get_draft_file_service, get_record_service
@@ -354,3 +355,6 @@ def verify_files(pid, pid_type, user):
             "{} files failed the checksum verification".format(num_errors), fg="red"
         )
         sys.exit(1)
+
+    # persist the 'last_check_at' timestamp for each file
+    db.session.commit()
diff --git a/invenio_utilities_tuw/cli/records.py b/invenio_utilities_tuw/cli/records.py
index 2cd49d2..b6a31fd 100644
--- a/invenio_utilities_tuw/cli/records.py
+++ b/invenio_utilities_tuw/cli/records.py
@@ -221,6 +221,9 @@ def verify_files(pid, pid_type, user):
         )
         sys.exit(1)
 
+    # persist the 'last_check_at' timestamp for each file
+    db.session.commit()
+
 
 @records.command("reindex")
 @option_pid_values
-- 
GitLab