workrave-auto.py
No OneTemporary
Actions

Size

24 KB

Subscribers

None

workrave-auto.py
View Options

	#!/usr/bin/env python

	from PyQt5.QtCore import QTimer, pyqtSignal, QObject, pyqtSlot, QCoreApplication, QMetaObject, Qt, QEventLoop, QThread
	from PyQt5 import QtGui, QtWidgets
	from dbus.mainloop.pyqt5 import DBusQtMainLoop
	import dbus
	import os
	import configparser
	import datetime
	import time
	import sys
	import signal
	import asyncio
	import caldav
	from caldav.elements import dav, cdav
	import icalendar
	import secretstorage
	import shelve
	from dateutil.tz import UTC, gettz
	from dateutil.relativedelta import relativedelta
	import asyncio

	class Login(QtWidgets.QDialog):
	def __init__(self, url, parent=None):
	super(Login, self).__init__(parent)
	self.textName = QtWidgets.QLineEdit(self)
	self.textPass = QtWidgets.QLineEdit(self)
	self.textPass.setEchoMode(2)
	title = QtWidgets.QLabel('Please enter credentials for: ' + url)
	lbl1 = QtWidgets.QLabel('Username', self)
	lbl2 = QtWidgets.QLabel('Password', self)
	self.buttonLogin = QtWidgets.QPushButton('OK', self)
	self.cancelLogin = QtWidgets.QPushButton('Cancel', self)
	self.buttonLogin.clicked.connect(self.handleLogin)
	self.cancelLogin.clicked.connect(self.handleReject)
	layout = QtWidgets.QVBoxLayout(self)
	layout2 = QtWidgets.QHBoxLayout()
	layout3 = QtWidgets.QHBoxLayout()
	layout4 = QtWidgets.QHBoxLayout()
	layout.addWidget(title)

	layout3.addWidget(lbl1)
	layout3.addWidget(self.textName)
	layout.addLayout(layout3)

	layout4.addWidget(lbl2)
	layout4.addWidget(self.textPass)
	layout.addLayout(layout4)

	layout2.addWidget(self.buttonLogin)
	layout2.addWidget(self.cancelLogin)
	layout.addLayout(layout2)

	def handleLogin(self):
	self.accept()

	def handleReject(self):
	self.reject()

	class CalendarSync(QObject):

	sync_result = pyqtSignal(list)

	def __init__(self, url):
	super(CalendarSync, self).__init__()
	self.url = url
	self.calendars = []
	self.persistence = False
	self.persistenceFile = ''
	self.cache = {}
	self.events = []
	self.sync_running = False

	def setCalendars(self, calendars):
	self.calendars = calendars

	def setPersistencePath(self, path):
	self.persistence = True
	self.persistenceFile = path + "/syncPersistence"

	def get_credentials(self):
	username = ''
	password = ''

	asyncio.set_event_loop(asyncio.new_event_loop())
	connection = secretstorage.dbus_init()
	collection = secretstorage.get_default_collection(connection)
	storedItems = collection.search_items({'xdg:schema' : 'at.aboehler.workrave-auto'})
	for item in storedItems:
	attributes = item.get_attributes()
	if 'server' in attributes:
	if attributes['server'] == self.url:
	username = attributes['username']
	password = item.get_secret()
	break

	if username == '' or password == '':
	login = Login(self.url)
	if login.exec_() == QtWidgets.QDialog.Accepted:
	username = login.textName.text()
	password = login.textPass.text()
	attributes = {'xdg:schema': 'at.aboehler.workrave-auto',
	'server' : self.url,
	'username' : username}
	item = collection.create_item('workrave-auto', attributes, password)
	else:
	print('Wrong')

	return username, password

	def get_sync_running(self):
	return self.sync_running

	def get_events(self):
	return self.events

	@pyqtSlot(bool, bool)
	def sync(self, useCache = True, onlyCache = False):
	self.sync_running = True
	if self.persistence and useCache:
	cache = shelve.open(self.persistenceFile)
	if onlyCache and useCache:
	if self.url in cache:
	self.events = cache[self.url]
	cache.close()
	self.sync_running = False
	self.sync_result.emit(self.events)
	return self.events
	else:
	self.sync_running = False
	self.sync_result.emit(self.events)
	return []
	username, password = self.get_credentials()
	client = caldav.DAVClient(self.url, username=username, password=password)
	self.events = []
	try:
	principal = client.principal()
	calendars = principal.calendars()
	for calendar in calendars:
	calName = calendar.get_properties([dav.DisplayName()])
	if calName['{DAV:}displayname'] in self.calendars:
	entries = calendar.date_search(datetime.date.today(), datetime.date.today() + datetime.timedelta(days=7))
	for entry in entries:
	cal = icalendar.Calendar.from_ical(entry._get_data())
	event = cal.walk('VEVENT')[0]
	if event.get('rrule'):
	pass
	else:
	self.events.append(event)
	if self.persistence and useCache:
	cache[self.url] = self.events
	except Exception as e:
	print('Probably wrong credentials?')
	print(str(e))
	print('We fail silently here, that\'s by design...')

	if self.persistence and useCache and self.url in cache:
	self.events = cache[self.url]
	cache.close()
	self.sync_running = False
	self.sync_result.emit(self.events)
	return self.events

	def wk2int(day):
	if day == 'mon':
	return 0
	elif day == 'tue':
	return 1
	elif day == 'wed':
	return 2
	elif day == 'thu':
	return 3
	elif day == 'fri':
	return 4
	elif day == 'sat':
	return 5
	elif day == 'sun':
	return 6
	else:
	return -1

	def time2diff(now, time):
	datetime.datetime.now()
	to = time.split(':')
	hr = int(to[0])
	mi = 0
	sec = 0
	if len(to) == 2:
	mi = int(to[1])
	if len(to) == 3:
	sec = int(to[2])
	return datetime.datetime.combine(datetime.date.today(), datetime.time(hr, mi, sec)) - now

	class autoEntry:
	def __init__(self, options):
	self.options = options
	self.next_action = options['default']
	self.next_run = 0
	self.in_interval = False
	self.startDt = False
	self.endDt = False
	self.update()

	def print_options(self):

	print(self.options)

	def update(self):
	self.in_interval = False
	now = datetime.datetime.now()
	dateOn = now.date()
	dateOff = now.date()
	timeOn = datetime.time(0,0)
	timeOff = datetime.time(23,59)
	recurring = False
	if 'weekday' in self.options:
	day = wk2int(self.options['weekday'])
	if now.date().weekday() == day:
	dateOn = now.date()
	dateOff = now.date()
	else:
	td = day - now.date().weekday()
	if td < 0:
	td += 7
	dateOn = now.date() + datetime.timedelta(days=td)
	dateOff = dateOn
	recurring = True
	# Don't forget to add a week if we are outside of the interval

	if 'startdate' in self.options:
	dateOn = datetime.datetime.strptime(self.options['startdate'], "%Y-%m-%d").date()
	if 'enddate' in self.options:
	dateOff = datetime.datetime.strptime(self.options['enddate'], "%Y-%m-%d").date()
	if 'starttime' in self.options:
	timeOn = datetime.datetime.strptime(self.options['starttime'], "%H:%M").time()
	if 'endtime' in self.options:
	timeOff = datetime.datetime.strptime(self.options['endtime'], "%H:%M").time()

	startDt = datetime.datetime.combine(dateOn, timeOn)
	endDt = datetime.datetime.combine(dateOff, timeOff)

	if startDt < now and endDt > now:
	# in interval
	self.next_action = self.options['default']
	self.next_run = (endDt - now).total_seconds()
	self.in_interval = True
	elif startDt < now and endDt < now:
	# event is in the past
	if recurring:
	self.next_action = self.options['mode']
	self.next_run = ((endDt + datetime.timedelta(days=7)) - now).total_seconds()
	else:
	self.next_action = False
	self.next_run = False
	elif startDt > now and endDt > now:
	# in the future
	self.next_action = self.options['mode']
	self.next_run = (startDt - now).total_seconds()
	else:
	print('Invalid settings')
	self.next_action = False
	self.next_run = False

	self.startDt = startDt
	self.endDt = endDt

	def get_next_action(self):
	self.update()
	return self.next_action

	def get_interval_action(self):
	return self.options['mode']

	def get_next_run(self):
	self.update()
	return int(self.next_run)

	def get_start(self):
	self.update()
	return self.startDt.timestamp()

	def get_end(self):
	self.update()
	return self.endDt.timestamp()

	def get_options(self):
	return self.options

	def get_string(self):
	return "Start: " + self.startDt.strftime('%Y-%m-%d %H:%M:%s') + ', End: ' + self.endDt.strftime('%Y-%m-%d %H:%M:%s')

	def is_in_interval(self):
	self.update()
	return self.in_interval

	def load_config(cfgFile):
	config = configparser.ConfigParser()
	config.read(cfgFile)
	return config

	def save_config(config, cfgFile):
	with open(cfgFile, 'w') as configFile:
	config.write(configFile)

	def recreate_config(cfgFile):
	config = configparser.ConfigParser()
	config['general'] = {
	'mode' : 'normal',
	'entries' : 'entry1,entry2',
	'calendards' : 'calendar1,calendar2',
	'calendarsyncinterval' : '60'
	}
	config['calendar1'] = {
	'url' : 'https://my.dav.server',
	'calendar' : 'Personal',
	'wholeday' : 'skip'
	}
	config['calendar2'] = {
	'url' : 'https://my.dav.server',
	'calendar' : 'Personal',
	'wholeday' : 'skip'
	}
	config['entry1'] = {
	'name' : 'workrave',
	'mode' : 'suspended',
	'weekday' : 'wed',
	'starttime' : '07:00',
	'endtime' : '15:10'
	}
	config['entry2'] = {
	'name' : 'workrave',
	'mode' : 'suspended',
	'weekday' : 'thu',
	'starttime' : '07:00',
	'endtime' : '18:00'
	}
	save_config(config, cfgFile)

	class workraveManager(QObject):

	start_sync = pyqtSignal(bool, bool)
	finished = pyqtSignal()
	timer_time = pyqtSignal(int)

	def __init__(self, cfg, parent=None):
	QObject.__init__(self)
	self.cfg = cfg
	self.parent = parent

	self.entryobjects = []

	self.default_mode = cfg['general']['mode']
	self.sync_interval = cfg['general']['calendarsyncinterval']
	self.inhibited = False
	self.first_run = True

	def build_entryobjects(self):
	self.entryobjects.clear()

	entries = self.parse_config(self.cfg)

	for entry in entries['entries']:
	entryCfg = entries['entries'][entry]
	entryCfg['default'] = self.default_mode
	self.entryobjects.append(autoEntry(entryCfg))

	for calendar in entries['calendars']:
	calendarCfg = entries['calendars'][calendar]
	calendarCfg['default'] = self.default_mode
	calendarCfg['cfgpath'] = self.cfg['general']['cfgpath']
	if self.first_run:
	self.entryobjects.extend(self.calendar2autoEntry(calendarCfg, onlyCache = True))
	else:
	self.entryobjects.extend(self.calendar2autoEntry(calendarCfg))

	self.entryobjects = self.deduplicate(self.entryobjects)

	self.first_run = False
	if len(entries['calendars']) > 0 and not self.syncTimer.isActive():
	self.syncTimer.start(int(self.sync_interval) * 1000 * 60)

	def deduplicate(self, entries):
	retlist = []
	print('Before simplification we had ' + str(len(entries)) + ' entries')

	for entry in entries:
	duplicate = False
	for ret in retlist:
	# New entry is contained in existing one
	if entry.get_start() > ret.get_start() and entry.get_end() < ret.get_end():
	print('Entry is contained in existing one, not adding')
	duplicate = True
	# New entry includes existing one
	elif entry.get_start() < ret.get_start() and entry.get_end() > ret.get_end():
	print('Existing entry is contained in new one, deleting existing one')
	duplicate = False
	retlist.remove(ret)
	# Starts before and ends between
	elif entry.get_start() < ret.get_start() and entry.get_end() < ret.get_end() and entry.get_end() > ret.get_start():
	print('New entry starts before and ends between existing one')
	duplicate = False
	dts = datetime.datetime.fromtimestamp(entry.get_start())
	dte = datetime.datetime.fromtimestamp(ret.get_end())
	tmp = {}
	tmp['default'] = self.default_mode
	tmp['mode'] = 'suspended'
	tmp['starttime'] = dts.strftime("%H:%M")
	tmp['startdate'] = dts.strftime("%Y-%m-%d")
	tmp['endtime'] = dte.strftime("%H:%M")
	tmp['enddate'] = dte.strftime("%Y-%m-%d")
	retlist.remove(ret)
	entry = autoEntry(tmp)
	elif entry.get_start() > ret.get_start() and entry.get_end() > ret.get_end() and entry.get_start() < ret.get_end():
	print('New entry starts between and ends after existing one')
	duplicate = False
	dts = datetime.datetime.fromtimestamp(ret.get_start())
	dte = datetime.datetime.fromtimestamp(entry.get_end())
	tmp = {}
	tmp['default'] = self.default_mode
	tmp['mode'] = 'suspended'
	tmp['starttime'] = dts.strftime("%H:%M")
	tmp['startdate'] = dts.strftime("%Y-%m-%d")
	tmp['endtime'] = dte.strftime("%H:%M")
	tmp['enddate'] = dte.strftime("%Y-%m-%d")
	retlist.remove(ret)
	entry = autoEntry(tmp)
	if not duplicate:
	print('Appending: ' + entry.get_string())
	retlist.append(entry)

	print('After simplification we have ' + str(len(retlist)) + ' entries')
	return retlist


	def calendar2autoEntry(self, cfg, onlyCache = False):
	ret = []
	syncEngine = CalendarSync(cfg['url'])
	syncEngine.setCalendars([a.strip() for a in cfg['calendar'].split(',')])
	syncEngine.setPersistencePath(cfg['cfgpath'])
	events = syncEngine.sync(useCache = True, onlyCache = onlyCache)
	for event in events:
	tmp = {}
	tmp['default'] = self.default_mode
	tmp['mode'] = 'suspended'
	if event.get('dtstart') and event.get('dtend'):
	dts = event['dtstart'].dt
	if type(dts) is datetime.datetime:
	dts = dts.astimezone(tz=None)
	elif type(dts) is datetime.date:
	if cfg['wholeday'] == 'skip':
	continue
	dte = event['dtend'].dt
	if type(dte) is datetime.datetime:
	dte = dte.astimezone(tz=None)
	elif type(dte) is datetime.date:
	if cfg['wholeday'] == 'skip':
	continue
	tmp['starttime'] = dts.strftime("%H:%M")
	tmp['startdate'] = dts.strftime("%Y-%m-%d")
	tmp['endtime'] = dte.strftime("%H:%M")
	tmp['enddate'] = dte.strftime("%Y-%m-%d")
	ret.append(autoEntry(tmp))
	return ret

	@pyqtSlot()
	def run_action(self):
	print('Timer fired, about to run action: ' + self.action)
	self.set_workrave_mode(self.action)
	# This prevents the timer from firing several times
	time.sleep(1)
	self.setup_timer()

	@pyqtSlot()
	def sync_calendars(self):
	print('Syncing calendars...')
	self.build_entryobjects()
	self.startup_check()
	self.setup_timer()

	def suspend_handler(self, suspended):
	if suspended:
	print('Going to sleep...')
	self.timer.stop()
	else:
	if self.inhibited:
	return
	print('Resuming, doing the recalculation math!')
	self.startup_check()
	self.setup_timer()

	def init_dbus(self):

	dbus_loop = DBusQtMainLoop(set_as_default=True)
	self.system_bus = dbus.SystemBus(mainloop = dbus_loop)

	manager_interface = 'org.freedesktop.login1.Manager'

	self.system_bus.add_signal_receiver(self.suspend_handler, 'PrepareForSleep', manager_interface)

	def parse_config(self, cfg):
	ret = {}
	if 'entries' in cfg['general']:
	entrylist = cfg['general']['entries']
	if entrylist != '':
	entrylist = entrylist.split(',')
	else:
	entrylist = []
	else:
	entrylist = []
	if 'calendars' in cfg['general']:
	calendarlist = cfg['general']['calendars']
	if calendarlist != '':
	calendarlist = calendarlist.split(',')
	else:
	calendarlist = []
	else:
	calendarlist = []
	ret['entries'] = {}
	ret['calendars'] = {}
	for entry in entrylist:
	ret['entries'][entry] = {}
	for key in cfg[entry]:
	ret['entries'][entry][key] = cfg[entry][key]
	for cal in calendarlist:
	ret['calendars'][cal] = {}
	for key in cfg[cal]:
	ret['calendars'][cal][key] = cfg[cal][key]
	return ret

	def startup_check(self):
	if self.inhibited:
	return

	print('Startup check...')
	in_interval = False
	for entry in self.entryobjects:
	if entry.is_in_interval():
	action = entry.get_interval_action()
	print('Interval currently running, setting mode: ' + action)
	self.set_workrave_mode(action)
	in_interval = True

	if not in_interval:
	print('Not in interval, setting default workrave mode: ' + self.default_mode)
	self.set_workrave_mode(self.default_mode)

	def set_workrave_mode(self, action):
	wr = dbus.SessionBus().get_object('org.workrave.Workrave', '/org/workrave/Workrave/Core')
	iface = dbus.Interface(wr, 'org.workrave.CoreInterface')
	mode = iface.GetOperationMode()
	if mode == action:
	print('Mode already set, no action')
	else:
	iface.SetOperationMode(action)

	@pyqtSlot()
	def get_timer_time(self):
	if self.timer.isActive():
	tm = self.timer.remainingTime()
	self.timer_time.emit(tm)
	return tm
	else:
	return 0

	@pyqtSlot()
	def startup(self):
	self.timer = QTimer()
	self.syncTimer = QTimer()
	self.timer.setSingleShot(True)
	self.timer.timeout.connect(self.run_action)
	self.syncTimer.timeout.connect(self.sync_calendars)
	self.build_entryobjects()
	self.init_dbus()
	self.startup_check()
	self.setup_timer()

	def is_inhibited(self):
	return self.inhibited

	def inhibit(self, action):
	if action:
	self.timer.stop()
	self.inhibited = True
	else:
	self.startup_check()
	self.setup_timer()
	self.inhibited = False

	def setup_timer(self):
	self.timer.stop()
	next_action_list = []
	next_run_list = []
	for entry in self.entryobjects:
	action = entry.get_next_action()
	run = entry.get_next_run()
	if action and run:
	next_action_list.append(action)
	next_run_list.append(run)
	if len(next_run_list) == 0:
	print('Not setting up any timer, nothing to do.')
	return
	sleep_time = min(next_run_list)
	self.action = next_action_list[next_run_list.index(sleep_time)]
	print('Setting up timer for ' + str(sleep_time) + ' seconds and then running action: ' + self.action)
	self.timer.start(sleep_time * 1000)

	def quitWR(self):
	self.timer.stop()
	self.finished.emit()

	class SystemTrayIcon(QtWidgets.QSystemTrayIcon):

	get_timer_time = pyqtSignal()
	sync_calendars = pyqtSignal()

	def __init__(self, icon, manager, parent=None):
	self.parent = parent
	QtWidgets.QSystemTrayIcon.__init__(self, icon, parent)
	menu = QtWidgets.QMenu(parent)
	self.remaining = menu.addAction('Remaining: ')
	self.presentationAction = menu.addAction("Inhibit")
	self.presentationAction.triggered.connect(self.inhibit)
	self.syncAction = menu.addAction("Sync Now!")
	self.syncAction.triggered.connect(self.sync)
	self.exitAction = menu.addAction("Exit")
	self.exitAction.triggered.connect(manager.quitWR)
	self.setContextMenu(menu)
	self.manager = manager
	self.activated.connect(self.tray_activated)

	@pyqtSlot(int)
	def timer_time(self, tm):
	rem = tm / 1000
	sec = datetime.timedelta(seconds=int(rem))
	d = datetime.datetime(1, 1, 1) + sec
	if rem == 0:
	self.remaining.setText("No action planned.")
	else:
	if d.day-1 == 0:
	fmt = "%02d:%02d:%02d" % (d.hour, d.minute, d.second)
	else:
	fmt = "%d days %02d:%02d:%02d" % (d.day-1, d.hour, d.minute, d.second)
	self.remaining.setText("Next Action: " + fmt)

	@pyqtSlot()
	def tray_activated(self):
	self.get_timer_time.emit()

	def sync(self):
	self.sync_calendars.emit()

	@pyqtSlot()
	def inhibit(self):
	if self.manager.is_inhibited():
	self.manager.inhibit(False)
	self.presentationAction.setText("Inhibit")
	else:
	self.manager.inhibit(True)
	self.presentationAction.setText("Resume")


	def main():
	signal.signal(signal.SIGINT, signal.SIG_DFL)
	app = QtWidgets.QApplication(sys.argv)
	app.setQuitOnLastWindowClosed(False);

	cfgPath = os.path.expanduser("~") + "/.workrave-auto/"
	if not os.path.isdir(cfgPath):
	os.makedirs(cfgPath)
	cfgFile = cfgPath + "config.ini"
	if not os.path.exists(cfgFile):
	recreate_config(cfgFile)

	cfg = load_config(cfgFile)
	cfg['general']['cfgpath'] = cfgPath

	w = QtWidgets.QWidget()
	m = workraveManager(cfg, w)
	objThread = QThread()
	m.moveToThread(objThread)
	objThread.started.connect(m.startup)
	#QMetaObject.invokeMethod(m, "startup", Qt.QueuedConnection)
	m.finished.connect(objThread.quit)
	m.finished.connect(app.exit)
	objThread.start()
	#QMetaObject.invokeMethod(objThread, "start", Qt.QueuedConnection)

	# Wait for up to 30 seconds for the systemTray to become available (required for, e.g., Xfce)
	count = 0
	while not QtWidgets.QSystemTrayIcon.isSystemTrayAvailable() and count < 30:
	time.sleep(1)
	count += 1

	trayIcon = SystemTrayIcon(QtGui.QIcon("/usr/share/icons/workrave-auto.png"), m, w)
	trayIcon.show()

	trayIcon.get_timer_time.connect(m.get_timer_time)
	m.timer_time.connect(trayIcon.timer_time)
	trayIcon.sync_calendars.connect(m.sync_calendars)
	sys.exit(app.exec_())

	if __name__ == '__main__':
	main()

File Metadata

Mime Type: text/x-python
Expires: Thu, Dec 5, 4:02 AM (7 h, 34 m)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 626242
Default Alt Text: workrave-auto.py (24 KB)

workrave-auto.pyNo OneTemporaryActions

workrave-auto.pyView Options

File Metadata

Event Timeline

workrave-auto.py
No OneTemporary
Actions

workrave-auto.py
View Options