#!/usr/bin/env python

# Imports an XML file into a PostgreSQL database

# Format: see http://vegbank.org/vegdocs/xml/vegbank_example_ver1.0.2.xml

import os

import psycopg2

from psycopg2.extensions import ISOLATION_LEVEL_SERIALIZABLE

import sys

import xml.dom.minidom

import xml_db

def env_flag(name): return name in os.environ and os.environ[name] != ''

def main():

    prog_name = sys.argv.pop(0)

    try:

        db_config = {}

        for name in ['host', 'user', 'password', 'database']:

            if os.environ[name] != '': db_config[name] = os.environ[name]

    except KeyError: raise Exception('Usage: env host=... user=... password=...'

        ' database=... [commit=1] '+prog_name+' <dataset')

    commit = env_flag('commit')

    # Process dataset

    db = psycopg2.connect(**db_config)

    db.set_isolation_level(ISOLATION_LEVEL_SERIALIZABLE)

    try:

        doc = xml.dom.minidom.parse(sys.stdin)

        print 'Inserted '+str(xml_db.xml2db(doc.documentElement, db))+' rows'

        if commit: db.commit()

    finally:

        db.rollback()

        db.close()

main()