'''
Created on 06.04.2011

@author: jwh
'''

from PySide.QtCore import *
from PySide.QtGui import *


class DatabaseLogic(object):
    '''
    classdoc
    '''


    def __init__(self,db):

        self.db=db
        self.db.host = "localhost"
        self.db.user = "postgres"
        self.db.password = "kingkong1"
        self.db.database = "vfs1"
        self.db.connect()
        self.cur = self.db.connection.cursor()

    def fillLocTable(self):
        self.cur.execute('''select * from "config"."Location" ;''')
        locs_table = QTableWidget()
        for j in range(self.cur.rowcount):
            b=self.cur.fetchone()
            for i in range(len(b)):
                newItem = QTableWidgetItem(str(b[i]))
                locs_table.setItem(j,i, newItem)

        self.cur.execute('''select attname from pg_attribute,pg_class where attrelid = pg_class.oid and relname = 'Location' and attnum>0;''')

        header_temp=self.cur.fetchall()
        header_items=["","",""]

        for i in range(len(header_temp)):
            header_items[i]=header_items[i].join(header_temp[i])

        for i in range(len(header_items)):
            newItem = QTableWidgetItem(str(header_items[i]))
            locs_table.setHorizontalHeaderItem(i,newItem)
        #self.cur.close()
        print "test"
        return locs_table

    def fillUserTree(self):
        users_tree=QTreeWidget()
        self.cur.execute("select usename from pg_user;")

        top_temp=self.cur.fetchall()
        top_items=["","","",""]

        for i in range(len(top_temp)):
            top_items[i]=top_items[i].join(top_temp[i])



        items = []
        for i in range(len(top_items)):
            items.append(QTreeWidgetItem(users_tree))
            items[-1].setText(0,top_items[i])

        users_tree.insertTopLevelItems(0,items)

        return users_tree