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

import datetime

import xml_dom

def map_items(func, items):

    return [(name, func(value)) for name, value in items]

def range_(items):

    items = dict(map_items(float, items))

    return str(items['to'] - items['from'])

def avg(items):

    count = 0

    sum_ = 0.

    for name, value in map_items(float, items):

        count += 1

        sum_ += value

    return str(sum_/count)

def date(items):

    items = dict(items)

    year = float(items['year'])

    date = datetime.date(int(year), int(items.get('month', '1')),

        int(items.get('day', '1'))) + datetime.timedelta(round((year % 1.)*365))

    return date.strftime('%Y-%m-%d')

def name(items):

    items = dict(items)

    return ' '.join([items['first'], items['last']])

def namePart(items):

    items = dict(items)

    def to_parts(name): return items[name].split(' ')

    parts = []

    if 'first' in items: parts += to_parts('first')[:1]

    if 'last' in items: parts += to_parts('last')[-1:]

    if 'middle' in items: parts += to_parts('middle')[1:-1]

    return ' '.join(parts)

# Function names must start with _ to avoid collisions with real tags

# Function names must be lowercase because name_of() returns name lowercased

# Functions take arguments (doc, node)

funcs = {'_range': range_, '_avg': avg, '_date': date, '_name': name,

    '_namepart': namePart}

def process(doc, node=None):

    if node == None: node = doc.documentElement

    name = xml_dom.name_of(node)

    if name in funcs: xml_dom.replace_with_text(doc, node,

        funcs[name](xml_dom.NodeTextEntryIter(node)))

    else:

        for child in xml_dom.NodeElemIter(node): process(doc, child)