analyzeall.py
author Radek Brich <radek.brich@devl.cz>
Tue, 14 Feb 2012 18:15:56 +0100
changeset 29 8636719a30f6
parent 28 27fc0504663d
child 30 a8b7cd92f39f
permissions -rwxr-xr-x
Add batchquery tool.

#!/usr/bin/env python3.2

from pgtoolkit import pgbrowser, toolbase
from pgtoolkit.highlight import highlight


class AnalyzeAllTool(toolbase.SimpleTool):
    def __init__(self):
        toolbase.SimpleTool.__init__(self, name='analyzeall', desc='Analyze all tables.')
        self.parser.add_argument('-s', dest='schema', nargs='*', help='Schema filter')
        self.parser.add_argument('--vacuum', action='store_true', help='Call VACUUM ANALYZE')
        self.parser.add_argument('--full', action='store_true', help='Call VACUUM FULL ANALYZE')
        self.init()

    def main(self):
        browser = pgbrowser.PgBrowser(self.pgm.get_conn('target'))
        
        query_pattern = 'ANALYZE %s.%s;'
        if self.args.vacuum:
            query_pattern = 'VACUUM ANALYZE %s.%s;'
        if self.args.full:
            query_pattern = 'VACUUM FULL ANALYZE %s.%s;'

        schema_list = self.args.schema
        if not schema_list:
            schema_list = [schema['name'] for schema in browser.list_schemas() if not schema['system']]

        for schema in schema_list:
            tables = browser.list_tables(schema=schema)
            with self.pgm.cursor('target') as curs:
                for table in tables:
                    query = query_pattern % (schema, table['name'])
                    print(query)
                    curs.execute(query, [])


tool = AnalyzeAllTool()
tool.main()