added all server files

1 files changed, 33 insertions, 0 deletions

diff --git a/sbhs_server/tables/management/commands/generate_checksum.py b/sbhs_server/tables/management/commands/generate_checksum.py
new file mode 100644
index 0000000..5e708f0
--- /dev/null
+++ b/sbhs_server/tables/management/commands/generate_checksum.py
@@ -0,0 +1,33 @@
+from django.core.management.base import BaseCommand
+from sbhs_server.tables.models import Experiment
+import hashlib
+
+class Command(BaseCommand):
+    args = ''
+    help = 'Calculates checksum for unchecked experiments'
+
+    def handle(self, *args, **options):
+        experiments = Experiment.objects.filter(checksum="NONE")
+
+        for e in experiments:
+            try:
+                f = open(e.log, "r")
+                # If log file doesn't exist, it means experiment is not done yet.
+                # This takes care of that.
+            except:
+                return
+
+            data = f.read()
+            f.close()
+            data = data.strip().split("\n")
+            clean_data = ""
+            for line in data:
+                columns = line.split(" ")
+                if len(columns) >= 6:
+                    clean_data += (" ".join(columns[0:6]) + "\n")
+
+            checksum = hashlib.sha1(clean_data).hexdigest()
+
+            e.checksum = checksum
+            e.save()
+            print checksum