def __init__(self):
super().__init__()
# settings
self.app_name = "OSFOffline"
self.app_author = "COS"
# views
self.start_screen = StartScreen()
self.tray = SystemTray()
self.preferences = Preferences()
AlertHandler.setup_alerts(self.tray.tray_icon,
self.tray.tray_alert_signal)
# connect all signal-slot pairs
self.setup_connections()
self.background_worker = BackgroundWorker()
def __init__(self):
super().__init__()
# settings
self.app_name = "OSFOffline"
self.app_author = "COS"
# views
self.start_screen = StartScreen()
self.tray = SystemTray()
self.preferences = Preferences()
AlertHandler.setup_alerts(self.tray.tray_icon, self.tray.tray_alert_signal)
# connect all signal-slot pairs
self.setup_connections()
self.background_worker = BackgroundWorker()
class OSFApp(QDialog):
login_signal = pyqtSignal()
start_tray_signal = pyqtSignal()
containing_folder_updated_signal = pyqtSignal()
def __init__(self):
super().__init__()
# settings
self.app_name = "OSFOffline"
self.app_author = "COS"
# views
self.start_screen = StartScreen()
self.tray = SystemTray()
self.preferences = Preferences()
AlertHandler.setup_alerts(self.tray.tray_icon,
self.tray.tray_alert_signal)
# connect all signal-slot pairs
self.setup_connections()
self.background_worker = BackgroundWorker()
def setup_connections(self):
# [ (signal, slot) ]
signal_slot_pairs = [
# system tray
(self.tray.open_osf_folder_action.triggered,
self.tray.open_osf_folder),
(self.tray.launch_osf_action.triggered, self.tray.start_osf),
(self.tray.sync_now_action.triggered, self.sync_now),
# (self.tray.currently_synching_action.triggered, self.controller.currently_synching),
(self.tray.preferences_action.triggered, self.open_preferences),
(self.tray.about_action.triggered, self.start_about_screen),
(self.tray.quit_action.triggered, self.quit),
(self.tray.tray_alert_signal, self.tray.update_currently_synching),
# main events
(self.login_signal, self.start_screen.open_window),
(self.start_tray_signal, self.tray.start),
# preferences
(self.preferences.preferences_window.desktopNotifications.
stateChanged, self.preferences.alerts_changed),
(self.preferences.preferences_window.startOnStartup.stateChanged,
self.preferences.startup_changed),
(self.preferences.preferences_window.changeFolderButton.clicked,
self.preferences.set_containing_folder),
(self.preferences.preferences_closed_signal, self.resume),
(self.preferences.preferences_window.accountLogOutButton.clicked,
self.logout),
(self.preferences.containing_folder_updated_signal,
self.tray.set_containing_folder),
# (self.preferences.containing_folder_updated_signal, self.preferences.update_containing_folder_text_box),
# start screen
(self.start_screen.done_logging_in_signal, self.start),
(self.start_screen.quit_application_signal, self.quit),
]
for signal, slot in signal_slot_pairs:
signal.connect(slot)
def _can_restart_background_worker(self):
try:
user = self.get_current_user()
except:
return False
if not user.logged_in:
return False
if not os.path.isdir(user.osf_local_folder_path):
return False
if not self.background_worker:
return False
if self.background_worker.is_alive():
return False
return True
def start(self):
logger.debug('Start in main called.')
try:
user = session.query(User).filter(User.logged_in).one()
except MultipleResultsFound:
session.query(User).delete()
self.login_signal.emit()
return
except NoResultFound:
self.login_signal.emit()
return
try:
# Simple request to ensure user logged in with valid oauth_token
user = asyncio.get_event_loop().run_until_complete(
AuthClient().populate_user_data(user))
except AuthError as e:
logging.exception(e.message)
self.login_signal.emit()
containing_folder = os.path.dirname(user.osf_local_folder_path)
while not validate_containing_folder(containing_folder):
logger.warning(
'Invalid containing folder: {}'.format(containing_folder))
AlertHandler.warn(
'Invalid containing folder. Please choose another.')
containing_folder = os.path.abspath(
self.set_containing_folder_initial())
user.osf_local_folder_path = os.path.join(containing_folder, 'OSF')
save(session, user)
self.tray.set_containing_folder(containing_folder)
if not os.path.isdir(user.osf_local_folder_path):
os.makedirs(user.osf_local_folder_path)
self.start_tray_signal.emit()
logger.debug('starting background worker from main.start')
self.background_worker = BackgroundWorker()
self.background_worker.start()
def resume(self):
logger.debug('resuming')
if self.background_worker.is_alive():
raise RuntimeError('Resume called without first calling pause')
self.background_worker = BackgroundWorker()
self.background_worker.start()
def pause(self):
logger.debug('pausing')
if self.background_worker and self.background_worker.is_alive():
self.background_worker.stop()
def quit(self):
try:
if self.background_worker.is_alive():
logger.info('Stopping background worker')
self.background_worker.stop()
try:
user = session.query(User).filter(User.logged_in).one()
except NoResultFound:
pass
else:
logger.info('Saving user data')
save(session, user)
session.close()
finally:
logger.info('Quitting application')
QApplication.instance().quit()
def sync_now(self):
if not self.background_worker:
self.start()
try:
self.pause()
except RuntimeError:
pass
self.resume()
def get_current_user(self):
return session.query(User).one()
def set_containing_folder_initial(self):
return QFileDialog.getExistingDirectory(
self, "Choose where to place OSF folder")
def logout(self):
user = session.query(User).filter(User.logged_in).one()
user.logged_in = False
try:
save(session, user)
except SQLAlchemyError:
session.query(User).delete()
self.tray.tray_icon.hide()
if self.preferences.isVisible():
self.preferences.close()
self.start_screen.open_window()
def open_preferences(self):
logger.debug('pausing for preference modification')
self.pause()
logger.debug('opening preferences')
self.preferences.open_window(Preferences.GENERAL)
def start_about_screen(self):
self.pause()
self.preferences.open_window(Preferences.ABOUT)
class OSFApp(QDialog):
login_signal = pyqtSignal()
start_tray_signal = pyqtSignal()
containing_folder_updated_signal = pyqtSignal()
def __init__(self):
super().__init__()
# settings
self.app_name = "OSFOffline"
self.app_author = "COS"
# views
self.start_screen = StartScreen()
self.tray = SystemTray()
self.preferences = Preferences()
AlertHandler.setup_alerts(self.tray.tray_icon, self.tray.tray_alert_signal)
# connect all signal-slot pairs
self.setup_connections()
self.background_worker = BackgroundWorker()
def setup_connections(self):
# [ (signal, slot) ]
signal_slot_pairs = [
# system tray
(self.tray.open_osf_folder_action.triggered, self.tray.open_osf_folder),
(self.tray.launch_osf_action.triggered, self.tray.start_osf),
(self.tray.sync_now_action.triggered, self.sync_now),
# (self.tray.currently_synching_action.triggered, self.controller.currently_synching),
(self.tray.preferences_action.triggered, self.open_preferences),
(self.tray.about_action.triggered, self.start_about_screen),
(self.tray.quit_action.triggered, self.quit),
(self.tray.tray_alert_signal, self.tray.update_currently_synching),
# main events
(self.login_signal, self.start_screen.open_window),
(self.start_tray_signal, self.tray.start),
# preferences
(self.preferences.preferences_window.desktopNotifications.stateChanged, self.preferences.alerts_changed),
(self.preferences.preferences_window.startOnStartup.stateChanged, self.preferences.startup_changed),
(self.preferences.preferences_window.changeFolderButton.clicked, self.preferences.set_containing_folder),
(self.preferences.preferences_closed_signal, self.resume),
(self.preferences.preferences_window.accountLogOutButton.clicked, self.logout),
(self.preferences.containing_folder_updated_signal, self.tray.set_containing_folder),
# (self.preferences.containing_folder_updated_signal, self.preferences.update_containing_folder_text_box),
# start screen
(self.start_screen.done_logging_in_signal, self.start),
(self.start_screen.quit_application_signal, self.quit),
]
for signal, slot in signal_slot_pairs:
signal.connect(slot)
def _can_restart_background_worker(self):
try:
user = self.get_current_user()
except:
return False
if not user.logged_in:
return False
if not os.path.isdir(user.osf_local_folder_path):
return False
if not self.background_worker:
return False
if self.background_worker.is_alive():
return False
return True
def start(self):
logger.debug('Start in main called.')
try:
user = session.query(User).filter(User.logged_in).one()
except MultipleResultsFound:
session.query(User).delete()
self.login_signal.emit()
return
except NoResultFound:
self.login_signal.emit()
return
try:
# Simple request to ensure user logged in with valid oauth_token
user = asyncio.get_event_loop().run_until_complete(AuthClient().populate_user_data(user))
except AuthError as e:
logging.exception(e.message)
self.login_signal.emit()
containing_folder = os.path.dirname(user.osf_local_folder_path)
while not validate_containing_folder(containing_folder):
logger.warning('Invalid containing folder: {}'.format(containing_folder))
AlertHandler.warn('Invalid containing folder. Please choose another.')
containing_folder = os.path.abspath(self.set_containing_folder_initial())
user.osf_local_folder_path = os.path.join(containing_folder, 'OSF')
save(session, user)
self.tray.set_containing_folder(containing_folder)
if not os.path.isdir(user.osf_local_folder_path):
os.makedirs(user.osf_local_folder_path)
self.start_tray_signal.emit()
logger.debug('starting background worker from main.start')
self.background_worker = BackgroundWorker()
self.background_worker.start()
def resume(self):
logger.debug('resuming')
if self.background_worker.is_alive():
raise RuntimeError('Resume called without first calling pause')
self.background_worker = BackgroundWorker()
self.background_worker.start()
def pause(self):
logger.debug('pausing')
if self.background_worker and self.background_worker.is_alive():
self.background_worker.stop()
def quit(self):
try:
if self.background_worker.is_alive():
logger.info('Stopping background worker')
self.background_worker.stop()
try:
user = session.query(User).filter(User.logged_in).one()
except NoResultFound:
pass
else:
logger.info('Saving user data')
save(session, user)
session.close()
finally:
logger.info('Quitting application')
QApplication.instance().quit()
def sync_now(self):
if not self.background_worker:
self.start()
try:
self.pause()
except RuntimeError:
pass
self.resume()
def get_current_user(self):
return session.query(User).one()
def set_containing_folder_initial(self):
return QFileDialog.getExistingDirectory(self, "Choose where to place OSF folder")
def logout(self):
user = session.query(User).filter(User.logged_in).one()
user.logged_in = False
try:
save(session, user)
except SQLAlchemyError:
session.query(User).delete()
self.tray.tray_icon.hide()
if self.preferences.isVisible():
self.preferences.close()
self.start_screen.open_window()
def open_preferences(self):
logger.debug('pausing for preference modification')
self.pause()
logger.debug('opening preferences')
self.preferences.open_window(Preferences.GENERAL)
def start_about_screen(self):
self.pause()
self.preferences.open_window(Preferences.ABOUT)