From ac6226de1d7c785f38224231c9c9d0296b9e3b94 Mon Sep 17 00:00:00 2001
From: Ryan Herbert <ryan.herbert@inria.fr>
Date: Mon, 12 Sep 2016 17:47:04 +0200
Subject: [PATCH] scripts/clear_results_and_scheduler_tasks.py

Added a script whichs clears duplicate sequence_file-config pairs from
the results_file table. The most recent entry containing said pair is
kept. Scheduler tasks associated to the deleted entries are also
cleared.
---
 .../clear_results_and_scheduler_tasks.py      | 65 +++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100755 server/web2py/applications/vidjil/scripts/clear_results_and_scheduler_tasks.py

diff --git a/server/web2py/applications/vidjil/scripts/clear_results_and_scheduler_tasks.py b/server/web2py/applications/vidjil/scripts/clear_results_and_scheduler_tasks.py
new file mode 100755
index 000000000..cf6b55e31
--- /dev/null
+++ b/server/web2py/applications/vidjil/scripts/clear_results_and_scheduler_tasks.py
@@ -0,0 +1,65 @@
+#!/usr/bin/python
+import argparse
+
+parser = argparse.ArgumentParser(description='Process results and scheduler tasks that are no longer in use.')
+parser.add_argument('--assume-yes', action='store_true', help='assume yes on all choices')
+
+args = parser.parse_args()
+
+def prompt(question):
+    affir = ['y', 'yes']
+    negat = ['n', 'no']
+    confirm = affir[0] if args.assume_yes else ""
+    while (confirm.lower() not in affir + negat):
+        if confirm != "":
+            print "Invalid option."
+        confirm = raw_input(question)
+        if confirm == "":
+            confirm = negat[0]
+            
+    if confirm in affir:
+        return True
+    elif confirm in negat:
+        return False
+    else:
+        raise Exception('Illegal State')
+
+results_files = db(
+                db.sequence_file.id == db.results_file.sequence_file_id
+            ).select(
+                db.results_file.ALL,
+                orderby=(db.results_file.sequence_file_id|db.results_file.config_id|~db.results_file.run_date)
+            )
+
+prev_sfid = -1
+prev_cid = -1
+for row in results_files:
+    if row.sequence_file_id != prev_sfid or row.config_id != prev_cid:
+        targets = (
+            (db.results_file.sequence_file_id == row.sequence_file_id) &
+            (db.results_file.config_id == row.config_id) &
+            (db.results_file.id != row.id)
+        )
+        if db(targets).count() > 0:
+            print "Conserving: %d" % row.id
+
+            task_targets = (targets &
+                (db.scheduler_task.id == db.results_file.scheduler_task_id)
+            )
+            task_ids = [r.id for r in db(task_targets).select(db.scheduler_task.id)]
+            print "Removing scheduler tasks: %s" % str(task_ids)
+
+            choice = prompt("confirm deleting these %d scheduler tasks ? [y/N]" % len(task_ids))
+            if choice:
+                for target in db(targets).select(db.results_file.scheduler_task_id):
+                    db(db.scheduler_task.id == target.scheduler_task_id).delete()
+
+            result_ids = [r.id for r in db(targets).select(db.results_file.id)]
+            print "Removing results_files: %s" % str(result_ids)
+
+            choice = prompt("confirm deleting these %d results files ? [y/N]" % len(result_ids))
+            if choice:
+                db(targets).delete()
+
+        prev_sfid = row.sequence_file_id
+        prev_cid = row.config_id
-- 
GitLab