summaryrefslogtreecommitdiffstats
path: root/util/local_database/xpathlite.py
blob: 6e4dd71fd86635ae7b925e5128ad4dae74489f51 (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
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
102
103
104
105
106
107
import sys
import os
import xml.dom.minidom

doc_cache = {}

class DraftResolution:
    # See http://www.unicode.org/cldr/process.html for description
    unconfirmed = 'unconfirmed'
    provisional = 'provisional'
    contributed = 'contributed'
    approved = 'approved'


def findChild(parent, tag_name, arg_value, draft=None):
    for node in parent.childNodes:
        if node.nodeType != node.ELEMENT_NODE:
            continue
        if node.nodeName != tag_name:
            continue
        if arg_value:
            if not node.attributes.has_key('type'):
                continue
            if node.attributes['type'].nodeValue != arg_value:
                continue
        if draft:
            if node.attributes.has_key('draft'):
                if node.attributes['draft'].nodeValue != draft:
                    continue
            elif draft != DraftResolution.approved:
                continue
        return node
    return False

def _findEntry(file, path, draft=None):
    doc = False
    if doc_cache.has_key(file):
        doc = doc_cache[file]
    else:
        doc = xml.dom.minidom.parse(file)
        doc_cache[file] = doc

    elt = doc.documentElement
    tag_spec_list = path.split("/")
    last_entry = None
    if draft is not None:
        last_entry = tag_spec_list[-1]
        tag_spec_list = tag_spec_list[:-1]
    for tag_spec in tag_spec_list:
        tag_name = tag_spec
        arg_value = ''
        left_bracket = tag_spec.find('[')
        if left_bracket != -1:
            tag_name = tag_spec[:left_bracket]
            arg_value = tag_spec[left_bracket+1:-1]
        elt = findChild(elt, tag_name, arg_value)
        if not elt:
            return ""
    if last_entry is not None:
        elt = findChild(elt, last_entry, '', draft)
        if not elt:
            return ""
    return elt.firstChild.nodeValue

def findAlias(file):
    doc = doc_cache[file]
    alias_elt = findChild(doc.documentElement, "alias", "")
    if not alias_elt:
        return False
    if not alias_elt.attributes.has_key('source'):
        return False
    return alias_elt.attributes['source'].nodeValue

def findEntry(base, path, draft=None):
    file = base + ".xml"

    if os.path.isfile(file):
        result = _findEntry(file, path, draft)
        if result:
            return result

        alias = findAlias(file)
        if alias:
            file = os.path.dirname(base) + "/" + alias + ".xml"
            if os.path.isfile(file):
                result = _findEntry(file, path, draft)
                if result:
                    return result

    file = base[:-3] + ".xml"
    if os.path.isfile(file):
        result = _findEntry(file, path, draft)
        if result:
            return result
        alias = findAlias(file)
        if alias:
            file = os.path.dirname(base) + "/" + alias + ".xml"
            if os.path.isfile(file):
                result = _findEntry(file, path, draft)
                if result:
                    return result

    if not draft:
        file = os.path.dirname(base) + "/root.xml"
        result = _findEntry(file, path, draft)
    return result