schemadiff.py
author Radek Brich <radek.brich@devl.cz>
Mon, 05 Mar 2012 18:36:46 +0100
changeset 31 c2e6e24b83d9
parent 16 cb7e13711a99
child 44 4fe39c59c515
permissions -rwxr-xr-x
Add browser - database schema browser using tuikit (curses UI). Add listdepends - tool which shows depending views for column. Update pgdatadiff - allow composite primary key. Update pgmanager - RowDict is now OrderedDict. Drop support for Python2.x.

#!/usr/bin/env python3.2

from pgtoolkit import pgmanager, pgbrowser, pgdiff, toolbase


class SchemaDiffTool(toolbase.SrcDstTool):
    def __init__(self):
        toolbase.SrcDstTool.__init__(self, name='schemadiff', desc='Database schema diff.')
        
        self.parser.add_argument('-s', dest='schema', nargs='*', help='Schema filter')
        self.parser.add_argument('-t', dest='table', nargs='*', help='Table filter')
        self.parser.add_argument('--sql', action='store_true', help='Output is SQL script.')
        
        self.init()

    def main(self):
        srcbrowser = pgbrowser.PgBrowser(self.pgm.get_conn('src'))
        dstbrowser = pgbrowser.PgBrowser(self.pgm.get_conn('dst'))
        
        pgd = pgdiff.PgDiff(srcbrowser, dstbrowser)

        if self.args.schema:
            pgd.filter_schemas(include=self.args.schema)
        
        if self.args.table:
            pgd.filter_tables(include=self.args.table)

        if self.args.sql:
            pgd.print_patch()
        else:
            pgd.print_diff()


tool = SchemaDiffTool()
tool.main()