#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import locale
locale.setlocale(locale.LC_ALL, '')

from tuikit import *
from pgtoolkit import pgbrowser
from pgtoolkit.toolbase import SimpleTool


class MyApplication(Application, SimpleTool):
    def __init__(self):
        Application.__init__(self)
        SimpleTool.__init__(self, name='browser', desc='PostgreSQL database browser.')
        self.init()
        
        self.top.connect('keypress', self.globalkeypress)

        browser = pgbrowser.PgBrowser(self.pgm.get_conn('target'))
        model = TreeModel()
        view = TreeView(model)
        self.top.add(view)
        
        # populate schemas
        schemas = browser.list_schemas()
        model.add('/',  [schema['name'] for schema in schemas if not schema['system']])
        # populate tables
        for schema in schemas:
            if schema['system']:
                continue
            tables = browser.list_tables(schema=schema['name'])
            schemanode = '/'+schema['name']
            model.add(schemanode,  [table['name'] for table in tables])
            view.collapse(schemanode)        
        
        vert = VerticalLayout()
        self.top.layout(vert)
        
        view.setfocus()

    def globalkeypress(self, keyname, char):
        if keyname == 'escape':
            self.terminate()


if __name__ == '__main__':
    app = MyApplication()
    app.start()