diff -r 000000000000 -r de9c2efb9d02 SqlBrowser/SqlConnectionWidget.py
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/SqlBrowser/SqlConnectionWidget.py Mon Dec 28 16:03:33 2009 +0000
@@ -0,0 +1,172 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2009 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing a widget showing the SQL connections.
+"""
+
+from PyQt4.QtCore import *
+from PyQt4.QtGui import *
+from PyQt4.QtSql import QSqlDatabase
+
+class SqlConnectionWidget(QWidget):
+ """
+ Class implementing a widget showing the SQL connections.
+
+ @signal tableActivated(QString) emitted after the entry for a table has been activated
+ @signal schemaRequested(QString) emitted when the schema display is requested
+ @signal cleared() emitted after the connection tree has been cleared
+ """
+ def __init__(self, parent = None):
+ """
+ Constructor
+
+ @param parent reference to the parent widget (QWidget)
+ """
+ QWidget.__init__(self, parent)
+
+ layout = QVBoxLayout(self)
+ layout.setMargin(0)
+
+ self.__connectionTree = QTreeWidget(self)
+ self.__connectionTree.setObjectName("connectionTree")
+ self.__connectionTree.setHeaderLabels([self.trUtf8("Database")])
+ self.__connectionTree.header().setResizeMode(QHeaderView.Stretch)
+ refreshAction = QAction(self.trUtf8("Refresh"), self.__connectionTree)
+ self.__schemaAction = QAction(self.trUtf8("Show Schema"), self.__connectionTree)
+
+ self.connect(refreshAction, SIGNAL("triggered()"), self.refresh)
+ self.connect(self.__schemaAction, SIGNAL("triggered()"), self.showSchema)
+
+ self.__connectionTree.addAction(refreshAction)
+ self.__connectionTree.addAction(self.__schemaAction)
+ self.__connectionTree.setContextMenuPolicy(Qt.ActionsContextMenu)
+
+ layout.addWidget(self.__connectionTree)
+
+ self.connect(self.__connectionTree,
+ SIGNAL("itemActivated(QTreeWidgetItem*, int)"),
+ self.__itemActivated)
+ self.connect(self.__connectionTree,
+ SIGNAL("currentItemChanged(QTreeWidgetItem*, QTreeWidgetItem*)"),
+ self.__currentItemChanged)
+
+ self.__activeDb = ""
+
+ def refresh(self):
+ """
+ Public slot to refresh the connection tree.
+ """
+ self.__connectionTree.clear()
+ self.emit(SIGNAL("cleared()"))
+
+ connectionNames = QSqlDatabase.connectionNames()
+
+ foundActiveDb = False
+ for name in connectionNames:
+ root = QTreeWidgetItem(self.__connectionTree)
+ db = QSqlDatabase.database(name, False)
+ root.setText(0, self.__dbCaption(db))
+ if name == self.__activeDb:
+ foundActiveDb = True
+ self.__setActive(root)
+ if db.isOpen():
+ tables = db.tables()
+ for table in tables:
+ itm = QTreeWidgetItem(root)
+ itm.setText(0, table)
+
+ if not foundActiveDb and connectionNames:
+ self.__activeDb = connectionNames[0]
+ self.__setActive(self.__connectionTree.topLevelItem(0))
+
+ self.__connectionTree.doItemsLayout()
+
+ def showSchema(self):
+ """
+ Public slot to show schema data of a database.
+ """
+ cItm = self.__connectionTree.currentItem()
+ if cItm is None or cItm.parent() is None:
+ return
+ self.__setActive(cItm.parent())
+ self.emit(SIGNAL("schemaRequested(QString)"), cItm.text(0))
+
+ def __itemActivated(self, itm, column):
+ """
+ Private slot handling the activation of an item.
+
+ @param itm reference to the item (QTreeWidgetItem)
+ @param column column that was activated (integer)
+ """
+ if itm is None:
+ return
+
+ if itm.parent() is None:
+ self.__setActive(itm)
+ else:
+ self.__setActive(itm.parent())
+ self.emit(SIGNAL("tableActivated(QString)"), itm.text(0))
+
+ def __currentItemChanged(self, current, previous):
+ """
+ Private slot handling a change of the current item.
+
+ @param current reference to the new current item (QTreeWidgetItem)
+ @param previous reference to the previous current item (QTreeWidgetItem)
+ """
+ self.__schemaAction.setEnabled(
+ current is not None and current.parent() is not None)
+
+ def __dbCaption(self, db):
+ """
+ Private method to assemble a string for the caption.
+
+ @param db reference to the database object (QSqlDatabase)
+ @return caption string (string)
+ """
+ nm = db.driverName()
+ nm += ":"
+ if db.userName():
+ nm += db.userName()
+ nm += "@"
+ nm += db.databaseName()
+ return nm
+
+ def __setBold(self, itm, bold):
+ """
+ Private slot to set the font to bold.
+
+ @param itm reference to the item to be changed (QTreeWidgetItem)
+ @param bold flag indicating bold (boolean)
+ """
+ font = itm.font(0)
+ font.setBold(bold)
+ itm.setFont(0, font)
+
+ def currentDatabase(self):
+ """
+ Public method to get the current database.
+
+ @return reference to the current database (QSqlDatabase)
+ """
+ return QSqlDatabase.database(self.__activeDb)
+
+ def __setActive(self, itm):
+ """
+ Private slot to set an item to active.
+
+ @param itm reference to the item to set as the active item (QTreeWidgetItem)
+ """
+ for index in range(self.__connectionTree.topLevelItemCount()):
+ if self.__connectionTree.topLevelItem(index).font(0).bold():
+ self.__setBold(self.__connectionTree.topLevelItem(index), False)
+
+ if itm is None:
+ return
+
+ self.__setBold(itm, True)
+ self.__activeDb = QSqlDatabase.connectionNames()[
+ self.__connectionTree.indexOfTopLevelItem(itm)]
