# Author: Nic Wolfe # URL: http://code.google.com/p/sickbeard/ # # This file is part of SickRage. # # SickRage 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. # # SickRage 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 SickRage. If not, see . import datetime import threading from sickbeard import logger class QueuePriorities: LOW = 10 NORMAL = 20 HIGH = 30 class GenericQueue(object): def __init__(self): self.queueItem = None self.currentItem = None self.queue = [] self.queue_name = "QUEUE" self.min_priority = 0 self.lock = threading.Lock() def pause(self): logger.log(u"Pausing queue") self.min_priority = 999999999999 def unpause(self): logger.log(u"Unpausing queue") self.min_priority = 0 def add_item(self, item): item.added = datetime.datetime.now() self.queue.append(item) return item def run(self, force=False): # only start a new task if one isn't already going if self.queueItem is None or not self.queueItem.isAlive(): # if the thread is dead then the current item should be finished if self.currentItem: self.currentItem.finish() self.currentItem = None # if there's something in the queue then run it in a thread and take it out of the queue if len(self.queue) > 0: # sort by priority def sorter(x, y): """ Sorts by priority descending then time ascending """ if x.priority == y.priority: if y.added == x.added: return 0 elif y.added < x.added: return 1 elif y.added > x.added: return -1 else: return y.priority - x.priority self.queue.sort(cmp=sorter) queueItem = self.queue.pop(0) if queueItem.priority < self.min_priority: return # launch the queue item in a thread queueItem.name = self.queue_name + '-' + queueItem.name queueItem.start() self.currentItem = queueItem queueItem.join() class QueueItem(threading.Thread): def __init__(self, name, action_id=0): super(QueueItem, self).__init__() self.name = name.replace(" ", "-").upper() self.inProgress = False self.priority = QueuePriorities.NORMAL self.action_id = action_id self.added = None self.stop = threading.Event() def run(self): """Implementing classes should call this""" self.inProgress = True def finish(self): """Implementing Classes should call this""" self.inProgress = False threading.currentThread().name = self.name