diff options
Diffstat (limited to 'allotter/management/commands/loadexam.py')
| -rw-r--r-- | allotter/management/commands/loadexam.py | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/allotter/management/commands/loadexam.py b/allotter/management/commands/loadexam.py new file mode 100644 index 0000000..2fedb66 --- /dev/null +++ b/allotter/management/commands/loadexam.py @@ -0,0 +1,40 @@ +from csv import reader +from django.core.management.base import BaseCommand, CommandError +from allotter.models import Exam + +class Command(BaseCommand): + args = '<file_name...>' + help = "Give the filename of the csv file that has all the exam code and exam name relation" + + def handle(self, *args, **options): + + clean_exam() + + for fname in args: + load_exam(fname) + + self.stdout.write('Done\n') + + +def clean_exam(): + """Removes all the objects from the database, required as if not done there might be a case of multile entries""" + data = Exam.objects.all() + data.delete() + +def load_exam(filename): + """Load exam code and exam name from the given csv file. The file should + declare a list of "exam_code;exam_name". + """ + try: + csvFile = open(filename, 'rb') + except IOError as (errno,strerror): + print "I/O error({0}): {1}".format(errno, strerror) + + csvReader = reader(csvFile, delimiter=";") + + for data in csvReader: + new_exam = Exam.objects.create() + new_exam.exam_code = data[0] + new_exam.exam_name = data[1] + new_exam.save() + print "Added ({0} : {1})".format(data[0], data[1])
\ No newline at end of file |