#!/usr/bin/env python3.2
#
# Print differencies between data in two tables of same schema.
#
# Requirements:
#  * First column of both tables must be numerical primary key. 
#  * Destination table must contain all columns from source table.
#    Order is not important.
#

from pgtools import pgmanager, pgbrowser, pgdatadiff
from toolbase import SrcDstTool


class TableDiffTool(SrcDstTool):
    def __init__(self):
        SrcDstTool.__init__(self, name='tablediff', desc='Table diff.')
        
        self.parser.add_argument('table', metavar='table', type=str, help='Table name')
        
        self.init()

    def main(self):
        if '.' in self.args.table:
            schema, table = self.args.table.split('.', 1)
        else:
            table = self.args.table

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

        columns = srcbrowser.list_columns(schema=schema, table=table)
        columns_sel = ', '.join(['"' + x['name'] + '"' for x in columns])
        
        table_fullname = '"' + schema + '"."'+ table + '"';
        query = 'SELECT ' + columns_sel + ' FROM ' + table_fullname + ' ORDER BY 1;'
        
        with self.pgm.cursor('src') as curs:
            curs.execute(query)
            src_rows = curs.fetchall()

        with self.pgm.cursor('dst') as curs:
            curs.execute(query)
            dst_rows = curs.fetchall()

        pgdd = pgdatadiff.PgDataDiff(table_fullname,
            src_rows, dst_rows, [x['name'] for x in columns])
        #pgdd.print_diff()
        pgdd.print_patch()


tool = TableDiffTool()
tool.main()