Project

General

Profile

1
# XML-database conversion
2

    
3
import re
4
import traceback
5
from xml.dom import Node
6

    
7
import exc
8
import sql
9
import strings
10
import util
11
import xml_dom
12

    
13
def name_of(node): return re.sub(r'^.*\.', r'', node.tagName)
14

    
15
ptr_suffix = '_id'
16

    
17
def is_ptr(node_name): return node_name.lower().endswith(ptr_suffix)
18

    
19
def ptr_type_guess(node_name):
20
    assert is_ptr(node_name)
21
    return node_name[:-len(ptr_suffix)]
22

    
23
def ptr_target(node):
24
    assert is_ptr(name_of(node))
25
    return xml_dom.first_elem(node)
26

    
27
def find_by_name(node, name):
28
    for parent in xml_dom.NodeParentIter(node):
29
        if name_of(parent) == name: return parent
30
        else:
31
            for child in xml_dom.NodeElemIter(parent):
32
                child_name = name_of(child)
33
                if is_ptr(child_name):
34
                    target = ptr_target(child)
35
                    if target.tagName == name: return target
36
                elif child_name == name: return child
37
    return None
38

    
39
def get(db, node, limit=None, start=None):
40
    def pkey(table): return sql.pkey(db, table)
41
    
42
    node = node.firstChild
43
    table = name_of(node)
44
    pkey_ = pkey(table)
45
    
46
    fields = []
47
    conds = {}
48
    for child in xml_dom.NodeElemIter(node):
49
        child_name = name_of(child)
50
        if xml_dom.is_empty(child): fields.append(child_name)
51
        elif xml_dom.is_text(child): conds[child_name] = xml_dom.value(child)
52
        else: raise Exception('Joins not supported yet')
53
    id_ = xml_dom.get_id(node)
54
    if id_ != None: conds[pkey(table)] = id_ # replace any existing pkey value
55
    if fields == []: fields.append(pkey_)
56
    
57
    return sql.select(db, table, fields, conds, limit, start)
58

    
59
def put(db, node, row_ct_ref=None, on_error=exc.raise_, pool=None,
60
    store_ids=False, parent_id=None):
61
    '''store_ids enables searching the tree for missing fields'''
62
    def pkey(table): return sql.pkey(db, table, True)
63
    
64
    def put_(node, parent_id=None):
65
        return put(db, node, row_ct_ref, on_error, pool, store_ids, parent_id)
66
    
67
    def on_error_(e):
68
        exc.add_msg(e, 'node:\n'+str(node))
69
        on_error(e)
70
    
71
    table = name_of(node)
72
    try: pkey_ = pkey(table)
73
    except sql.DatabaseErrors, e: on_error_(e); return None
74
    row = {}
75
    children = []
76
    
77
    # Divide children into fields and children with fkeys to parent
78
    for child in xml_dom.NodeElemIter(node):
79
        child_name = name_of(child)
80
        if xml_dom.is_empty(child): row[child_name] = None
81
        elif xml_dom.is_text(child):
82
            row[child_name] = strings.to_unicode(xml_dom.value(child))
83
        elif is_ptr(child_name): row[child_name] = put_(ptr_target(child))
84
        else: children.append(child)
85
    try: del row[pkey_]
86
    except KeyError: pass
87
    
88
    # Add fkey to parent
89
    if parent_id != None:
90
        parent_ptr = node.getAttribute('fkey')
91
        if parent_ptr == '': parent_ptr = pkey(name_of(node.parentNode))
92
        row[parent_ptr] = parent_id
93
    
94
    # Insert node
95
    try:
96
        for try_num in xrange(2):
97
            try:
98
                id_ = sql.put(db, table, row, pkey_, row_ct_ref)
99
                if store_ids: xml_dom.set_id(node, id_)
100
                break
101
            except sql.NullValueException, e:
102
                col = e.cols[0]
103
                if try_num > 0: raise # exception still raised after retry
104
                if store_ids and is_ptr(col):
105
                    # Search for required column in ancestors and their children
106
                    target = find_by_name(node, ptr_type_guess(col))
107
                    if target == None: raise
108
                    row[col] = xml_dom.get_id(target)
109
                else: raise
110
    except sql.DatabaseErrors, e: on_error_(e); return None
111
    
112
    def put_child(node, parent_id):
113
        call = lambda: put_(node, parent_id)
114
        if pool != None: pool.apply_async(call)
115
        else: call()
116
    
117
    # Insert children with fkeys to parent
118
    for child in children: put_child(child, id_)
119
    
120
    return id_
(7-7/29)