#!/usr/bin/env python3.2

from pgtoolkit import pgbrowser, toolbase, prettysize

class ListTablesTool(toolbase.SimpleTool):

    def __init__(self):

        toolbase.SimpleTool.__init__(self, name='listtables', desc='List tables in database.')

        self.parser.add_argument('-o', dest='options', type=str, nargs='*', help='Filter by options (eg. -o autovacuum_enabled=false).')

        self.init()

    def main(self):

        browser = pgbrowser.PgBrowser(self.pgm.get_conn('target'))

        # scan all tables from all shemas, remember names and sizes

        tables = []

        for schema in browser.schemas.values():

            for table in schema.tables.values():

                for option in self.args.options:

                    if option in table.options:

                        tables.append(table)

        # print result

        if len(tables):

            print('Found %d tables:' % len(tables))

        else:

            print('No table meets the conditions.')

        for table in tables:

            table_name = '%s.%s' % (table.schema.name, table.name)

            print(' ', table_name)

tool = ListTablesTool()

tool.main()