Coverage for pyanaconda/queuefactory.py: 0%

# Coverage for pyanaconda/queuefactory.py : 0%

## 25 statements 0 run 25 missing 0 excluded

Hot-keys on this page

r m x p toggle line displays

j k next/prev highlighted chunk

0 (zero) top of page

1 (one) first highlighted chunk

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 | # # queue.py: factory for creating communications channels # # Copyright (C) 2013 Red Hat, Inc. All rights reserved. # # This program 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 2 of the License, or # (at your option) any later version. # # This program 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 this program. If not, see <http://www.gnu.org/licenses/>. import queue from pyanaconda.core.util import lowerASCII, upperASCII class QueueFactory(object): """Constructs a new object wrapping a Queue.Queue, complete with constants and sending functions for each type of message that can be put into the queue. Creating a new object using this class is done like so: q = QueueFactory("progress") And then adding messages to it is done like so: q.addMessage("init", 0) q.addMessage("step", 1) The first call will create a new constant named PROGRESS_CODE_INIT and a method named send_init that takes zero arguments. The second call will create a new constant named PROGRESS_CODE_STEP and a method named send_step that takes one argument. Reusing names within the same class is not allowed. """ def __init__(self, name): self.name = name self.__counter = 0 self.__names = [] self.q = queue.Queue() def _makeMethod(self, constant, methodName, argc): def __method(*args): if len(args) != argc: raise TypeError("%s() takes exactly %d arguments (%d given)" % (methodName, argc, len(args))) self.q.put((constant, args)) __method.__name__ = methodName return __method def addMessage(self, name, argc): if name in self.__names: raise AttributeError("%s queue already has a message named %s" % (self.name, name)) # Add a constant. const_name = upperASCII(self.name) + "_CODE_" + upperASCII(name) setattr(self, const_name, self.__counter) self.__counter += 1 # Add a convenience method for putting things into the queue. method_name = "send_" + lowerASCII(name) method = self._makeMethod(getattr(self, const_name), method_name, argc) setattr(self, method_name, method) self.__names.append(name)