blob: 5e708f0aa295a896c13dfa9a2bae84af2d354eb6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
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
|
