Project

General

Profile

1
# XML "function" nodes that evaluate their contents to text
2

    
3
import datetime
4

    
5
import exc
6
import xml_dom
7

    
8
class SyntaxException(Exception):
9
    def __init__(self, cause=None):
10
        Exception.__init__(self, 'Invalid syntax in XML function: '+str(cause))
11

    
12
def map_items(func, items):
13
    return [(name, func(value)) for name, value in items]
14

    
15
def conv_items(type_, items):
16
    try: return map_items(type_, items)
17
    except ValueError, e: raise SyntaxException(e)
18

    
19
def alt(items):
20
    items = list(items)
21
    items.sort()
22
    return items[0][1] # value of lowest-numbered item
23

    
24
def range_(items):
25
    items = dict(conv_items(float, items))
26
    try:
27
        to = items['to']
28
        from_ = items['from']
29
    except KeyError, e: raise SyntaxException(e)
30
    return str(to - from_)
31

    
32
def avg(items):
33
    count = 0
34
    sum_ = 0.
35
    for name, value in conv_items(float, items):
36
        count += 1
37
        sum_ += value
38
    return str(sum_/count)
39

    
40
def date(items):
41
    items = dict(items)
42
    if 'date' in items:
43
        str_ = items['date']
44
        try: year = float(str_)
45
        except ValueError:
46
            try: import dateutil.parser
47
            except ImportError: return str_
48
            try: date = dateutil.parser.parse(str_)
49
            except ValueError, e: raise SyntaxException(e)
50
        else: date = (datetime.date(int(year), 1, 1) +
51
            datetime.timedelta(round((year % 1.)*365)))
52
    else:
53
        try: year = items['year']
54
        except KeyError, e: raise SyntaxException(e)
55
        month = items.get('month', '1')
56
        day = items.get('day', '1')
57
        try:
58
            year = int(year)
59
            month = int(month)
60
            day = int(day)
61
        except ValueError, e: raise SyntaxException(e)
62
        date = datetime.date(year, month, day)
63
    return date.strftime('%Y-%m-%d')
64

    
65
_name_parts_slices_items = [
66
    ('first', slice(None, 1)),
67
    ('middle', slice(1, -1)),
68
    ('last', slice(-1, None)),
69
]
70
name_parts_slices = dict(_name_parts_slices_items)
71
name_parts = [name for name, slice_ in _name_parts_slices_items]
72

    
73
def name(items):
74
    items = dict(items)
75
    parts = []
76
    for part in name_parts:
77
        if part in items: parts.append(items[part])
78
    return ' '.join(parts)
79

    
80
def name_part(items):
81
    out_items = []
82
    for part, value in items:
83
        try: slice_ = name_parts_slices[part]
84
        except KeyError, e: raise SyntaxException(e)
85
        else: out_items.append((part, ' '.join(value.split(' ')[slice_])))
86
    return name(out_items)
87

    
88
# Function names must start with _ to avoid collisions with real tags
89
# Functions take arguments (items)
90
funcs = {'_alt': alt, '_range': range_, '_avg': avg, '_date': date,
91
    '_name': name, '_namePart': name_part}
92

    
93
def process(node, on_error=exc.raise_):
94
    name = node.tagName
95
    if name.startswith('_') and name in funcs:
96
        try: value = funcs[name](xml_dom.NodeTextEntryIter(node))
97
        except SyntaxException, e:
98
            str_ = str(node)
99
            exc.add_msg(e, 'function:\n'+str_)
100
            xml_dom.replace(node, node.ownerDocument.createComment('\n'+str_))
101
            on_error(e)
102
        else: xml_dom.replace_with_text(node, value)
103
    else:
104
        for child in xml_dom.NodeElemIter(node): process(child, on_error)
(10-10/11)