def launch_daemon(self):
self.ui.launch_btn.hide()
self.hide()
self.ui.account_tab.setEnabled(True)
number_of_storages = self.storage_manager.get_number_of_storages()
self.event_handler = DemerioConductor(Mapping(demerio_dir, config_file), FileFec(PARTS_NEEDED, number_of_storages), self.storage_manager, ignore_patterns = IGNORE_PATTERNS)
self.event_handler.conductor_exception.connect(self.tray.conductor_problem)
self.event_handler.event_started.connect(self.tray.event_started)
self.event_handler.event_finished.connect(self.tray.event_finished)
self.event_handler.reconstruct_started.connect(self.init_progress_bar)
self.event_handler.reconstruct_finished.connect(self.end_progress_bar)
self.event_handler.reconstruct_update.connect(self.update_progress_bar)
observer = Observer()
self.daemon = DemerioDaemon(self.event_handler, observer, demerio_dir, should_exit=False)
self.daemon.start()
self.open_demerio_folder()
class MainWindow(QDialog):
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
self.ui = uic.loadUi(ui_full_path('main_window.ui'), self)
self.ui.label_version.setText("Demerio version %s" % (get_versions()['version'],))
self.ui.progress_bar.setVisible(False)
self.ui.tabWidget.setCurrentIndex(0)
self.ui.launch_btn.clicked.connect(self.launch_daemon)
self.ui.output_btn.clicked.connect(self.validate)
self.tray = SystemTray(self)
self.tray.show()
self.create_tray_actions()
self.init_storage_manager()
self.add_clouds_to_view()
self.daemon = None
self.check_if_already_connected()
def create_tray_actions(self):
self.tray.add_action("Open Demerio Folder", self.open_demerio_folder)
self.tray.add_action("Preference", self.show)
self.tray.add_separator()
self.tray.add_action("Quit", self.quit_app)
"""
should be done in cloud widget class with a proper model and view
"""
def init_storage_manager(self):
self.storage_manager = StorageManager()
drive_api = GoogleDriveAPI(demerio_config_dir)
self.storage_manager.add_storage("google_drive", drive_api)
dropbox = DropboxAPI(demerio_config_dir)
self.storage_manager.add_storage("dropbox", dropbox)
box = BoxAPI(demerio_config_dir)
self.storage_manager.add_storage("box", box)
def add_clouds_to_view(self):
self.ui.listWidget.setResizeMode(QListView.Adjust)
for cloud_name in self.storage_manager.map_of_storage.keys():
cloud_widget = CloudWidget(cloud_name, self)
cloud_widget.pushed.connect(self.authenticate)
item = QListWidgetItem()
item.setSizeHint(cloud_widget.sizeHint())
self.ui.listWidget.addItem(item)
self.ui.listWidget.setItemWidget(item, cloud_widget)
@pyqtSlot()
def authenticate(self):
cloud_storage = self.storage_manager.map_of_storage[self.sender().name]
cloud_storage.authorize()
if cloud_storage.is_connected():
self.sender().ui.cloudAction.setEnabled(False)
self.update_launch_btn_state()
@pyqtSlot()
def open_demerio_folder(self):
QDesktopServices.openUrl(QUrl.fromLocalFile(demerio_dir))
def update_launch_btn_state(self):
for cloud in self.storage_manager.map_of_storage.values():
if not cloud.is_connected():
return False
self.ui.launch_btn.setEnabled(True)
self.ui.label.setText("")
@pyqtSlot()
def launch_daemon(self):
self.ui.launch_btn.hide()
self.hide()
self.ui.account_tab.setEnabled(True)
number_of_storages = self.storage_manager.get_number_of_storages()
self.event_handler = DemerioConductor(Mapping(demerio_dir, config_file), FileFec(PARTS_NEEDED, number_of_storages), self.storage_manager, ignore_patterns = IGNORE_PATTERNS)
self.event_handler.conductor_exception.connect(self.tray.conductor_problem)
self.event_handler.event_started.connect(self.tray.event_started)
self.event_handler.event_finished.connect(self.tray.event_finished)
self.event_handler.reconstruct_started.connect(self.init_progress_bar)
self.event_handler.reconstruct_finished.connect(self.end_progress_bar)
self.event_handler.reconstruct_update.connect(self.update_progress_bar)
observer = Observer()
self.daemon = DemerioDaemon(self.event_handler, observer, demerio_dir, should_exit=False)
self.daemon.start()
self.open_demerio_folder()
@pyqtSlot(int)
def init_progress_bar(self, number_of_files):
self.ui.progress_bar.setVisible(True)
self.ui.progress_bar.setMaximum(number_of_files)
self.ui.progress_bar.setTextVisible(True)
@pyqtSlot(int)
def update_progress_bar(self, value):
self.ui.progress_bar.setValue(value)
@pyqtSlot()
def end_progress_bar(self):
self.ui.progress_bar.setVisible(False)
self.ui.start_label.setText("")
@pyqtSlot()
def validate(self):
self.dir_selected = QFileDialog.getExistingDirectory(self, "Select Directory")
self.ui.start_label.setText("Your datas will be reconstructed in : %s directory" %(self.dir_selected,))
self.event_handler.reconstruct_dir(self.dir_selected)
@pyqtSlot()
def quit_app(self):
if self.daemon is not None:
self.daemon.should_exit = True
self.daemon.join()
QCoreApplication.instance().quit()
def check_if_already_connected(self):
QApplication.processEvents()
for index in range(self.ui.listWidget.count()):
QApplication.processEvents()
cloud_widget = self.ui.listWidget.itemWidget(self.ui.listWidget.item(index))
if self.storage_manager.map_of_storage[cloud_widget.name].is_connected():
cloud_widget.ui.cloudAction.setEnabled(False)
self.update_launch_btn_state()
if self.ui.launch_btn.isEnabled():
self.launch_daemon()
else:
self.show()