# # This file is part of SickGear. # # SickGear is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # SickGear is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with SickGear. If not, see . import datetime import sickgear MESSAGE = 'notice' ERROR = 'error' class Notifications(object): """ A queue of Notification objects. """ def __init__(self): self._messages = [] self._errors = [] def __del__(self): pass def message(self, title, message=''): """ Add a regular notification to the queue title: The title of the notification message: The message portion of the notification """ self._messages.append(Notification(title, message, MESSAGE)) def error(self, title, message=''): """ Add an error notification to the queue title: The title of the notification message: The message portion of the notification """ self._errors.append(Notification(title, message, ERROR)) def get_notifications(self, remote_ip='127.0.0.1'): """ Return all the available notifications in a list. Marks them all as seen as it returns them. Also removes timed out Notifications from the queue. Returns: A list of Notification objects """ # filter out expired notifications self._errors = [x for x in self._errors if not x.is_expired()] self._messages = [x for x in self._messages if not x.is_expired()] # return any notifications that haven't been shown to the client already return [x.see(remote_ip) for x in self._errors + self._messages if x.is_new(remote_ip)] # static notification queue object notifications = Notifications() class Notification(object): """ Represents a single notification. Tracks its own timeout and a list of which clients have seen it before. """ def __init__(self, title, message='', n_type=None, timeout=None): self.title = title self.message = message self._when = datetime.datetime.now() self._seen = [] if n_type: self.type = n_type else: self.type = MESSAGE if timeout: self._timeout = timeout else: self._timeout = datetime.timedelta(minutes=1) def __del__(self): pass def is_new(self, remote_ip='127.0.0.1'): """ Returns True if the notification hasn't been displayed to the current client (aka IP address). """ return remote_ip not in self._seen def is_expired(self): """ Returns True if the notification is older than the specified timeout value. """ return datetime.datetime.now() - self._when > self._timeout def see(self, remote_ip='127.0.0.1'): """ Returns this notification object and marks it as seen by the client ip """ self._seen.append(remote_ip) return self class ProgressIndicator(object): def __init__(self, percent_complete=0, current_status=None): self.percent_complete = percent_complete self.currentStatus = {'title': ''} if None is current_status else current_status class ProgressIndicators(object): _pi = {'bulkChange': [], 'massAdd': [], 'dailyUpdate': [] } @staticmethod def get_indicator(name): if name not in ProgressIndicators._pi: return [] # if any of the progress indicators are done take them off the list for curPI in ProgressIndicators._pi[name]: if None is not curPI and 100 == curPI.percent_complete(): ProgressIndicators._pi[name].remove(curPI) # return the list of progress indicators associated with this name return ProgressIndicators._pi[name] @staticmethod def set_indicator(name, indicator): ProgressIndicators._pi[name].append(indicator) class QueueProgressIndicator(object): """ A class used by the UI to show the progress of the queue or a part of it. """ def __init__(self, name, queue_item_list): self.queueItemList = queue_item_list self.name = name def num_total(self): return len(self.queueItemList) def num_finished(self): return len([x for x in self.queueItemList if not x.is_in_queue()]) def num_remaining(self): return len([x for x in self.queueItemList if x.is_in_queue()]) def next_name(self): for curItem in [ sickgear.show_queue_scheduler.action.currentItem] + sickgear.show_queue_scheduler.action.queue: if curItem in self.queueItemList: return curItem.name return "Unknown" def percent_complete(self): num_finished = self.num_finished() num_total = self.num_total() if 0 == num_total: return 0 return int(float(num_finished) / float(num_total) * 100) class LoadingTVShow(object): def __init__(self, path): self.dir = path self.show_obj = None