1 files changed, 9 insertions, 0 deletions
diff --git a/SQLiteStudio3/guiSQLiteStudio/dbtree/dbtreeview.cpp b/SQLiteStudio3/guiSQLiteStudio/dbtree/dbtreeview.cpp
index 9382d7c..797652a 100644
--- a/SQLiteStudio3/guiSQLiteStudio/dbtree/dbtreeview.cpp
+++ b/SQLiteStudio3/guiSQLiteStudio/dbtree/dbtreeview.cpp
@@ -45,6 +45,15 @@ DbTreeItem *DbTreeView::currentItem()
     return dynamic_cast<DbTreeItem*>(model()->itemFromIndex(currentIndex()));
 }
 
+DbTreeItem* DbTreeView::currentDbItem()
+{
+    DbTreeItem* item = currentItem();
+    if (item->getType() == DbTreeItem::Type::DB)
+        return item;
+
+    return item->findParentItem(DbTreeItem::Type::DB);
+}
+
 DbTreeItem *DbTreeView::itemAt(const QPoint &pos)
 {
     return dynamic_cast<DbTreeItem*>(model()->itemFromIndex(indexAt(pos)));