def __init__(self):
super().__init__()
self._translate = QCoreApplication.translate
self.containing_folder = ''
self.preferences_window = Ui_Settings()
self.preferences_window.setupUi(self)
self.preferences_window.changeFolderButton_2.clicked.connect(self.update_sync_nodes)
self.preferences_window.pushButton.clicked.connect(self.sync_all)
self.preferences_window.pushButton_2.clicked.connect(self.sync_none)
self.tree_items = []
self.checked_items = []
self.setup_slots()
self._executor = QtCore.QThread()
self.node_fetcher = NodeFetcher()
class Preferences(QDialog):
"""
This class is a wrapper for the Ui_Preferences and its controls
"""
GENERAL = 0
OSF = 1
ABOUT = 2
PROJECT_NAME_COLUMN = 1
PROJECT_SYNC_COLUMN = 0
preferences_closed_signal = pyqtSignal()
containing_folder_updated_signal = pyqtSignal((str,))
def __init__(self):
super().__init__()
self._translate = QCoreApplication.translate
self.containing_folder = ''
self.preferences_window = Ui_Settings()
self.preferences_window.setupUi(self)
self.preferences_window.changeFolderButton_2.clicked.connect(self.update_sync_nodes)
self.preferences_window.pushButton.clicked.connect(self.sync_all)
self.preferences_window.pushButton_2.clicked.connect(self.sync_none)
self.tree_items = []
self.checked_items = []
self.setup_slots()
self._executor = QtCore.QThread()
self.node_fetcher = NodeFetcher()
def get_guid_list(self):
guid_list = []
for tree_item, node_id in self.tree_items:
if tree_item.checkState(self.PROJECT_SYNC_COLUMN) == Qt.Checked:
guid_list.append(node_id)
return guid_list
def closeEvent(self, event):
guid_list = self.get_guid_list()
if guid_list != self.checked_items:
reply = QMessageBox()
reply.setText('Unsaved changes')
reply.setIcon(QMessageBox.Warning)
reply.setInformativeText('You have unsaved changes to your synced projects.\n\n '
'Please review your changes and press \'update\' if you would like to save them. \n\n '
'Are you sure you would like to leave without saving? \n')
default = reply.addButton('Exit without saving', QMessageBox.YesRole)
reply.addButton('Review changes', QMessageBox.NoRole)
reply.setDefaultButton(default)
if reply.exec_() != 0:
return event.ignore()
try:
user = session.query(User).filter(User.logged_in).one()
except SQLAlchemyError:
pass
else:
self.preferences_closed_signal.emit()
self.reset_tree_widget()
event.accept()
def alerts_changed(self):
if self.preferences_window.desktopNotifications.isChecked():
AlertHandler.show_alerts = True
else:
AlertHandler.show_alerts = False
def startup_changed(self):
# todo: probably should give notification to show that this setting has been changed.
if self.preferences_window.startOnStartup.isChecked():
# todo: make it so that this application starts on login
# self.settings = QSettings(RUN_PATH, QSettings.NativeFormat)
pass
else:
# todo: make it so that this application does NOT start on login
pass
def set_containing_folder(self):
new_containing_folder = QFileDialog.getExistingDirectory(self, "Choose where to place OSF folder")
osf_path = os.path.join(new_containing_folder, "OSF")
if new_containing_folder == "":
# cancel, closed, or no folder chosen
return
elif not os.path.exists(osf_path):
os.makedirs(osf_path)
elif os.path.isfile(osf_path):
# FIXME: Consolidate redundant messages
AlertHandler.warn(
"An OSF file exists where you would like to create the OSF folder. Delete it, or choose a different location")
logging.warning("An OSF file exists where you would like to create the OSF folder.")
return
user = session.query(User).filter(User.logged_in).one()
user.osf_local_folder_path = os.path.join(osf_path)
self.preferences_window.containingFolderTextEdit.setText(self._translate("Preferences", self.containing_folder))
self.open_window(tab=Preferences.GENERAL) # todo: dynamically update ui????
self.containing_folder_updated_signal.emit(new_containing_folder)
def update_sync_nodes(self):
user = session.query(User).filter(User.logged_in).one()
guid_list = self.get_guid_list()
# FIXME: This needs a try-except block but is waiting on a preferences refactor to be merged
user.guid_for_top_level_nodes_to_sync = guid_list
save(session, user)
self.checked_items = guid_list
self.close()
def sync_all(self):
for tree_item, node_id in self.tree_items:
tree_item.setCheckState(self.PROJECT_SYNC_COLUMN, Qt.Checked)
def sync_none(self):
for tree_item, node_id in self.tree_items:
tree_item.setCheckState(self.PROJECT_SYNC_COLUMN, Qt.Unchecked)
def open_window(self, tab=GENERAL):
if self.isVisible():
self.preferences_window.tabWidget.setCurrentIndex(tab)
self.selector(tab)
else:
self.preferences_window.tabWidget.setCurrentIndex(tab)
self.selector(tab)
self.show()
self.raise_()
self.activateWindow()
def selector(self, selected_index):
if selected_index == self.GENERAL:
user = session.query(User).filter(User.logged_in).one()
containing_folder = os.path.dirname(user.osf_local_folder_path)
self.preferences_window.containingFolderTextEdit.setText(self._translate("Preferences", containing_folder))
elif selected_index == self.OSF:
user = session.query(User).filter(User.logged_in).one()
self.preferences_window.label.setText(self._translate("Preferences", user.full_name))
self._executor = QtCore.QThread()
self.node_fetcher = NodeFetcher()
self.preferences_window.treeWidget.setCursor(QtCore.Qt.BusyCursor)
self.node_fetcher.finished[list].connect(self.populate_item_tree)
self.node_fetcher.moveToThread(self._executor)
self._executor.started.connect(self.node_fetcher.fetch)
self._executor.start()
def reset_tree_widget(self):
self.tree_items.clear()
self.checked_items.clear()
self.preferences_window.treeWidget.clear()
@QtCore.pyqtSlot(list)
def populate_item_tree(self, nodes):
self.reset_tree_widget()
_translate = QCoreApplication.translate
try:
user = session.query(User).filter(User.logged_in).one()
except SQLAlchemyError:
return
for node in nodes:
tree_item = QTreeWidgetItem(self.preferences_window.treeWidget)
tree_item.setCheckState(self.PROJECT_SYNC_COLUMN, Qt.Unchecked)
tree_item.setText(self.PROJECT_NAME_COLUMN, _translate("Preferences", path.make_folder_name(node.name, node_id=node.id)))
if node.id in user.guid_for_top_level_nodes_to_sync:
tree_item.setCheckState(self.PROJECT_SYNC_COLUMN, Qt.Checked)
if node.id not in self.checked_items:
self.checked_items.append(node.id)
self.tree_items.append((tree_item, node.id))
self.preferences_window.treeWidget.resizeColumnToContents(self.PROJECT_SYNC_COLUMN)
self.preferences_window.treeWidget.resizeColumnToContents(self.PROJECT_NAME_COLUMN)
self.preferences_window.treeWidget.unsetCursor()
def setup_slots(self):
self.preferences_window.tabWidget.currentChanged.connect(self.selector)