-
Notifications
You must be signed in to change notification settings - Fork 0
/
xml_parser.py
61 lines (54 loc) · 1.91 KB
/
xml_parser.py
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
from xml.sax import handler, make_parser
import DataCarrier
class XML_parser(handler.ContentHandler):
def __init__(self):
super().__init__()
self.data = DataCarrier.DataCarrier()
self.content = ""
self.student = ""
self.insert_remove = ""
self.action_id = ""
self.workplace = ""
self.get_subject = False
self.subject = ""
self.prcv = ""
self.zsls = ""
def parse_file(self, f):
parser = make_parser()
parser.setContentHandler(self)
parser.parse(f)
def startElement(self, name, attrs):
if name == "actor":
self.student = attrs["personalNumber"]
elif name == "processedData":
self.insert_remove = attrs["activity"]
elif name == "timetableAction":
self.action_id = attrs["tt:id"]
elif name == "tt:subject":
self.get_subject = True
self.prcv = attrs["kind"]
self.content = ""
def endElement(self, name):
if self.get_subject:
self.subject = self.content
self.get_subject = False
elif name == "tt:term":
self.zsls = self.content
elif name == "tt:department":
self.workplace = self.content
elif name == "event":
line = [self.student, self.insert_remove, self.action_id, self.workplace, self.subject, self.prcv,
self.zsls]
self.data.all_actions += 1
if self.insert_remove == "insert":
self.data.insert_line(line)
self.data.sign_in += 1
elif self.insert_remove == "delete":
self.data.remove_line(line)
self.data.delete += 1
self.data.sign_in -= 1
else:
print("Chyba I/O. Končím.")
exit(3)
def characters(self, content):
self.content += content