Tracker.py

import datetime, os, sys, threading, yaml
from gi.repository import Gtk
import BuilderObject, DatePicker, PillEdit, TemplateObjects

if sys.platform == 'linux':
    import notifications.LinuxNotifications as Notifications
elif sys.platform == 'win32':
    import notifications.Win32Notifications as Notifications
elif sys.platform == 'darwin':
    import notifications.DarwinNotifications as Notifications
else:
    import notifications.DummyNotifications as Notifications

SaveDataFilename = "{}/savedData/pills.yaml".format(sys.path[0])

class Tracker(BuilderObject.BuilderObject):
    def __init__(self):
        super().__init__("tracker")
        self.window = self.builder.get_object("tracker_window")
        self.date_label = self.builder.get_object("date_label")
        self.pills_container = self.builder.get_object("pills_container")
        self.no_pills_label = self.builder.get_object("no_pills_label")
        self.date = None
        self.pills = []
        self.pill_to_widget = {}
        self.load_pills()
        self.set_date(datetime.date.today())
        self.launch_time_check_thread(0.1)

    def on_tracker_window_destroy(self, widget):
        Gtk.main_quit()

    def on_add_pill_button_clicked(self, widget):
        pill_edit = PillEdit.PillEdit(self)
        pill_edit.new_mode()
        pill_edit.window.show_all()

    def on_date_menu_select_activate(self, widget):
        date_picker = DatePicker.DatePicker(self)
        date_picker.calendar.select_month(self.date.month - 1, self.date.year)
        date_picker.calendar.select_day(self.date.day)
        date_picker.window.show_all()

    def on_date_menu_today_activate(self, widget):
        self.set_date(datetime.date.today())

    def set_date(self, date):
        self.date = date
        self.date_label.set_text(date.strftime("%A %x"))
        for pill in self.pills:
            widget = self.pill_to_widget.get(pill, None)
            if widget is None:
                continue
            for time_tracker in widget.time_trackers.values():
                dates_taken = pill.dates_taken[time_tracker.time.get_hour_minute()]
                time_tracker.checkbox.set_active(self.date in dates_taken)
        self.save_pills()


    def add_pill(self, pill):
        self.pills_container.remove(self.no_pills_label)
        self.pills.append(pill)
        widget = TemplateObjects.PillFrame(pill, self)
        self.pills_container.add(widget.frame)
        self.pill_to_widget.update({pill: widget})
        self.save_pills()

    def delete_pill(self, pill):
        try:
            self.pills.remove(pill)
        except ValueError:
            # It might have already been removed due to some race condition.
            print("Couldn't remove pill {} from tracker.pills!".format(pill.name), file = sys.stderr)
        pillWidget = self.pill_to_widget.pop(pill, None) # Get the pill's widget, if it exists
        if pillWidget is not None: # If there was a widget, remove it
            self.pills_container.remove(pillWidget.frame)
            pillWidget = None
        if len(self.pills) == 0:
            self.pills_container.pack_start(self.no_pills_label, True, True, 0)
        self.save_pills()

    def save_pills(self):
        if not os.path.isdir(os.path.dirname(SaveDataFilename)):
            os.mkdir(os.path.dirname(SaveDataFilename), 0o755)
        with open(SaveDataFilename, 'w') as save_data:
            yaml.dump(self.pills, save_data)

    def load_pills(self):
        if os.path.isfile(SaveDataFilename): # only try if it really exists
            with open(SaveDataFilename, 'r') as save_data:
                for pill in yaml.safe_load(save_data):
                    self.add_pill(pill)

    def time_check(self):
        today = datetime.date.today()
        if self.date != today and today - datetime.timedelta(days = 1) == self.date:
            self.set_date(today)

        now = datetime.datetime.now().time()
        for pill in self.pills:
            widget = self.pill_to_widget.get(pill)
            for time in pill.times:
                time_tracker = None
                if widget is not None:
                    time_tracker = widget.time_trackers.get(time.get_hour_minute())
                if now >= time.get_datetime_time():
                    if today in pill.dates_taken[time.get_hour_minute()]:
                        # We've taken this pill.
                        if time_tracker is not None:
                            time_tracker.alert.set_visible(False)
                    else:
                        # uh oh!
                        if time.notifications and not time_tracker.alert.get_visible():
                            # send a notification!
                            Notifications.send_notification("Pill Tracker", "It's time for your {} dose of {}!".format(time.name, pill.name))
                        if time_tracker is not None:
                            time_tracker.alert.set_visible(True)
                else:
                    # It's not time to take this yet.
                    if time_tracker is not None:
                        time_tracker.alert.set_visible(False)
        self.launch_time_check_thread(0.25)

    def launch_time_check_thread(self, time):
        thread = threading.Timer(time, self.time_check)
        thread.daemon = True
        thread.start()