schemadiff.py
author Radek Brich <radek.brich@devl.cz>
Sat, 29 Sep 2012 14:01:47 +0200
changeset 50 f71d3abbb18f
parent 44 4fe39c59c515
child 56 94e091c23ebb
permissions -rwxr-xr-x
Add test_prettysize to tests.py.
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
2
efee419b7a2d Fix pgmanager exceptions.
Radek Brich <radek.brich@devl.cz>
parents:
diff changeset
     1
#!/usr/bin/env python3.2
44
4fe39c59c515 Update longqueries tool: add client IP, waiting attributes.
Radek Brich <radek.brich@devl.cz>
parents: 16
diff changeset
     2
#
4fe39c59c515 Update longqueries tool: add client IP, waiting attributes.
Radek Brich <radek.brich@devl.cz>
parents: 16
diff changeset
     3
# Print differences in database schema.
4fe39c59c515 Update longqueries tool: add client IP, waiting attributes.
Radek Brich <radek.brich@devl.cz>
parents: 16
diff changeset
     4
#
4fe39c59c515 Update longqueries tool: add client IP, waiting attributes.
Radek Brich <radek.brich@devl.cz>
parents: 16
diff changeset
     5
# Prints changes from source to destination.
4fe39c59c515 Update longqueries tool: add client IP, waiting attributes.
Radek Brich <radek.brich@devl.cz>
parents: 16
diff changeset
     6
# SQL patch updates source database schema to destination schema.
4fe39c59c515 Update longqueries tool: add client IP, waiting attributes.
Radek Brich <radek.brich@devl.cz>
parents: 16
diff changeset
     7
#
2
efee419b7a2d Fix pgmanager exceptions.
Radek Brich <radek.brich@devl.cz>
parents:
diff changeset
     8
9
2fcc8ef0b97d Reorganize again :-) Add setup.py.
Radek Brich <radek.brich@devl.cz>
parents: 7
diff changeset
     9
from pgtoolkit import pgmanager, pgbrowser, pgdiff, toolbase
2
efee419b7a2d Fix pgmanager exceptions.
Radek Brich <radek.brich@devl.cz>
parents:
diff changeset
    10
efee419b7a2d Fix pgmanager exceptions.
Radek Brich <radek.brich@devl.cz>
parents:
diff changeset
    11
7
685b20d2d3ab Reorganize directories. PgDataDiff - reworked. PgManager - add fetchone_adapted, fetchall_adapted to cursor.
Radek Brich <radek.brich@devl.cz>
parents: 5
diff changeset
    12
class SchemaDiffTool(toolbase.SrcDstTool):
5
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    13
    def __init__(self):
7
685b20d2d3ab Reorganize directories. PgDataDiff - reworked. PgManager - add fetchone_adapted, fetchall_adapted to cursor.
Radek Brich <radek.brich@devl.cz>
parents: 5
diff changeset
    14
        toolbase.SrcDstTool.__init__(self, name='schemadiff', desc='Database schema diff.')
5
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    15
        
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    16
        self.parser.add_argument('-s', dest='schema', nargs='*', help='Schema filter')
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    17
        self.parser.add_argument('-t', dest='table', nargs='*', help='Table filter')
16
cb7e13711a99 MyManager - create_conn() - allow same parameters as for PgManager.
Radek Brich <brich.radek@ifortuna.cz>
parents: 9
diff changeset
    18
        self.parser.add_argument('--sql', action='store_true', help='Output is SQL script.')
5
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    19
        
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    20
        self.init()
2
efee419b7a2d Fix pgmanager exceptions.
Radek Brich <radek.brich@devl.cz>
parents:
diff changeset
    21
5
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    22
    def main(self):
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    23
        srcbrowser = pgbrowser.PgBrowser(self.pgm.get_conn('src'))
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    24
        dstbrowser = pgbrowser.PgBrowser(self.pgm.get_conn('dst'))
2
efee419b7a2d Fix pgmanager exceptions.
Radek Brich <radek.brich@devl.cz>
parents:
diff changeset
    25
        
5
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    26
        pgd = pgdiff.PgDiff(srcbrowser, dstbrowser)
2
efee419b7a2d Fix pgmanager exceptions.
Radek Brich <radek.brich@devl.cz>
parents:
diff changeset
    27
5
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    28
        if self.args.schema:
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    29
            pgd.filter_schemas(include=self.args.schema)
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    30
        
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    31
        if self.args.table:
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    32
            pgd.filter_tables(include=self.args.table)
2
efee419b7a2d Fix pgmanager exceptions.
Radek Brich <radek.brich@devl.cz>
parents:
diff changeset
    33
16
cb7e13711a99 MyManager - create_conn() - allow same parameters as for PgManager.
Radek Brich <brich.radek@ifortuna.cz>
parents: 9
diff changeset
    34
        if self.args.sql:
cb7e13711a99 MyManager - create_conn() - allow same parameters as for PgManager.
Radek Brich <brich.radek@ifortuna.cz>
parents: 9
diff changeset
    35
            pgd.print_patch()
cb7e13711a99 MyManager - create_conn() - allow same parameters as for PgManager.
Radek Brich <brich.radek@ifortuna.cz>
parents: 9
diff changeset
    36
        else:
cb7e13711a99 MyManager - create_conn() - allow same parameters as for PgManager.
Radek Brich <brich.radek@ifortuna.cz>
parents: 9
diff changeset
    37
            pgd.print_diff()
5
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    38
2
efee419b7a2d Fix pgmanager exceptions.
Radek Brich <radek.brich@devl.cz>
parents:
diff changeset
    39
5
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    40
tool = SchemaDiffTool()
57cf8fdff5ed Clean up tools.
Radek Brich <radek.brich@devl.cz>
parents: 2
diff changeset
    41
tool.main()