DBLP2Bib/XMLFileToEndNote.py at master · ashriram/DBLP2Bib · GitHub

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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#!/usr/bin/python

import xml.sax
import sys
import io
# reload(sys)    # to re-enable sys.setdefaultencoding()
# sys.setdefaultencoding('utf8')

skip = 0

CONFABBRV = {'ISCA': 37, 'HPCA': 16, 'MICRO': 43,
             'PACT': 19, 'ASPLOS': 15, 'PLDI': 31}

output = ""


def ord(n):
    return str(n)+("th" if 4 <= n % 100 <= 20 else {1: "st", 2: "nd", 3: "rd"}.get(n % 10, "th"))


def conftitle(conf, year):
    return "Proc. of the " + ord(int(year)-2010+CONFABBRV[conf]) + " " + conf


class MovieHandler(xml.sax.ContentHandler):
    def __init__(self):
        self.CurrentData = ""
        self.type = ""
        self.format = ""
        self.year = ""
        self.rating = ""
        self.stars = ""
        self.description = ""

    # Call when an element starts
    def startElement(self, tag, attributes):
        global skip
        global output
        self.CurrentData = tag
        if tag == "inproceedings":
            output = output + "\n\n"
            skip = 0
        if tag == "proceedings":
            skip = 1

    # Call when an elements ends
    def endElement(self, tag):
        global skip
        global output
        if skip:
            return
        if tag == "inproceedings":
            output = output + "%R " + "\n"
        if self.CurrentData == "author":
            output = output + "%A " + self.type.strip() + "\n"
        elif self.CurrentData == "title":
            output = output + "%T " + self.format.strip() + "\n"
        elif self.CurrentData == "year":
            output = output + "%D " + self.year.strip() + "\n"
        elif self.CurrentData == "pages":
            output = output + "%P " + self.rating.strip() + "\n"
        elif self.CurrentData == "booktitle":
            if (self.year != ''):
                output = output + \
                    "%J " + conftitle(self.stars.strip(),
                                      self.year.strip()) + "\n"
        elif self.CurrentData == "description":
            output = output + "Description:" + self.description + "\n"
        self.CurrentData = ""

    # Call when a character is read
    def characters(self, content):
        if self.CurrentData == "author":
            self.type = content
        elif self.CurrentData == "title":
            self.format = content
        elif self.CurrentData == "year":
            self.year = content
        elif self.CurrentData == "pages":
            self.rating = content
        elif self.CurrentData == "booktitle":
            self.stars = content
        elif self.CurrentData == "description":
            self.description = content


if (__name__ == "__main__"):

    # create an XMLReader
    parser = xml.sax.make_parser()
    # turn off namepsaces
    parser.setFeature(xml.sax.handler.feature_namespaces, 0)

    # override the default ContextHandler
    Handler = MovieHandler()
    parser.setContentHandler(Handler)

    parser.parse(str(sys.argv[1]))
#   xml.sax.parseString(text,Handler)
    with io.open(str(sys.argv[2]), "w") as text_file:
        text_file.write(output)