def OnCreate(self, form):
# if HAS_PYSIDE:
# self.parent = self.FormToPySideWidget(form)
# else:
self.parent = self.FormToPyQtWidget(form)
self.tv = QTreeView()
self.tv.setExpandsOnDoubleClick(False)
root_layout = QVBoxLayout(self.parent)
# self.le_filter = QLineEdit(self.parent)
# root_layout.addWidget(self.le_filter)
root_layout.addWidget(self.tv)
self.parent.setLayout(root_layout)
self._model = QtGui.QStandardItemModel()
self._init_model()
self.tv.setModel(self._model)
self.tv.setColumnWidth(0, 200)
self.tv.setColumnWidth(1, 300)
self.tv.header().setStretchLastSection(True)
self.tv.expandAll()
self.tv.doubleClicked.connect(self.on_navigate_to_method_requested)
def __init__(self):
QTreeView.__init__(self)
# MVC
# 可拖拽
self.dragEnabled()
# 可释放
self.acceptDrops()
# 拖拽 indicator
self.showDropIndicator()
# 拖拽模式
self.setDragDropMode(QAbstractItemView.InternalMove)
# 标题拉伸最后一列
self.header().setStretchLastSection(True)
# 无 focus, 去掉蓝框
self.setFocusPolicy(Qt.NoFocus)
# 设置样式
self.setStyleSheet(get_theme("TreeView"))
def __init__(self):
super(MainWindow, self).__init__()
self.model = Project()
self.view = QTreeView(self)
self.view.setModel(self.model)
self.view.setDragEnabled(True)
self.view.setDragDropMode(QAbstractItemView.InternalMove)
self.setCentralWidget(self.view)
class Grid(object):
def __init__(self, data):
self.data = data
def create_widget(self, parent=None):
self.tree_view = QTreeView(parent)
self.model = ListModel(self.data)
self.tree_view.setModel(self.model)
return self.tree_view
class DictTreeView(object):
def __init__(self, data):
self.data = data
self.treeView = QTreeView()
self.treeView.setModel(DictModel(self.data))
def set_on_clicked(self, callback):
def execute(index):
callback(index.internalPointer().data)
self.treeView.clicked.connect(execute)
def __init__(self):
QTreeView.__init__(self)
# MVC
self.dragEnabled()
self.acceptDrops()
self.showDropIndicator()
self.setDragDropMode(QAbstractItemView.InternalMove)
self.header().setStretchLastSection(True)
self.setFocusPolicy(Qt.NoFocus)
self.setStyleSheet(get_theme("TreeView"))
def __init__(self, parent=None):
super(getFilesDlg, self).__init__(parent)
self.setMinimumSize(500,500)
self.fileDlgPaths = []
layout = QVBoxLayout()
self.btnBox = QDialogButtonBox(QDialogButtonBox.Ok |
QDialogButtonBox.Cancel)
self.btnBox.accepted.connect(self.getPaths)
self.btnBox.rejected.connect(self.close)
self.fsModel = QFileSystemModel()
self.treeView = QTreeView()
self.treeView.setSelectionMode(QTreeView.ExtendedSelection)
self.treeView.setModel(self.fsModel)
self.treeView.setColumnWidth(0, 361)
self.treeView.setColumnHidden(1, True)
self.treeView.setColumnHidden(3, True)
layout.addWidget(self.treeView)
layout.addWidget(self.btnBox)
self.setLayout(layout)
self.fsModel.setRootPath(environ['HOMEPATH'])
# self.treeView.setRootIndex(self.fsModel.index("\\"))
self.treeView.expand(self.treeView.rootIndex())
def new_tab(self):
"""Open new tab."""
tasklist = QTreeView()
tasklist.hide()
tasklist.setObjectName('taskList')
tasklist.setMinimumWidth(100)
tasklist.setMaximumWidth(250)
new_tab = QWebView()
new_tab.setObjectName('webView')
inspector = QWebInspector(self)
inspector.setObjectName('webInspector')
inspector.hide()
page_layout = QVBoxLayout()
page_layout.setSpacing(0)
page_layout.setContentsMargins(0, 0, 0, 0)
page_layout.addWidget(new_tab)
page_layout.addWidget(inspector)
page_widget = QFrame()
page_widget.setObjectName('pageWidget')
page_widget.setLayout(page_layout)
complete_tab_layout = QHBoxLayout()
complete_tab_layout.setSpacing(0)
complete_tab_layout.setContentsMargins(0, 0, 0, 0)
complete_tab_layout.addWidget(tasklist)
complete_tab_layout.addWidget(page_widget)
complete_tab_widget = QFrame()
complete_tab_widget.setLayout(complete_tab_layout)
new_tab.load(QUrl(self.startpage))
self.tabs.setUpdatesEnabled(False)
if self.new_tab_behavior == "insert":
self.tabs.insertTab(self.tabs.currentIndex()+1, complete_tab_widget,
unicode(new_tab.title()))
elif self.new_tab_behavior == "append":
self.tabs.appendTab(complete_tab_widget, unicode(new_tab.title()))
self.tabs.setCurrentWidget(complete_tab_widget)
self.tabs.setTabText(self.tabs.currentIndex(),
unicode(self.tabs.currentWidget().findChild(QFrame, unicode('pageWidget')).findChild(QWebView, unicode('webView')).title()))
self.tabs.setUpdatesEnabled(True)
# tab.page().mainFrame().setScrollBarPolicy(Qt.Horizontal, Qt.ScrollBarAlwaysOff)
# tab.page().mainFrame().setScrollBarPolicy(Qt.Vertical, Qt.ScrollBarAlwaysOff)
new_tab.titleChanged.connect(self.change_tab)
new_tab.urlChanged.connect(self.change_tab)
new_tab.loadStarted.connect(self.load_start)
new_tab.loadFinished.connect(self.load_finish)
new_tab.loadProgress.connect(self.pbar.setValue)
new_tab.page().linkHovered.connect(self.linkHover)
inspector.setPage(new_tab.page())
def __init__(self, parent=None):
QTreeView.__init__(self)
self.header().setHidden(True)
self.setAnimated(True)
# Modelo
self.model = QFileSystemModel(self)
# path = QDir.toNativeSeparators(QDir.homePath())
# self.model.setRootPath(path)
self.setModel(self.model)
# Se ocultan algunas columnas
self.hideColumn(1)
self.hideColumn(2)
self.hideColumn(3)
# Conexion
self.doubleClicked.connect(self._open_file)
def __init__(self):
super(MainWindow, self).__init__()
self.model = Project()
self.view = QTreeView(self)
self.view.setModel(self.model)
self.view.setDragEnabled(True)
self.view.setDragDropMode(QAbstractItemView.InternalMove)
self.setCentralWidget(self.view)
def __init__(self, parent, mframe, existing_filters):
"""Create a FilterTab with the given parent TabDialog, logical
parent ModuleFrame mframe, and existing_filters list of Clause
objects.
"""
super(FilterTab, self).__init__(parent)
self.mframe = mframe
self.parent = parent
self.attributes = self.mframe.agent.datatree.generateAttributeList()
self.clause_list = list()
self.clause_dict = dict()
# Right now we only look at the first passed in filter.
# TODO: At GUI to switch between existing filters
if existing_filters is not None and len(existing_filters) > 0:
for clause in existing_filters[0].conditions.clauses:
self.clause_list.append(str(clause))
self.clause_dict[str(clause)] = clause
self.clause_model = QStringListModel(self.clause_list)
layout = QVBoxLayout(self)
self.sidesplitter = QSplitter(Qt.Horizontal)
# You can only select one attribute at a time to build the
# filter clauses
self.data_view = QTreeView(self)
self.data_view.setModel(self.mframe.agent.datatree)
self.data_view.setDragEnabled(True)
self.data_view.setDropIndicatorShown(True)
self.data_view.expandAll()
self.sidesplitter.addWidget(self.data_view)
self.sidesplitter.setStretchFactor(1,1)
self.filter_widget = self.buildFilterWidget()
self.sidesplitter.addWidget(self.filter_widget)
self.sidesplitter.setStretchFactor(1,0)
layout.addWidget(self.sidesplitter)
# Apply buttons
buttonWidget = QWidget()
buttonLayout = QHBoxLayout(buttonWidget)
self.applyButton = QPushButton("Apply")
self.applyButton.clicked.connect(self.applyFilter)
self.closeButton = QPushButton("Apply & Close")
self.closeButton.clicked.connect(self.applyCloseFilter)
buttonLayout.addWidget(self.applyButton)
buttonLayout.addWidget(self.closeButton)
buttonWidget.setLayout(buttonLayout)
layout.addWidget(buttonWidget)
self.setLayout(layout)
def __init__(self, path, parent=None):
QTreeView.__init__(self)
self.header().setHidden(True)
self.setAnimated(True)
# Modelo
self.model = FileSystemModel()
# path = QDir.toNativeSeparators(QDir.homePath())
self.model.setRootPath(path)
# self.setModel(self.model)
# self.setRootIndex(QModelIndex(self.model.index(path)))
# self.model.setNameFilters(["*.exe", "*.log", "*.s"])
self.model.setNameFilterDisables(False)
self.model.setFilter(QDir.AllDirs | QDir.NoDotAndDotDot)
# Se ocultan algunas columnas
self.hideColumn(1)
self.hideColumn(2)
self.hideColumn(3)
# Conexion
self.doubleClicked.connect(self._open_file)
def OnCreate(self, form):
if HAS_PYSIDE:
self.parent = self.FormToPySideWidget(form)
else:
self.parent = self.FormToPyQtWidget(form)
self._idp_hooks = AutoReIDPHooks(self)
if not self._idp_hooks.hook():
print 'IDP_Hooks.hook() failed'
self.tv = QTreeView()
self.tv.setExpandsOnDoubleClick(False)
root_layout = QVBoxLayout(self.parent)
# self.le_filter = QLineEdit(self.parent)
# root_layout.addWidget(self.le_filter)
root_layout.addWidget(self.tv)
self.parent.setLayout(root_layout)
self._model = QtGui.QStandardItemModel()
self._init_model()
self.tv.setModel(self._model)
self.tv.setColumnWidth(0, 200)
self.tv.setColumnWidth(1, 300)
self.tv.header().setStretchLastSection(True)
self.tv.expandAll()
self.tv.doubleClicked.connect(self.on_navigate_to_method_requested)
# self.le_filter.textChanged.connect(self.on_filter_text_changed)
self.tv.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.tv.customContextMenuRequested.connect(self._tree_customContextMenuRequesssted)
rename_action = QAction('Rename...', self.tv)
rename_action.setShortcut('n')
rename_action.triggered.connect(self._tv_rename_action_triggered)
self.tv.addAction(rename_action)
def _init_gui(self):
self._widget = QSplitter(self, Qt.Horizontal)
self._hierarchy_view = QTreeView(self._widget)
self._details_view = QTableView(self._widget)
self._widget.addWidget(self._hierarchy_view)
self._widget.addWidget(self._details_view)
self._widget.setStretchFactor(0, 2)
self._widget.setStretchFactor(1, 1)
self.setCentralWidget(self._widget)
self._hierarchy_view.setModel(self._hierarchy_model)
self._details_view.setModel(self._details_model)
self._hierarchy_view.expanded.connect(self._mo_item_expand)
def __init__(self, data, header=None):
self.treeView = QTreeView()
self.header = header
self.model = None
self.set_data(data)
# Expand the root item. I'm not sure if this is a good idea in
# general, but it works well on some examples. Maybe a
# heuristic is needed.
root_index = self.model.index(0, 0, QModelIndex())
self.treeView.setExpanded(root_index, True)
# Size all the columns to the size of their current
# contents. Note that this ignores contents for values in the
# tree that are collapsed by default, so it's of limited
# use. However, it's better than the default sizing.
for i in range(self.model.columnCount(None)):
self.treeView.resizeColumnToContents(i)
class MainWindow(QMainWindow):
def __init__(self):
super(MainWindow, self).__init__()
self.model = Project()
self.view = QTreeView(self)
self.view.setModel(self.model)
self.view.setDragEnabled(True)
self.view.setDragDropMode(QAbstractItemView.InternalMove)
self.setCentralWidget(self.view)
def mousePressEvent(self, e):
print e.x(), e.y()
return QMainWindow.mousePressEvent(self, e)
class MainWindow(QMainWindow):
def __init__(self):
super(MainWindow, self).__init__()
self.model = Project()
self.view = QTreeView(self)
self.view.setModel(self.model)
self.view.setDragEnabled(True)
self.view.setDragDropMode(QAbstractItemView.InternalMove)
self.setCentralWidget(self.view)
def mousePressEvent(self, e):
print e.x(), e.y()
return QMainWindow.mousePressEvent(self, e)
class NestedListTreeView(object):
def __init__(self, data, header=None):
self.treeView = QTreeView()
self.header = header
self.model = None
self.set_data(data)
# Expand the root item. I'm not sure if this is a good idea in
# general, but it works well on some examples. Maybe a
# heuristic is needed.
root_index = self.model.index(0, 0, QModelIndex())
self.treeView.setExpanded(root_index, True)
# Size all the columns to the size of their current
# contents. Note that this ignores contents for values in the
# tree that are collapsed by default, so it's of limited
# use. However, it's better than the default sizing.
for i in range(self.model.columnCount(None)):
self.treeView.resizeColumnToContents(i)
def set_on_clicked(self, callback):
def execute(index):
callback(index.internalPointer().data)
self.treeView.clicked.connect(execute)
def set_data(self, data):
self.data = data
self.model = ListModel(self.data, header=self.header)
self.treeView.setModel(self.model)
def refresh_data(self):
"""Refresh the UI's view of all data in the tree view. This may be
inefficient with very large data sets.
"""
root_index = self.model.index(0, 0, QModelIndex())
self.treeView.dataChanged(root_index, root_index)
def __init__(self, parent=None):
"""Initialize the parent class of this instance."""
super(window, self).__init__(parent)
app.aboutToQuit.connect(self.myExitHandler)
self.style_sheet = self.styleSheet('style')
# app.setStyle(QStyleFactory.create('Macintosh'))
#app.setStyleSheet(self.style_sheet)
self.layout().setSpacing(0)
self.layout().setContentsMargins(0,0,0,0)
app.setOrganizationName("Eivind Arvesen")
app.setOrganizationDomain("https://github.com/eivind88/raskolnikov")
app.setApplicationName("Raskolnikov")
app.setApplicationVersion("0.0.1")
settings = QSettings()
self.data_location = QDesktopServices.DataLocation
self.temp_location = QDesktopServices.TempLocation
self.cache_location = QDesktopServices.CacheLocation
self.startpage = "https://duckduckgo.com/"
self.new_tab_behavior = "insert"
global bookmarks
global saved_tabs
print "Currently saved_tabs:\n", saved_tabs
global menubar
menubar = QMenuBar()
# Initialize a statusbar for the window
self.statusbar = self.statusBar()
self.statusbar.setFont(QFont("Helvetica Neue", 11, QFont.Normal))
self.statusbar.setStyleSheet(self.style_sheet)
self.statusbar.setMinimumHeight(15)
self.pbar = QProgressBar()
self.pbar.setMaximumWidth(100)
self.statusbar.addPermanentWidget(self.pbar)
self.statusbar.hide()
self.setMinimumSize(504, 235)
# self.setWindowModified(True)
# app.alert(self, 0)
self.setWindowTitle("Raskolnikov")
# toolbar = self.addToolBar('Toolbar')
# toolbar.addAction(exitAction)
# self.setUnifiedTitleAndToolBarOnMac(True)
self.setWindowIcon(QIcon(""))
# Create input widgets
self.bbutton = QPushButton(u"<")
self.fbutton = QPushButton(u">")
self.hbutton = QPushButton(u"⌂")
self.edit = QLineEdit("")
self.edit.setFont(QFont("Helvetica Neue", 12, QFont.Normal))
self.edit.setPlaceholderText("Enter URL")
# self.edit.setMinimumSize(400, 24)
self.rbutton = QPushButton(u"↻")
self.dbutton = QPushButton(u"☆")
self.tbutton = QPushButton(u"⁐")
# ↆ ⇧ √ ⌘ ⏎ ⏏ ⚠ ✓ ✕ ✖ ✗ ✘ ::: ❤ ☮ ☢ ☠ ✔ ☑ ♥ ✉ ☣ ☤ ✘ ☒ ♡ ツ ☼ ☁ ❅ ✎
self.nbutton = QPushButton(u"+")
self.nbutton.setObjectName("NewTab")
self.nbutton.setMinimumSize(35, 30)
self.nbutton.setSizePolicy(QSizePolicy.Minimum,QSizePolicy.Minimum)
self.edit.setTextMargins(2, 1, 2, 0)
# create a horizontal layout for the input
input_layout = QHBoxLayout()
input_layout.setSpacing(4)
input_layout.setContentsMargins(0, 0, 0, 0)
# add the input widgets to the input layout
input_layout.addWidget(self.bbutton)
input_layout.addWidget(self.fbutton)
input_layout.addWidget(self.hbutton)
input_layout.addWidget(self.edit)
input_layout.addWidget(self.rbutton)
input_layout.addWidget(self.dbutton)
input_layout.addWidget(self.tbutton)
# create a widget to hold the input layout
self.input_widget = QFrame()
self.input_widget.setObjectName("InputWidget")
self.input_widget.setStyleSheet(self.style_sheet)
# set the layout of the widget
self.input_widget.setLayout(input_layout)
self.input_widget.setVisible(True)
# CREATE BOOKMARK-LINE HERE
self.bookmarks_layout = QHBoxLayout()
self.bookmarks_layout.setSpacing(0)
self.bookmarks_layout.setContentsMargins(0, 0, 0, 0)
for i in bookmarks:
link = QToolButton()
link.setDefaultAction(QAction(unicode(i), link))
link.setObjectName(unicode(i))
link.setFont(QFont("Helvetica Neue", 11, QFont.Normal))
link.clicked.connect(self.handleBookmarks)
self.bookmarks_layout.addWidget(link)
self.bookmarks_widget = QFrame()
self.bookmarks_widget.setObjectName("BookmarkWidget")
self.bookmarks_widget.setStyleSheet(self.style_sheet)
self.bookmarks_widget.setLayout(self.bookmarks_layout)
if not bookmarks:
self.bookmarks_widget.hide()
# Task list
self.tasklist = QStandardItemModel()
#parentItem = self.tasklist.invisibleRootItem()
#self.tasklist.header().hide()
self.tasklist.setHorizontalHeaderItem(0, QStandardItem('Tasks'))
parentItem = QStandardItem("Parent")
self.tasklist.appendRow(parentItem)
for i in range(4):
item = QStandardItem("Item %d" % i)
parentItem.appendRow(item)
#parentItem = item
#self.list.activated[str].connect(self.handleBookmarks)
#self.list.view().setSizePolicy(QSizePolicy.Minimum,QSizePolicy.Minimum)
# create tabs
self.tabs = QTabWidget()
self.tabs.setDocumentMode(True)
self.tabs.setTabsClosable(True)
self.tabs.setMovable(True)
self.tabs.tabBar().hide()
self.tabs.setFont(QFont("Helvetica Neue", 11, QFont.Normal))
self.tabs.setCornerWidget(self.nbutton)
self.tabs.cornerWidget().setObjectName("CornerWidget")
self.tabs.cornerWidget().setMinimumSize(10, 24)
self.tabs.cornerWidget().setSizePolicy(QSizePolicy.Minimum,QSizePolicy.Minimum)
if saved_tabs:
for tab in saved_tabs['tabs']:
tasklist = QTreeView()
tasklist.hide()
tasklist.setObjectName('taskList')
tasklist.setMinimumWidth(100)
tasklist.setMaximumWidth(250)
new_tab = QWebView()
new_tab.setObjectName('webView')
inspector = QWebInspector(self)
inspector.setObjectName('webInspector')
inspector.hide()
page_layout = QVBoxLayout()
page_layout.setSpacing(0)
page_layout.setContentsMargins(0, 0, 0, 0)
page_layout.addWidget(new_tab)
page_layout.addWidget(inspector)
page_widget = QFrame()
page_widget.setObjectName('pageWidget')
page_widget.setLayout(page_layout)
complete_tab_layout = QHBoxLayout()
complete_tab_layout.setSpacing(0)
complete_tab_layout.setContentsMargins(0, 0, 0, 0)
complete_tab_layout.addWidget(tasklist)
complete_tab_layout.addWidget(page_widget)
complete_tab_widget = QFrame()
complete_tab_widget.setLayout(complete_tab_layout)
#for page in tab['history']:
# new_tab.load(QUrl(page['url']))
#print tab['current_history']
#for item in new_tab.history().items():
# print item
#new_tab.history().goToItem(new_tab.history().itemAt(tab['current_history']))
new_tab.load(QUrl(tab['history'][tab['current_history']]['url']))
tab['current_history']
self.tabs.setUpdatesEnabled(False)
if self.new_tab_behavior == "insert":
self.tabs.insertTab(self.tabs.currentIndex()+1, complete_tab_widget,
unicode(new_tab.title()))
elif self.new_tab_behavior == "append":
self.tabs.appendTab(complete_tab_widget, unicode(new_tab.title()))
self.tabs.setUpdatesEnabled(True)
new_tab.titleChanged.connect(self.change_tab)
new_tab.urlChanged.connect(self.change_tab)
new_tab.loadStarted.connect(self.load_start)
new_tab.loadFinished.connect(self.load_finish)
new_tab.loadProgress.connect(self.pbar.setValue)
new_tab.page().linkHovered.connect(self.linkHover)
inspector.setPage(new_tab.page())
for index, tab in enumerate(saved_tabs['tabs']):
self.tabs.setTabText(index, tab['history'][tab['current_history']]['title'])
self.tabs.setCurrentIndex(saved_tabs['current_tab'])
else:
self.new_tab()
tabs_layout = QVBoxLayout()
tabs_layout.setSpacing(0)
tabs_layout.setContentsMargins(0, 0, 0, 0)
tabs_layout.addWidget(self.tabs)
self.tabs_widget = QFrame()
self.tabs_widget.setObjectName("TabLine")
self.tabs_widget.setStyleSheet(self.style_sheet)
self.tabs_widget.setLayout(tabs_layout)
self.tabs_widget.setVisible(True)
# Webkit settings
gsettings = self.tabs.currentWidget().findChild(QFrame, unicode('pageWidget')).findChild(QWebView, unicode('webView')).settings().globalSettings()
# Basic settings
gsettings.setAttribute(QWebSettings.AutoLoadImages, True)
gsettings.setAttribute(QWebSettings.JavascriptEnabled, True)
gsettings.setAttribute(QWebSettings.JavascriptCanOpenWindows, False)
gsettings.setAttribute(QWebSettings.JavascriptCanAccessClipboard, False)
gsettings.setAttribute(QWebSettings.PluginsEnabled, False) # Flash isn't stable at present
gsettings.setAttribute(QWebSettings.JavaEnabled, False) # Java applet's aren't supported by PySide
gsettings.setAttribute(QWebSettings.DeveloperExtrasEnabled, True)
gsettings.setAttribute(QWebSettings.AcceleratedCompositingEnabled,
True)
# Performace settings
gsettings.setAttribute(QWebSettings.DnsPrefetchEnabled, True)
gsettings.setAttribute(QWebSettings.AcceleratedCompositingEnabled, True)
gsettings.setAttribute(QWebSettings.DnsPrefetchEnabled, True)
# Other settings
gsettings.setAttribute(QWebSettings.PrivateBrowsingEnabled, False)
# Create a vertical layout and add widgets
vlayout = QVBoxLayout()
vlayout.setSpacing(0)
vlayout.setContentsMargins(0, 0, 0, 0)
# toolbar.addWidget(self.input_widget)
vlayout.addWidget(self.input_widget)
vlayout.addWidget(self.bookmarks_widget)
vlayout.addWidget(self.tabs_widget)
# create a widget to hold the vertical layout
wrapper_widget = QWidget()
wrapper_widget.setLayout(vlayout)
self.setCentralWidget(wrapper_widget)
self.bbutton.clicked.connect(self.tabs.currentWidget().findChild(QFrame, unicode('pageWidget')).findChild(QWebView, unicode('webView')).back)
self.fbutton.clicked.connect(self.tabs.currentWidget().findChild(QFrame, unicode('pageWidget')).findChild(QWebView, unicode('webView')).forward)
self.hbutton.clicked.connect(self.goHome)
self.edit.returnPressed.connect(self.set_url)
# Add button signal to "go" slot
self.rbutton.clicked.connect(self.tabs.currentWidget().findChild(QFrame, unicode('pageWidget')).findChild(QWebView, unicode('webView')).reload)
self.dbutton.clicked.connect(self.bookmark)
self.tbutton.clicked.connect(self.toggleTaskBar)
self.nbutton.clicked.connect(self.new_tab)
self.tabs.tabCloseRequested.connect(self.tabs.removeTab)
self.tabs.currentChanged.connect(self.change_tab)
widgets = (input_layout.itemAt(i).widget() for i in range(
input_layout.count()))
for widget in widgets:
if isinstance(widget, QPushButton):
widget.setFixedSize(33, 21)
widget.setFont(QFont("Helvetica Neue", 12, QFont.Normal))
widget.pressed.connect(self.press_button)
widget.released.connect(self.release_button)
# make a ctrl+q quit
sequence = QKeySequence(Qt.CTRL + Qt.Key_Q)
QShortcut(sequence, self, SLOT("close()"))
# make an accelerator to toggle fullscreen
sequence = QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_F)
QShortcut(sequence, self, self.toggle_fullscreen)
# make an accelerator to toggle input visibility
sequence = QKeySequence(Qt.CTRL + Qt.ALT + Qt.Key_L)
QShortcut(sequence, self, self.toggle_input)
# make an accelerator to focus adress-bar
sequence = QKeySequence(Qt.CTRL + Qt.Key_L)
QShortcut(sequence, self, self.focus_adress)
# make an accelerator to reload page
sequence = QKeySequence(Qt.CTRL + Qt.Key_R)
QShortcut(sequence, self, self.tabs.currentWidget().findChild(QFrame, unicode('pageWidget')).findChild(QWebView, unicode('webView')).reload)
# make an accelerator to create new tab
sequence = QKeySequence(Qt.CTRL + Qt.Key_T)
QShortcut(sequence, self, self.new_tab)
# make an accelerator to close tab
sequence = QKeySequence(Qt.CTRL + Qt.Key_W)
QShortcut(sequence, self, self.close_tab)
# make an accelerator to navigate tabs
sequence = QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_Left)
QShortcut(sequence, self, self.previous_tab)
sequence = QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_Right)
QShortcut(sequence, self, self.next_tab)
# make an accelerator to toggle inspector
sequence = QKeySequence(Qt.CTRL + Qt.ALT + Qt.Key_U)
QShortcut(sequence, self, self.handleShowInspector)
# make an accelerator to toggle bookmark
sequence = QKeySequence(Qt.CTRL + Qt.Key_D)
QShortcut(sequence, self, self.bookmark)
# make an accelerator to toggle task/project-list
sequence = QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_L)
QShortcut(sequence, self, self.toggleTaskBar)
# finally set the attribute need to rotate
# try:
# self.setAttribute(Qt.WA_Maemo5AutoOrientation, True)
# except:
# print "not maemo"
self.statusbar.show()
class MainWindow(QMainWindow):
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
self.dirty = False
# Models
# - Filesystem Model
self.fileSystemModel = QFileSystemModel()
# rootPath = QDesktopServices.storageLocation(
# QDesktopServices.HomeLocation)
# fileSystemRoot = self.fileSystemModel.setRootPath(rootPath)
fileSystemRoot = self.fileSystemModel.setRootPath(
"/home/ryan/Programming/Python/projects/r3tagger/r3tagger/tests"
)
# Views
# - Filesystem View
self.fileSystemView = QTreeView()
self.fileSystemView.setModel(self.fileSystemModel)
self.fileSystemView.setRootIndex(fileSystemRoot)
self.fileSystemView.doubleClicked.connect(self.updateAlbumModel)
self.fileSystemView.expanded.connect(self.fixFileSystemColumns)
self.fileSystemView.collapsed.connect(self.fixFileSystemColumns)
# - Album View
self.albumView = albumcollection.MusicCollectionView()
self.albumView.setSelectionMode(QAbstractItemView.MultiSelection)
self.albumView.clicked.connect(self.updateEditing)
self.albumView.expanded.connect(self.fixAlbumViewColumns)
self.albumView.collapsed.connect(self.fixAlbumViewColumns)
self.albumView.model().dataChanged.connect(self.fixAlbumViewColumns)
self.albumView.model().dataChanged.connect(self._setDirty)
model = self.albumView.model()
model.dataChanged.connect(self.updateEditing)
# Editing Group
self.editingGroup = QFormLayout()
self.lineArtist = QLineEdit()
self.lineAlbum = QLineEdit()
self.lineTitle = QLineEdit()
self.lineTrack = QLineEdit()
self.lineDate = QLineEdit()
self.lineGenre = QLineEdit()
self.editingGroup.addRow("Artist:", self.lineArtist)
self.editingGroup.addRow("Album:", self.lineAlbum)
self.editingGroup.addRow("Title:", self.lineTitle)
self.editingGroup.addRow("Track:", self.lineTrack)
self.editingGroup.addRow("Date:", self.lineDate)
self.editingGroup.addRow("Genre:", self.lineGenre)
self.tagsToAttribs = {
"artist": self.lineArtist,
"album": self.lineAlbum,
"title": self.lineTitle,
"tracknumber": self.lineTrack,
"date": self.lineDate,
"genre": self.lineGenre,
}
# Confirm / Cancel / Clear Group
self.buttonGroup = QHBoxLayout()
self.buttonGroup.addStretch()
confirm = QPushButton("Confirm")
confirm.clicked.connect(self.confirmChanges)
self.buttonGroup.addWidget(confirm)
cancel = QPushButton("Cancel")
cancel.clicked.connect(self.cancelChanges)
self.buttonGroup.addWidget(cancel)
clear = QPushButton("Clear")
clear.clicked.connect(self.clearAlbumView)
self.buttonGroup.addWidget(clear)
self.buttonGroup.addStretch()
# Statusbar
# status = self.statusBar()
# status.setSizeGripEnabled(False)
# status.showMessage("Ready", 5000)
# Docks
dockAllowed = Qt.LeftDockWidgetArea | Qt.RightDockWidgetArea
# - Filesystem Dock
fileSystemDock = QDockWidget("Navigate", self)
fileSystemDock.setObjectName("fileSystemDock")
fileSystemDock.setAllowedAreas(dockAllowed)
fileSystemDock.setWidget(self.fileSystemView)
self.addDockWidget(Qt.LeftDockWidgetArea, fileSystemDock)
# - Editing Dock
editingWidget = QWidget()
editingWidget.setLayout(self.editingGroup)
editingDock = QDockWidget("Editing", self)
editingDock.setObjectName("editingDock")
editingDock.setAllowedAreas(dockAllowed)
editingDock.setWidget(editingWidget)
self.addDockWidget(Qt.RightDockWidgetArea, editingDock)
# Actions
fileAddSongAction = self._createAction(
text="Add &Songs",
slot=self.fileAddSong,
shortcut=QKeySequence.Open,
icon="fileOpen",
tip="Add files (songs)",
)
fileAddAlbumAction = self._createAction(
text="Add &Album",
slot=self.fileAddAlbum,
shortcut=QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_O),
icon="fileOpen",
tip="Add directory (album)",
)
fileSaveAction = self._createAction(
text="&Save Changes",
slot=self.confirmChanges,
shortcut=QKeySequence.Save,
icon="fileSave",
tip="Save Changes",
)
fileQuitAction = self._createAction(
text="&Quit", slot=self.close, shortcut=QKeySequence.Quit, icon="fileQuit", tip="Quit Program"
)
editRecognizeAction = self._createAction(
text="&Recognize",
slot=self.editRecognize,
shortcut=QKeySequence(Qt.CTRL + Qt.Key_R),
icon="editRecognize",
tip="Recognize music",
)
editReorganizeAction = self._createAction(
text="Reorganize",
slot=self.editReorganize,
shortcut=QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_R),
icon="editReorganize",
tip="Reorganize music",
)
editSettingsAction = self._createAction(
text="Settings",
slot=self.editSettings,
shortcut=QKeySequence.Preferences,
icon="editSettings",
tip="Edit settings",
)
helpDocsAction = self._createAction(
text="Documentation",
slot=self.helpDocs,
shortcut=QKeySequence.HelpContents,
icon="helpDocs",
tip="Documentation",
)
helpAboutAction = self._createAction(
text="About",
slot=self.helpAbout,
shortcut=QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_F1),
icon="helpAbout",
tip="About",
)
toggleEditing = editingDock.toggleViewAction()
toggleEditing.setIcon(QIcon(":/toggleEditing.png"))
toggleFileNav = fileSystemDock.toggleViewAction()
toggleFileNav.setIcon(QIcon(":/toggleFileNav.png"))
# Menus
fileMenu = self.menuBar().addMenu("&File")
self._addActions(fileMenu, (fileAddSongAction, fileAddAlbumAction, fileSaveAction, fileQuitAction))
editMenu = self.menuBar().addMenu("&Edit")
self._addActions(editMenu, (editReorganizeAction, editRecognizeAction, editSettingsAction))
helpMenu = self.menuBar().addMenu("&Help")
self._addActions(helpMenu, (helpDocsAction, helpAboutAction))
# Toolbars
editToolbar = self.addToolBar("EditToolbar")
editToolbar.setObjectName("editToolbar")
self._addActions(editToolbar, (editRecognizeAction, editReorganizeAction))
toggleToolbar = self.addToolBar("ToggleToolbar")
toggleToolbar.setObjectName("toggleToolbar")
self._addActions(toggleToolbar, (toggleFileNav, toggleEditing))
# Settings
settings = QSettings()
if settings.contains("MainWindow/Geometry"):
self.restoreGeometry(settings.value("MainWindow/Geometry"))
if settings.contains("MainWindow/State"):
self.restoreState(settings.value("MainWindow/State"))
self.setWindowTitle("r3tagger")
# Final Layout
centralWidget = QWidget()
centralLayout = QVBoxLayout()
centralLayout.addWidget(self.albumView)
centralLayout.addLayout(self.buttonGroup)
centralWidget.setLayout(centralLayout)
self.setCentralWidget(centralWidget)
def _createAction(self, text, slot=None, shortcut=None, icon=None, tip=None, checkable=False, signal="triggered"):
action = QAction(text, self)
if icon is not None:
action.setIcon(QIcon(":/{0}.png".format(icon)))
if shortcut is not None:
action.setShortcut(shortcut)
if tip is not None:
action.setToolTip(tip)
action.setStatusTip(tip)
if slot is not None:
getattr(action, signal).connect(slot)
if checkable:
action.setCheckable(True)
return action
def _addActions(self, target, actions):
for action in actions:
if action is None:
target.addSeparator()
else:
target.addAction(action)
def _setDirty(self):
self.dirty = True
def _fixColumns(self, index, view, model):
for column in range(model.columnCount(index)):
view.resizeColumnToContents(column)
def closeEvent(self, event):
if self.dirty:
reply = QMessageBox.question(
self,
"r3tagger - Unsaved Changes",
"Save unsaved changes?",
QMessageBox.Yes | QMessageBox.No | QMessageBox.Cancel,
)
if reply == QMessageBox.Cancel:
event.ignore()
return None
elif reply == QMessageBox.Yes:
self.confirmChanges()
settings = QSettings()
settings.setValue("MainWindow/Geometry", self.saveGeometry())
settings.setValue("MainWindow/State", self.saveState())
def fixFileSystemColumns(self, index):
self._fixColumns(index, self.fileSystemView, self.fileSystemModel)
def fixAlbumViewColumns(self, index):
model = self.albumView.model()
self._fixColumns(index, self.albumView, model)
def updateAlbumModel(self, index):
path = self.fileSystemModel.fileInfo(index).absoluteFilePath()
self.addPath(path)
self.fixAlbumViewColumns(index)
def addPath(self, path):
if os.path.isfile(path):
track = controller.build_track(path)
containerAlbum = controller.album_from_tracks([track], u"Singles")
self.albumView.model().addAlbum(containerAlbum)
else:
for album in controller.build_albums(path, recursive=True):
self.albumView.model().addAlbum(album)
def updateEditing(self, index):
self.albumView.correctListingSelection(index)
selectedTracks = self.albumView.selectedTracks()
albumOfSingles = controller.album_from_tracks(selectedTracks)
selected = self.albumView.selectedAlbums()
selected.append(albumOfSingles)
tags = controller.find_shared_tags(*selected) if selected else {}
for tag, edit in self.tagsToAttribs.items():
if not tags:
self.clearEditing()
break
edit.setText(tags.get(tag, ""))
edit.setCursorPosition(0)
def confirmChanges(self):
tags = {}
for field, lineEdit in self.tagsToAttribs.items():
tag = lineEdit.text()
tags[field] = tag
view = self.albumView
for album in view.selectedAlbums():
controller.retag_album(album, tags)
for track in view.selectedTracks():
controller.retag_track(track, tags)
self.saveChanges()
def clearAlbumView(self):
model = self.albumView.model()
model.clear()
model.setHeaders()
def clearEditing(self):
for lineEdit in self.tagsToAttribs.values():
lineEdit.setText("")
def saveChanges(self):
for track in self.albumView.model():
if track.dirty:
track.saveChanges()
self.dirty = False
self.resetModel()
def cancelChanges(self):
for track in self.albumView.model():
track.reset()
self.resetModel()
def resetModel(self):
model = self.albumView.model()
model.beginResetModel()
self.clearEditing()
expanded = []
rowCount = model.rowCount(QModelIndex())
for row in range(rowCount):
index = model.index(row, 0, QModelIndex())
if self.albumView.isExpanded(index):
expanded.append(index)
model.endResetModel()
for expandedIndex in expanded:
self.albumView.setExpanded(expandedIndex, True)
def fileAddSong(self):
selectedFiles, selectedFilter = QFileDialog.getOpenFileNames(parent=self, caption="Add Songs")
for song in selectedFiles:
self.addPath(song)
def fileAddAlbum(self):
selectedDir = QFileDialog.getExistingDirectory(parent=self, caption="Add Album")
self.addPath(selectedDir)
def editRecognize(self):
pass
def editReorganize(self):
pass
def editSettings(self):
pass
def helpDocs(self):
pass
def helpAbout(self):
pass
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
self.dirty = False
# Models
# - Filesystem Model
self.fileSystemModel = QFileSystemModel()
# rootPath = QDesktopServices.storageLocation(
# QDesktopServices.HomeLocation)
# fileSystemRoot = self.fileSystemModel.setRootPath(rootPath)
fileSystemRoot = self.fileSystemModel.setRootPath(
"/home/ryan/Programming/Python/projects/r3tagger/r3tagger/tests"
)
# Views
# - Filesystem View
self.fileSystemView = QTreeView()
self.fileSystemView.setModel(self.fileSystemModel)
self.fileSystemView.setRootIndex(fileSystemRoot)
self.fileSystemView.doubleClicked.connect(self.updateAlbumModel)
self.fileSystemView.expanded.connect(self.fixFileSystemColumns)
self.fileSystemView.collapsed.connect(self.fixFileSystemColumns)
# - Album View
self.albumView = albumcollection.MusicCollectionView()
self.albumView.setSelectionMode(QAbstractItemView.MultiSelection)
self.albumView.clicked.connect(self.updateEditing)
self.albumView.expanded.connect(self.fixAlbumViewColumns)
self.albumView.collapsed.connect(self.fixAlbumViewColumns)
self.albumView.model().dataChanged.connect(self.fixAlbumViewColumns)
self.albumView.model().dataChanged.connect(self._setDirty)
model = self.albumView.model()
model.dataChanged.connect(self.updateEditing)
# Editing Group
self.editingGroup = QFormLayout()
self.lineArtist = QLineEdit()
self.lineAlbum = QLineEdit()
self.lineTitle = QLineEdit()
self.lineTrack = QLineEdit()
self.lineDate = QLineEdit()
self.lineGenre = QLineEdit()
self.editingGroup.addRow("Artist:", self.lineArtist)
self.editingGroup.addRow("Album:", self.lineAlbum)
self.editingGroup.addRow("Title:", self.lineTitle)
self.editingGroup.addRow("Track:", self.lineTrack)
self.editingGroup.addRow("Date:", self.lineDate)
self.editingGroup.addRow("Genre:", self.lineGenre)
self.tagsToAttribs = {
"artist": self.lineArtist,
"album": self.lineAlbum,
"title": self.lineTitle,
"tracknumber": self.lineTrack,
"date": self.lineDate,
"genre": self.lineGenre,
}
# Confirm / Cancel / Clear Group
self.buttonGroup = QHBoxLayout()
self.buttonGroup.addStretch()
confirm = QPushButton("Confirm")
confirm.clicked.connect(self.confirmChanges)
self.buttonGroup.addWidget(confirm)
cancel = QPushButton("Cancel")
cancel.clicked.connect(self.cancelChanges)
self.buttonGroup.addWidget(cancel)
clear = QPushButton("Clear")
clear.clicked.connect(self.clearAlbumView)
self.buttonGroup.addWidget(clear)
self.buttonGroup.addStretch()
# Statusbar
# status = self.statusBar()
# status.setSizeGripEnabled(False)
# status.showMessage("Ready", 5000)
# Docks
dockAllowed = Qt.LeftDockWidgetArea | Qt.RightDockWidgetArea
# - Filesystem Dock
fileSystemDock = QDockWidget("Navigate", self)
fileSystemDock.setObjectName("fileSystemDock")
fileSystemDock.setAllowedAreas(dockAllowed)
fileSystemDock.setWidget(self.fileSystemView)
self.addDockWidget(Qt.LeftDockWidgetArea, fileSystemDock)
# - Editing Dock
editingWidget = QWidget()
editingWidget.setLayout(self.editingGroup)
editingDock = QDockWidget("Editing", self)
editingDock.setObjectName("editingDock")
editingDock.setAllowedAreas(dockAllowed)
editingDock.setWidget(editingWidget)
self.addDockWidget(Qt.RightDockWidgetArea, editingDock)
# Actions
fileAddSongAction = self._createAction(
text="Add &Songs",
slot=self.fileAddSong,
shortcut=QKeySequence.Open,
icon="fileOpen",
tip="Add files (songs)",
)
fileAddAlbumAction = self._createAction(
text="Add &Album",
slot=self.fileAddAlbum,
shortcut=QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_O),
icon="fileOpen",
tip="Add directory (album)",
)
fileSaveAction = self._createAction(
text="&Save Changes",
slot=self.confirmChanges,
shortcut=QKeySequence.Save,
icon="fileSave",
tip="Save Changes",
)
fileQuitAction = self._createAction(
text="&Quit", slot=self.close, shortcut=QKeySequence.Quit, icon="fileQuit", tip="Quit Program"
)
editRecognizeAction = self._createAction(
text="&Recognize",
slot=self.editRecognize,
shortcut=QKeySequence(Qt.CTRL + Qt.Key_R),
icon="editRecognize",
tip="Recognize music",
)
editReorganizeAction = self._createAction(
text="Reorganize",
slot=self.editReorganize,
shortcut=QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_R),
icon="editReorganize",
tip="Reorganize music",
)
editSettingsAction = self._createAction(
text="Settings",
slot=self.editSettings,
shortcut=QKeySequence.Preferences,
icon="editSettings",
tip="Edit settings",
)
helpDocsAction = self._createAction(
text="Documentation",
slot=self.helpDocs,
shortcut=QKeySequence.HelpContents,
icon="helpDocs",
tip="Documentation",
)
helpAboutAction = self._createAction(
text="About",
slot=self.helpAbout,
shortcut=QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_F1),
icon="helpAbout",
tip="About",
)
toggleEditing = editingDock.toggleViewAction()
toggleEditing.setIcon(QIcon(":/toggleEditing.png"))
toggleFileNav = fileSystemDock.toggleViewAction()
toggleFileNav.setIcon(QIcon(":/toggleFileNav.png"))
# Menus
fileMenu = self.menuBar().addMenu("&File")
self._addActions(fileMenu, (fileAddSongAction, fileAddAlbumAction, fileSaveAction, fileQuitAction))
editMenu = self.menuBar().addMenu("&Edit")
self._addActions(editMenu, (editReorganizeAction, editRecognizeAction, editSettingsAction))
helpMenu = self.menuBar().addMenu("&Help")
self._addActions(helpMenu, (helpDocsAction, helpAboutAction))
# Toolbars
editToolbar = self.addToolBar("EditToolbar")
editToolbar.setObjectName("editToolbar")
self._addActions(editToolbar, (editRecognizeAction, editReorganizeAction))
toggleToolbar = self.addToolBar("ToggleToolbar")
toggleToolbar.setObjectName("toggleToolbar")
self._addActions(toggleToolbar, (toggleFileNav, toggleEditing))
# Settings
settings = QSettings()
if settings.contains("MainWindow/Geometry"):
self.restoreGeometry(settings.value("MainWindow/Geometry"))
if settings.contains("MainWindow/State"):
self.restoreState(settings.value("MainWindow/State"))
self.setWindowTitle("r3tagger")
# Final Layout
centralWidget = QWidget()
centralLayout = QVBoxLayout()
centralLayout.addWidget(self.albumView)
centralLayout.addLayout(self.buttonGroup)
centralWidget.setLayout(centralLayout)
self.setCentralWidget(centralWidget)
class FilterTab(QWidget):
"""This class is the GUI for creating filters for the FilterBox
module. This GUI will be added as a tab to the ModuleFrame
tab dialog.
"""
applySignal = Signal(Clause)
def __init__(self, parent, mframe, existing_filters):
"""Create a FilterTab with the given parent TabDialog, logical
parent ModuleFrame mframe, and existing_filters list of Clause
objects.
"""
super(FilterTab, self).__init__(parent)
self.mframe = mframe
self.parent = parent
self.attributes = self.mframe.agent.datatree.generateAttributeList()
self.clause_list = list()
self.clause_dict = dict()
# Right now we only look at the first passed in filter.
# TODO: At GUI to switch between existing filters
if existing_filters is not None and len(existing_filters) > 0:
for clause in existing_filters[0].conditions.clauses:
self.clause_list.append(str(clause))
self.clause_dict[str(clause)] = clause
self.clause_model = QStringListModel(self.clause_list)
layout = QVBoxLayout(self)
self.sidesplitter = QSplitter(Qt.Horizontal)
# You can only select one attribute at a time to build the
# filter clauses
self.data_view = QTreeView(self)
self.data_view.setModel(self.mframe.agent.datatree)
self.data_view.setDragEnabled(True)
self.data_view.setDropIndicatorShown(True)
self.data_view.expandAll()
self.sidesplitter.addWidget(self.data_view)
self.sidesplitter.setStretchFactor(1,1)
self.filter_widget = self.buildFilterWidget()
self.sidesplitter.addWidget(self.filter_widget)
self.sidesplitter.setStretchFactor(1,0)
layout.addWidget(self.sidesplitter)
# Apply buttons
buttonWidget = QWidget()
buttonLayout = QHBoxLayout(buttonWidget)
self.applyButton = QPushButton("Apply")
self.applyButton.clicked.connect(self.applyFilter)
self.closeButton = QPushButton("Apply & Close")
self.closeButton.clicked.connect(self.applyCloseFilter)
buttonLayout.addWidget(self.applyButton)
buttonLayout.addWidget(self.closeButton)
buttonWidget.setLayout(buttonLayout)
layout.addWidget(buttonWidget)
self.setLayout(layout)
def applyFilter(self):
"""Emits the applySignal with the Clause object currently
represented by this FilterTab.
"""
num_clauses = len(self.clause_list)
if num_clauses == 0:
self.applySignal.emit(None)
else:
self.applySignal.emit(Clause("and", *self.clause_dict.values()))
def applyCloseFilter(self):
"""Calls applyFilter and then closes the containing TabDialog."""
self.applyFilter()
self.parent.close()
def buildFilterWidget(self):
"""Creates the filter portion of the widget by laying out
the subwidgets for relations, workspace and existing
clauses.
"""
filter_widget = QWidget()
filter_layout = QVBoxLayout(filter_widget)
filter_layout.addWidget(self.buildRelationsWidget())
filter_layout.addItem(QSpacerItem(5,5))
filter_layout.addWidget(self.buildWorkFrame())
filter_layout.addItem(QSpacerItem(5,5))
filter_layout.addWidget(self.buildFilterListView())
filter_widget.setLayout(filter_layout)
return filter_widget
def buildFilterListView(self):
"""Creates the QListView that contains all of the basic Clause
objects.
"""
groupBox = QGroupBox("Clauses")
layout = QVBoxLayout(groupBox)
self.list_view = QListView(groupBox)
self.list_view.setModel(self.clause_model)
layout.addWidget(self.list_view)
layout.addItem(QSpacerItem(5,5))
self.delButton = QPushButton("Remove Selected Clause")
self.delButton.clicked.connect(self.deleteClause)
layout.addWidget(self.delButton)
groupBox.setLayout(layout)
return groupBox
def buildWorkFrame(self):
"""Creates the grouped set of widgets that allow users to build
basic Clause objects.
"""
groupBox = QGroupBox("Clause Workspace")
layout = QHBoxLayout(groupBox)
attributeCompleter = QCompleter(self.attributes)
attributeCompleter.setCompletionMode(QCompleter.InlineCompletion)
self.dropAttribute = DropLineEdit(self, self.mframe.agent.datatree, "",
attributeCompleter)
self.dropRelation = DropTextLabel("__")
self.dropValue = FilterValueLineEdit(groupBox,
self.mframe.agent.datatree, self.dropAttribute)
# Clear dropValue when dropAttribute changes
self.dropAttribute.textChanged.connect(self.dropValue.clear)
# Enter in dropValue works like addButton
self.dropValue.returnPressed.connect(self.addClause)
self.addButton = QPushButton("Add", groupBox)
self.addButton.clicked.connect(self.addClause)
layout.addWidget(self.dropAttribute)
layout.addItem(QSpacerItem(5,5))
layout.addWidget(self.dropRelation)
layout.addItem(QSpacerItem(5,5))
layout.addWidget(self.dropValue)
layout.addItem(QSpacerItem(5,5))
layout.addWidget(self.addButton)
groupBox.setLayout(layout)
return groupBox
def buildRelationsWidget(self):
"""Creates the set of draggable relations. These relations are
whatever is available in the relations dict of Table.
"""
relations_widget = QWidget()
layout = QHBoxLayout(relations_widget)
for relation in Table.relations:
layout.addWidget(DragTextLabel(relation))
relations_widget.setLayout(layout)
return relations_widget
def addClause(self):
"""Adds a basic Clause to the current filter."""
if self.dropRelation.text() in Table.relations \
and len(self.dropValue.text()) > 0 \
and len(self.dropAttribute.text()) > 0:
clause = Clause(self.dropRelation.text(),
TableAttribute(self.dropAttribute.text()),
self.dropValue.text())
# Guard double add
if str(clause) not in self.clause_dict:
self.clause_list.append(str(clause))
self.clause_dict[str(clause)] = clause
self.clause_model.setStringList(self.clause_list)
def deleteClause(self):
"""Removes the selected basic Clause objects from the current
filter.
"""
clause = self.clause_model.data(
self.list_view.selectedIndexes()[0], Qt.DisplayRole)
if clause is not None and clause in self.clause_list:
self.clause_list.remove(clause)
del self.clause_dict[clause]
self.clause_model.setStringList(self.clause_list)
def create_widget(self, parent=None):
self.tree_view = QTreeView(parent)
self.model = ListModel(self.data)
self.tree_view.setModel(self.model)
return self.tree_view
class AutoREView(idaapi.PluginForm):
ADDR_ROLE = QtCore.Qt.UserRole + 1
def __init__(self, data):
super(AutoREView, self).__init__()
self._data = data
self.tv = None
self._model = None
def Show(self):
return idaapi.PluginForm.Show(self, 'AutoRE', options=idaapi.PluginForm.FORM_PERSIST)
def OnCreate(self, form):
if HAS_PYSIDE:
self.parent = self.FormToPySideWidget(form)
else:
self.parent = self.FormToPyQtWidget(form)
self._idp_hooks = AutoReIDPHooks(self)
if not self._idp_hooks.hook():
print 'IDP_Hooks.hook() failed'
self.tv = QTreeView()
self.tv.setExpandsOnDoubleClick(False)
root_layout = QVBoxLayout(self.parent)
# self.le_filter = QLineEdit(self.parent)
# root_layout.addWidget(self.le_filter)
root_layout.addWidget(self.tv)
self.parent.setLayout(root_layout)
self._model = QtGui.QStandardItemModel()
self._init_model()
self.tv.setModel(self._model)
self.tv.setColumnWidth(0, 200)
self.tv.setColumnWidth(1, 300)
self.tv.header().setStretchLastSection(True)
self.tv.expandAll()
self.tv.doubleClicked.connect(self.on_navigate_to_method_requested)
# self.le_filter.textChanged.connect(self.on_filter_text_changed)
self.tv.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
self.tv.customContextMenuRequested.connect(self._tree_customContextMenuRequesssted)
rename_action = QAction('Rename...', self.tv)
rename_action.setShortcut('n')
rename_action.triggered.connect(self._tv_rename_action_triggered)
self.tv.addAction(rename_action)
# def __event_filter(self, source, event):
# if event.type() == QtCore.QEvent.
def _tree_customContextMenuRequesssted(self, pos):
idx = self.tv.indexAt(pos)
if not idx.isValid():
return
addr = idx.data(role=self.ADDR_ROLE)
if not addr:
return
name_idx = idx.sibling(idx.row(), 1)
old_name = name_idx.data()
menu = QMenu()
rename_action = menu.addAction('Rename `%s`...' % old_name)
rename_action.setShortcut('n')
action = menu.exec_(self.tv.mapToGlobal(pos))
if action == rename_action:
return self._rename_ea_requested(addr, name_idx)
def _tv_rename_action_triggered(self):
selected = self.tv.selectionModel().selectedIndexes()
if not selected:
return
idx = selected[0]
if not idx.isValid():
return
addr = idx.data(role=self.ADDR_ROLE)
if not addr:
return
name_idx = idx.sibling(idx.row(), 1)
if not name_idx.isValid():
return
return self._rename_ea_requested(addr, name_idx)
def _rename_ea_requested(self, addr, name_idx):
old_name = name_idx.data()
if idaapi.IDA_SDK_VERSION >= 700:
new_name = idaapi.ask_str(str(old_name), 0, 'New name:')
else:
new_name = idaapi.askstr(0, str(old_name), 'New name:')
if new_name is None:
return
self._rename(addr, new_name)
renamed_name = idaapi.get_ea_name(addr)
name_idx.model().setData(name_idx, renamed_name)
@classmethod
def _rename(cls, ea, new_name):
if not ea or ea == idaapi.BADADDR:
return
if idaapi.IDA_SDK_VERSION >= 700:
return idaapi.force_name(ea, new_name, idaapi.SN_NOCHECK)
return idaapi.do_name_anyway(ea, new_name, 0)
def OnClose(self, form):
if self._idp_hooks:
self._idp_hooks.unhook()
def _tv_init_header(self, model):
item_header = QtGui.QStandardItem("EA")
item_header.setToolTip("Address")
model.setHorizontalHeaderItem(0, item_header)
item_header = QtGui.QStandardItem("Function name")
model.setHorizontalHeaderItem(1, item_header)
item_header = QtGui.QStandardItem("API called")
model.setHorizontalHeaderItem(2, item_header)
# noinspection PyMethodMayBeStatic
def _tv_make_tag_item(self, name):
rv = QtGui.QStandardItem(name)
rv.setEditable(False)
return [rv, QtGui.QStandardItem(), QtGui.QStandardItem()]
def _tv_make_ref_item(self, tag, ref):
ea_item = QtGui.QStandardItem(('%0' + get_addr_width() + 'X') % ref['ea'])
ea_item.setEditable(False)
ea_item.setData(ref['ea'], self.ADDR_ROLE)
name_item = QtGui.QStandardItem(ref['name'])
name_item.setEditable(False)
name_item.setData(ref['ea'], self.ADDR_ROLE)
apis = ', '.join(ref['tags'][tag])
api_name = QtGui.QStandardItem(apis)
api_name.setEditable(False)
api_name.setData(ref['ea'], self.ADDR_ROLE)
api_name.setToolTip(apis)
return [ea_item, name_item, api_name]
def _init_model(self):
self._model.clear()
root_node = self._model.invisibleRootItem()
self._tv_init_header(self._model)
for tag, refs in self._data.items():
item_tag_list = self._tv_make_tag_item(tag)
item_tag = item_tag_list[0]
root_node.appendRow(item_tag_list)
for ref in refs:
ref_item_list = self._tv_make_ref_item(tag, ref)
item_tag.appendRow(ref_item_list)
def on_navigate_to_method_requested(self, index):
addr = index.data(role=self.ADDR_ROLE)
if addr is not None:
idaapi.jumpto(addr)
def __init__(self, data):
self.data = data
self.treeView = QTreeView()
self.treeView.setModel(DictModel(self.data))
result, index = model.find_GivenName(name)
if result:
if index:
tv.setCurrentIndex(index)
return
tv.clearSelection()
app = QApplication([])
model = treeModel()
dialog = QDialog()
dialog.setMinimumSize(300, 150)
layout = QVBoxLayout(dialog)
tv = QTreeView(dialog)
tv.setModel(model)
tv.setAlternatingRowColors(True)
layout.addWidget(tv)
label = QLabel("Search for the following person")
layout.addWidget(label)
buts = []
frame = QFrame(dialog)
layout2 = QHBoxLayout(frame)
for person in model.people:
but = QPushButton(person.fname, frame)
buts.append(but)
layout2.addWidget(but)
def __init__(self, mw, parent=None):
"""
Default class constructor.
:param `mw`: The application's MainWindow instance.
:type `mw`: `QMainWindow`_
:param `parent`: Pointer to a parent widget instance.
:type `parent`: `QWidget`_
"""
super(LayerManager, self).__init__(parent)
tr = self.tr
self.layerModel = layerModel = QStandardItemModel(0, 8, self)
self.layerModelSorted = layerModelSorted = QSortFilterProxyModel()
layerModelSorted.setDynamicSortFilter(True)
layerModelSorted.setSourceModel(layerModel)
self.treeView = treeView = QTreeView()
treeView.setRootIsDecorated(False)
treeView.setAlternatingRowColors(True)
treeView.setModel(layerModelSorted)
treeView.setSortingEnabled(True)
treeView.sortByColumn(0, Qt.AscendingOrder)
mainLayout = QVBoxLayout()
mainLayout.addWidget(treeView)
self.setLayout(mainLayout)
self.setWindowTitle(tr("Layer Manager"))
self.setMinimumSize(750, 550)
## layerModel.setHeaderData(0, Qt.Horizontal, self.tr("Name"))
## layerModel.setHeaderData(1, Qt.Horizontal, self.tr("Visible"))
## layerModel.setHeaderData(2, Qt.Horizontal, self.tr("Frozen"))
## layerModel.setHeaderData(3, Qt.Horizontal, self.tr("Z Value"))
## layerModel.setHeaderData(4, Qt.Horizontal, self.tr("Color"))
## layerModel.setHeaderData(5, Qt.Horizontal, self.tr("Linetype"))
## layerModel.setHeaderData(6, Qt.Horizontal, self.tr("Lineweight"))
## layerModel.setHeaderData(7, Qt.Horizontal, self.tr("Print"))
localMeth = layerModel.setHeaderData
localQtHorizontal = Qt.Horizontal
[
localMeth(i, localQtHorizontal, tr(strng))
for i, strng in ((0, "Name"), (1, "Visible"), (2, "Frozen"),
(3, "Z Value"), (4, "Color"), (5, "Linetype"),
(6, "Lineweight"), (7, "Print"))
]
## addLayer = self.addLayer # local optimization
## addLayer("0", True, False, 0.0, qRgb(0, 0, 0), "Continuous", "Default", True)
## addLayer("1", True, False, 1.0, qRgb(0, 0, 0), "Continuous", "Default", True)
## addLayer("2", True, False, 2.0, qRgb(0, 0, 0), "Continuous", "Default", True)
## addLayer("3", True, False, 3.0, qRgb(0, 0, 0), "Continuous", "Default", True)
## addLayer("4", True, False, 4.0, qRgb(0, 0, 0), "Continuous", "Default", True)
## addLayer("5", True, False, 5.0, qRgb(0, 0, 0), "Continuous", "Default", True)
## addLayer("6", True, False, 6.0, qRgb(0, 0, 0), "Continuous", "Default", True)
## addLayer("7", True, False, 7.0, qRgb(0, 0, 0), "Continuous", "Default", True)
## addLayer("8", True, False, 8.0, qRgb(0, 0, 0), "Continuous", "Default", True)
## addLayer("9", True, False, 9.0, qRgb(0, 0, 0), "Continuous", "Default", True)
addLayer = self.addLayer
[
addLayer("%s" % i, True, False, float(i), qRgb(0, 0, 0),
"Continuous", "Default", True) for i in range(0, 10)
]
## for i in range(0, layerModel.columnCount()):
## treeView.resizeColumnToContents(i)
localtreeViewMeth = treeView.resizeColumnToContents
[localtreeViewMeth(i) for i in range(0, layerModel.columnCount())]
QApplication.setOverrideCursor(Qt.ArrowCursor)
class getFilesDlg(QDialog):
# sendPaths is a signal emitted by getPaths containing a list of file paths from
# the users selection via this dialog.
sendPaths = Signal(list)
def __init__(self, parent=None):
super(getFilesDlg, self).__init__(parent)
self.setMinimumSize(500,500)
self.fileDlgPaths = []
layout = QVBoxLayout()
self.btnBox = QDialogButtonBox(QDialogButtonBox.Ok |
QDialogButtonBox.Cancel)
self.btnBox.accepted.connect(self.getPaths)
self.btnBox.rejected.connect(self.close)
self.fsModel = QFileSystemModel()
self.treeView = QTreeView()
self.treeView.setSelectionMode(QTreeView.ExtendedSelection)
self.treeView.setModel(self.fsModel)
self.treeView.setColumnWidth(0, 361)
self.treeView.setColumnHidden(1, True)
self.treeView.setColumnHidden(3, True)
layout.addWidget(self.treeView)
layout.addWidget(self.btnBox)
self.setLayout(layout)
self.fsModel.setRootPath(environ['HOMEPATH'])
# self.treeView.setRootIndex(self.fsModel.index("\\"))
self.treeView.expand(self.treeView.rootIndex())
def getPaths(self):
# For some reason duplicates were being returned when they weren't supposed to.
# This obtains the selected files from the dialog and only returns individual
# paths.
indexes = self.treeView.selectedIndexes()
if indexes:
self.fileDlgPaths = []
for i in indexes:
# Possible permission error occuring here
# unable to replicate at this time
path = self.fsModel.filePath(i)
if path not in self.fileDlgPaths:
self.fileDlgPaths.append(path.replace('/','\\'))
self.close() # To close the dialog on an accept signal
self.sendPaths.emit(self.fileDlgPaths)
class BrowserWindow(QMainWindow):
MO_ROLE = Qt.UserRole+1
def __init__(self, conn):
super(BrowserWindow, self).__init__()
self._conn = conn
self._resolver = AsyncResolver()
self._resolver.object_resolved.connect(self._data_resolved)
self._resolver.start()
self._init_models()
self._init_gui()
self._init_data()
self._init_connections()
def __del__(self):
self._resolver.stop_work()
self._resolver.terminate()
def _init_models(self):
self._hierarchy_model = QStandardItemModel()
self._hierarchy_model.setColumnCount(2)
self._hierarchy_model.setHorizontalHeaderLabels(['class', 'dn'])
self._details_model = QStandardItemModel()
self._details_model.setColumnCount(2)
self._details_model.setHorizontalHeaderLabels(['Property', 'Value'])
def _init_gui(self):
self._widget = QSplitter(self, Qt.Horizontal)
self._hierarchy_view = QTreeView(self._widget)
self._details_view = QTableView(self._widget)
self._widget.addWidget(self._hierarchy_view)
self._widget.addWidget(self._details_view)
self._widget.setStretchFactor(0, 2)
self._widget.setStretchFactor(1, 1)
self.setCentralWidget(self._widget)
self._hierarchy_view.setModel(self._hierarchy_model)
self._details_view.setModel(self._details_model)
self._hierarchy_view.expanded.connect(self._mo_item_expand)
def _init_data(self):
item = self._row_for_mo(self._conn.resolve_dn(''))
self._hierarchy_model.insertRow(0, item)
def _init_connections(self):
self.connect(self._resolver,
SIGNAL('object_resolved(QVariant)'),
self,
SLOT('_data_resolved(QVariant)'))
self._hierarchy_view.activated.connect(self._item_activated)
#self.connect(self._hierarchy_view.selectionModel(),
# SIGNAL('currentChanged(QModelIndex,QModelIndex)'),
# self,
# SLOT('_current_changed(QModelIndex, QModelIndex)'))
self.connect(self._hierarchy_view.selectionModel(),
SIGNAL('activated(QModelIndex)'),
self,
SLOT('_item_activated(QModelIndex)'))
def _row_for_mo(self, mo):
row = [QStandardItem(mo.ucs_class), QStandardItem(mo.dn)]
for item in row:
item.setEditable(False)
row[0].appendColumn([QStandardItem('Loading...')])
row[0].setData(mo, self.MO_ROLE)
return row
def _add_mo_in_tree(self, mo, index=QtCore.QModelIndex()):
item = None
if index.isValid():
item = self._hierarchy_model.itemFromIndex(index)
else:
item = self._get_item_for_dn(self._parent_dn(mo.dn))
if item:
item.appendColumn([self._row_for_mo(mo)[0]])
self.auto_width()
def _add_mos_in_tree(self, mos, index=QtCore.QModelIndex()):
item = None
if index.isValid():
item = self._hierarchy_model.itemFromIndex(index)
else:
if not mos:
return
item = self._get_item_for_dn(self._parent_dn(mos[0].dn))
while item.columnCount():
item.removeColumn(0)
items = map(self._row_for_mo, mos)
if items:
for x in xrange(len(items[0])):
item.appendColumn([row[x] for row in items])
self.auto_width()
@staticmethod
def _parent_dn(dn):
parent_dn, _, rn = dn.rpartition('/')
return parent_dn
def _get_item_for_dn(self, dn):
parent_dn = dn
items = self._hierarchy_model.findItems(parent_dn, column=1)
if items:
return self._hierarchy_model.item(items[0].row())
return None
@QtCore.Slot('_data_resolved(QVariant)')
def _data_resolved(self, datav):
print 'Data resolved: ', datav
index, data = datav
if isinstance(data, UcsmObject):
self._add_mo_in_tree(data, index=index)
else:
self._add_mos_in_tree(data, index=index)
@QtCore.Slot('_current_changed(QModelIndex,QModelIndex)')
def _current_changed(self, curr, prev):
self._item_activated(curr)
@QtCore.Slot('_item_activated(QModelIndex)')
def _item_activated(self, index):
print 'Activated: %s data %s' % (index, index.data(self.MO_ROLE))
if index.sibling(0, 0).isValid():
index = index.sibling(0, 0)
data = index.data(self.MO_ROLE)
self.set_detail_object(data)
def _mo_item_expand(self, index):
obj = index.data(self.MO_ROLE)
print 'Expanded object: %s' % obj
try:
self._resolver.add_task(lambda: (index,
self._conn.resolve_children(obj.dn)))
except (KeyError, AttributeError):
QtGui.QMessageBox.critical(0, 'Error', 'Object does not have dn')
def auto_width(self):
for view in [self._hierarchy_view, self._details_view]:
for col in xrange(view.model().columnCount()):
view.resizeColumnToContents(col)
def set_detail_object(self, object):
self._details_model.removeRows(0, self._details_model.rowCount())
for k, v in object.attributes.iteritems():
row = [QStandardItem(k), QStandardItem(v)]
for item in row:
item.setEditable(False)
self._details_model.appendRow(row)
self.auto_width()
class AutoREView(idaapi.PluginForm):
ADDR_ROLE = QtCore.Qt.UserRole + 1
def __init__(self, data):
super(AutoREView, self).__init__()
self._data = data
def Show(self):
return idaapi.PluginForm.Show(self,
'AutoRE',
options=idaapi.PluginForm.FORM_PERSIST)
def OnCreate(self, form):
# if HAS_PYSIDE:
# self.parent = self.FormToPySideWidget(form)
# else:
self.parent = self.FormToPyQtWidget(form)
self.tv = QTreeView()
self.tv.setExpandsOnDoubleClick(False)
root_layout = QVBoxLayout(self.parent)
# self.le_filter = QLineEdit(self.parent)
# root_layout.addWidget(self.le_filter)
root_layout.addWidget(self.tv)
self.parent.setLayout(root_layout)
self._model = QtGui.QStandardItemModel()
self._init_model()
self.tv.setModel(self._model)
self.tv.setColumnWidth(0, 200)
self.tv.setColumnWidth(1, 300)
self.tv.header().setStretchLastSection(True)
self.tv.expandAll()
self.tv.doubleClicked.connect(self.on_navigate_to_method_requested)
# self.le_filter.textChanged.connect(self.on_filter_text_changed)
def OnClose(self, form):
# print 'TODO: OnClose(): clear the pointer to form in the plugin'
pass
def _tv_init_header(self, model):
item_header = QtGui.QStandardItem("EA")
item_header.setToolTip("Address")
model.setHorizontalHeaderItem(0, item_header)
item_header = QtGui.QStandardItem("Function name")
model.setHorizontalHeaderItem(1, item_header)
item_header = QtGui.QStandardItem("API called")
model.setHorizontalHeaderItem(2, item_header)
def _tv_make_tag_item(self, name):
rv = QtGui.QStandardItem(name)
rv.setEditable(False)
return [rv, QtGui.QStandardItem(), QtGui.QStandardItem()]
def _tv_make_ref_item(self, tag, ref):
ea_item = QtGui.QStandardItem(
('%0' + get_addr_width() + 'X') % ref['ea'])
ea_item.setEditable(False)
ea_item.setData(ref['ea'], self.ADDR_ROLE)
name_item = QtGui.QStandardItem(ref['name'])
name_item.setEditable(False)
name_item.setData(ref['ea'], self.ADDR_ROLE)
apis = ', '.join(ref['tags'][tag])
api_name = QtGui.QStandardItem(apis)
api_name.setEditable(False)
api_name.setData(ref['ea'], self.ADDR_ROLE)
api_name.setToolTip(apis)
return [ea_item, name_item, api_name]
def _init_model(self):
self._model.clear()
root_node = self._model.invisibleRootItem()
self._tv_init_header(self._model)
for tag, refs in self._data.items():
item_tag_list = self._tv_make_tag_item(tag)
item_tag = item_tag_list[0]
root_node.appendRow(item_tag_list)
for ref in refs:
ref_item_list = self._tv_make_ref_item(tag, ref)
item_tag.appendRow(ref_item_list)
def on_navigate_to_method_requested(self, index):
addr = index.data(role=self.ADDR_ROLE)
if addr is not None:
idaapi.jumpto(addr)
result, index = model.find_GivenName(name)
if result:
if index:
tv.setCurrentIndex(index)
return
tv.clearSelection()
app = QApplication([])
model = treeModel()
dialog = QDialog()
dialog.setMinimumSize(300, 150)
layout = QVBoxLayout(dialog)
tv = QTreeView(dialog)
tv.setModel(model)
tv.setAlternatingRowColors(True)
layout.addWidget(tv)
label = QLabel("Search for the following person")
layout.addWidget(label)
buts = []
frame = QFrame(dialog)
layout2 = QHBoxLayout(frame)
for person in model.people:
but = QPushButton(person.fname, frame)
buts.append(but)
layout2.addWidget(but)
def selectionChanged(self, selected, deselected):
QTreeView.selectionChanged(self, selected, deselected)
self.selection.emit(selected, deselected)
def selectionChanged(self, selected, deselected):
QTreeView.selectionChanged(self, selected, deselected)
self.selection.emit(selected, deselected)
