# XML-database conversion

import re

from xml.dom import Node

import db_util

import xml_util

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

ptr_suffix = '_id'

def is_ptr(node_name): return node_name.endswith(ptr_suffix)

def ptr_type(node_name):

    assert is_ptr(node_name)

    return node_name[:-len(ptr_suffix)]

def ptr_target(node):

    assert is_ptr(name_of(node))

    return xml_util.first_elem(node)

def find_by_name(node, name):

    for parent in xml_util.NodeParentIter(node):

        if name_of(parent) == name: return parent

        else:

            for child in xml_util.NodeElemIter(parent):

                child_name = name_of(child)

                if is_ptr(child_name):

                    if ptr_type(child_name) == name: return ptr_target(child)

                elif child_name == name: return child

    return None

def xml2db(node, db):

    pkeys = {}

    def pkey(table):

        if table not in pkeys: pkeys[table] = db_util.pkey(db, table)

        return pkeys[table]

    def main(node):

        for child in xml_util.NodeElemIter(node):

            if not xml_util.is_text(child): obj(child) # not XML metadata

    def obj(node):

        table = name_of(node)

        pkey_ = pkey(table)

        row = {}

        children = []

        # Divide children into fields and children with fkeys to parent

        for child in xml_util.NodeElemIter(node):

            child_name = name_of(child)

            if xml_util.is_text(child): row[child_name] = xml_util.value(child)

            elif is_ptr(child_name):

                child = ptr_target(child)

                obj(child)

                row[child_name] = xml_util.get_id(child)

            else: children.append(child)

        try: del row[pkey_]

        except KeyError: pass

        # Add fkey to parent

        parent_id = xml_util.get_id(node.parentNode)

        if parent_id != '': row[pkey(name_of(node.parentNode))] = parent_id

        # Insert node

        for try_num in range(2):

            try:

                xml_util.set_id(node, db_util.insert_or_get(db, table, row,

                    pkey_, row_ct_ref))

                break

            except db_util.NullValueException, ex:

                if try_num > 0: raise # exception still raised after retry

                # Search for required column in ancestors and their children

                target = find_by_name(node, ptr_type(ex.col))

                if target == None: raise

                row[ex.col] = xml_util.get_id(target)

        # Insert children with fkeys to parent

        for child in children: obj(child)

    row_ct_ref = [0]

    main(node)

    return row_ct_ref[0]