#!/usr/bin/env python3.2

import sys

import argparse

from pgtools import pgmanager, pgbrowser, pgdiff

from common import config

parser = argparse.ArgumentParser(description='Database schema diff.')

parser.add_argument('source', metavar='source', type=str, help='Source database')

parser.add_argument('destination', metavar='destination', type=str, help='Destination database')

parser.add_argument('-s', dest='schema', nargs='*', help='Schema filter')

parser.add_argument('-t', dest='table', nargs='*', help='Table filter')

args = parser.parse_args()

config = config.ConfigParser()

config.add_argument('databases', type=dict)

config.add_argument('meta_db')

config.add_argument('meta_query')

config.load('pgtoolkit.conf')

pgm = pgmanager.get_instance()

pgm.create_conn(name='meta', dns=config.meta_db)

def build_conn(name, targetname):

    with pgm.cursor('meta') as curs:

        curs.execute(config.meta_query, [targetname])

        row = curs.fetchone_dict()

        curs.connection.commit()

        pgm.create_conn(name=name, **row)

build_conn('src', args.source)

build_conn('dst', args.destination)

pgm.close_conn('meta')

srcbrowser = pgbrowser.PgBrowser(pgm.get_conn('src'))

dstbrowser = pgbrowser.PgBrowser(pgm.get_conn('dst'))

pgd = pgdiff.PgDiff(srcbrowser, dstbrowser)

if args.schema:

    pgd.filter_schemas(include=args.schema)

if args.table:

    pgd.filter_tables(include=args.table)

pgd.print_diff()