blob: 1e9cfde91e6e8531cffe28554527435a994580a7 (
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
# System library imports.
from os.path import basename
from xml.dom.minidom import parse
from htmlentitydefs import name2codepoint
import re
# Django imports.
from django.core.management.base import BaseCommand
# Local imports.
from exam.models import Question
def decode_html(html_str):
"""Un-escape or decode HTML strings to more usable Python strings.
From here: http://wiki.python.org/moin/EscapingHtml
"""
return re.sub('&(%s);' % '|'.join(name2codepoint),
lambda m: unichr(name2codepoint[m.group(1)]), html_str)
def clear_questions():
"""Delete all questions from the database."""
for question in Question.objects.all():
question.delete()
def load_questions_xml(filename):
"""Load questions from the given XML file."""
q_bank = parse(filename).getElementsByTagName("question")
for question in q_bank:
summary_node = question.getElementsByTagName("summary")[0]
summary = (summary_node.childNodes[0].data).strip()
desc_node = question.getElementsByTagName("description")[0]
description = (desc_node.childNodes[0].data).strip()
points_node = question.getElementsByTagName("points")[0]
points = int((points_node.childNodes[0].data).strip()) \
if points_node else 1
test_node = question.getElementsByTagName("test")[0]
test = decode_html((test_node.childNodes[0].data).strip())
new_question = Question(summary=summary,
description=description,
points=points,
test=test)
new_question.save()
class Command(BaseCommand):
args = '<q_file1.xml q_file2.xml>'
help = 'loads the questions from given XML files'
def handle(self, *args, **options):
"""Handle the command."""
# Delete existing stuff.
clear_questions()
# Load from files.
for fname in args:
self.stdout.write('Importing from {0} ... '.format(basename(fname)))
load_questions_xml(fname)
self.stdout.write('Done\n')
|