Linux下基于python的定时提醒工具

#! /usr/bin/env python # -*- coding: utf-8 -*- # ************************************************************* # Filename @ kreminder.py # Author @ Huoty # Create date @ 2015-12-25 16:59:16 # Description @ # ************************************************************* from os import getcwd, path, makedirs from time import strftime, localtime from re import match from apscheduler.schedulers.background import BackgroundScheduler from PyQt4.QtGui import QApplication from systray import SysTray import sys import logging logging.basicConfig() def log(msg, type = "info"): """ Write information to log file. type: info, warn, error """ log_dir = path.expandvars('$HOME') + "/.kreminder/log" log_file = log_dir + "/kreminder.log" if not path.exists(log_dir): makedirs(log_dir) curr_time = "[ " + strftime("%Y-%m-%d %H:%M:%S", localtime()) + " " + " ]" if type == "info": content = curr_time + " <INFO> " + msg + "\n" elif type == "warn": content = curr_time + " <WARNING> " + msg + "\n" elif type == "error": content = curr_time + " <ERROR> " + msg + "\n" else: content = curr_time + "没有定义的日志类型！\n" file_object = open(log_file, "a") file_object.write(content) file_object.close() def parse_conf_file(file_object): status = 0 # 状态机： 0 - 初始状态， 1 - 读取休息提醒配置， 2 - 读取代办事项配置 remind_rest_dict = {} todo_list_dict = {} file_content = [ line.strip() for line in file_object.readlines() ] for line in file_content: if len(line) == 0 or line[0] == "#": continue elif line == "[REST_INTERVAL]": status = 1 continue elif line == "[TODO_LIST]": status = 2 continue else: if line.count("=") != 1: continue else: key, val = [ item.strip(' "') for item in line.split("=") ] if status == 1: if not cmp(key, "interval"): remind_rest_dict["interval"] = int(val) elif not cmp(key, "message"): remind_rest_dict["message"] = unicode(val, "utf-8") else: log(key + " 是未定义的配置项.", "error") elif status == 2: todo_list_dict[key] = unicode(val, "utf-8") else: log("在解析配置文件时出现未定义的状态： " + str(status), "error") return remind_rest_dict, todo_list_dict def get_conf(conf_item = "all"): """ Return configuration information by specific configuration Item conf_item: all, rest, todo """ user_conf_file = path.expandvars('$HOME') + "/.kreminder/config/kreminder.conf" global_conf_file = "/etc/kreminder.conf" if path.exists(user_conf_file) and path.isfile(user_conf_file): fp = open(user_conf_file, "r") elif path.exists(global_conf_file) and path.isfile(global_conf_file): fp = open(global_conf_file, "r") else: log("没有找到配置文件.", "error") return None all_conf = parse_conf_file(fp) if not cmp(conf_item, "all"): return all_conf elif not cmp(conf_item, "rest"): return all_conf[0] elif not cmp(conf_item, "todo"): return all_conf[1] else: log("该配置项（%s）还未定义." % conf_item, "warn") return None # Script starts from here if __name__ == "__main__": app = QApplication(sys.argv) tray_icon = SysTray() tray_icon.show() tray_icon.showMessage(u"启动通知", u"Kreminder 已经启动，点击系统托盘图标可以获取帮助.", 1, 15000) # scheduler of Reminding rest rest_config = get_conf("rest") if rest_config: try: rest_remind_scheduler = BackgroundScheduler() rest_remind_scheduler.add_job(tray_icon.showMessage, 'interval', \ minutes = rest_config["interval"], \ args = [u"休息提醒", rest_config["message"], 1, 15000], \ id = "remind_rest") rest_remind_scheduler.start() except Exception, e: log(e, "error") else: log("未获取到 rest 的相关配置.", "error") # scheduler of Reminding todo todo_config = get_conf("todo") if todo_config: try: todo_remind_scheduler = BackgroundScheduler() for todo_date, todo_msg in todo_config.items(): if match("Everyday [\d]{2}:[\d]{2}:[\d]{2}", todo_date.strip()): todo_time = (todo_date.split()[1]).split(":") todo_remind_scheduler.add_job(tray_icon.showMessage, 'cron', \ day = "*", \ hour = todo_time[0], \ minute = todo_time[1], \ second = todo_time[2], \ args = [u"待办事项", todo_msg, 1, 15000], \ id = todo_date) elif match("[\d]{4}-[\d]{2}-[\d]{2} [\d]{2}:[\d]{2}:[\d]{2}", todo_date.strip()): if todo_date > strftime("%Y-%m-%d %H:%M:%S", localtime()): todo_remind_scheduler.add_job(tray_icon.showMessage, "date", \ run_date = todo_date, \ args = [u"待办事项", todo_msg, 1, 15000], \ id = todo_date) else: log("设置的日期已经过期： " + todo_date , "warn") else: log("Todo 配置项的日期格式不正确： " + todo_date, "error") todo_remind_scheduler.start() except Exception, e: log(e, "error") else: log("未获取到 todo 的相关配置.", "error") # main process try: sys.exit(app.exec_()) except Exception, e: log(e, "error") rest_remind_scheduler.shutdown() todo_remind_scheduler.shutdown()