""" :copyright: Copyright since 2006 by Oliver Schoenborn, all rights reserved. :license: BSD, see LICENSE_BSD_Simple.txt for details. """ from typing import Tuple, List, Sequence, Mapping, Dict, Callable, Any, Optional, Union from .topicmgr import ( TopicManager, TreeConfig ) from .listener import IListenerExcHandler, Listener, UserListener from .notificationmgr import INotificationHandler TopicFilter = Callable[[str], bool] ListenerFilter = Callable[[Listener], bool] class Publisher: """ Represent the class that send messages to listeners of given topics and that knows how to subscribe/unsubscribe listeners from topics. """ def __init__(self, treeConfig: TreeConfig = None): """ :param treeConfig: the TreeConfig instance to use; if None, a new one is created """ self.__treeConfig = treeConfig or TreeConfig() self.__topicMgr = TopicManager(self.__treeConfig) def getTopicMgr(self) -> TopicManager: """Get the topic manager created for this publisher.""" return self.__topicMgr def getListenerExcHandler(self) -> IListenerExcHandler: """ Get the listener exception handler that was registered via setListenerExcHandler(), or None of none registered. """ return self.__treeConfig.listenerExcHandler def setListenerExcHandler(self, handler: IListenerExcHandler): """Set the function to call when a listener raises an exception during a sendMessage().""" self.__treeConfig.listenerExcHandler = handler def addNotificationHandler(self, handler: INotificationHandler): """Add a handler for tracing pubsub activity.""" self.__treeConfig.notificationMgr.addHandler(handler) def clearNotificationHandlers(self): """ Remove all notification handlers that were added via self.addNotificationHandler(). """ self.__treeConfig.notificationMgr.clearHandlers() def setNotificationFlags(self, **kwargs: Mapping[str, Optional[bool]]): """ Set the notification flags on or off for each type of pubsub activity. The kwargs keys can be any of the following: - subscribe: if True, get notified whenever a listener subscribes to a topic; - unsubscribe: if True, get notified whenever a listener unsubscribes from a topic; - deadListener: if True, get notified whenever a subscribed listener has been garbage-collected; - sendMessage: if True, get notified whenever sendMessage() is called; - newTopic: if True, get notified whenever a new topic is created; - delTopic: if True, get notified whenever a topic is "deleted" from topic tree; - all: set all of the above to the given value (True or False). The kwargs that are not given or None are left at their current value. Those that are False will cause corresponding notification to be silenced. The 'all' is set first, then the others. E.g. mgr.setFlagStates(all=True, delTopic=False) will toggle all notifications on, but will turn off the 'delTopic' notification. """ self.__treeConfig.notificationMgr.setFlagStates(**kwargs) def getNotificationFlags(self) -> Mapping[str, bool]: """Return a dictionary with the notification flag states.""" return self.__treeConfig.notificationMgr.getFlagStates() def setTopicUnspecifiedFatal(self, newVal: bool = True, checkExisting: bool = True) -> bool: """ Changes the creation policy for topics. By default, pubsub will accept topic names for topics that don't have a message data specification (MDS). This default behavior makes pubsub easier to use initially, but allows topic names with typos to go uncaught in common operations such as sendMessage() and subscribe(). In a large application, this can lead to nasty bugs. Pubsub's default behavior is equivalent to setTopicUnspecifiedFatal(false). When called with newVal=True, any future pubsub operation that requires a topic (such as subscribe and sendMessage) will require an MDS; if none is available, pubsub will raise a TopicDefnError exception. If checkExisting is not given or True, all existing topics are validated. A TopicDefnError exception is raised if one is found to be incomplete (has hasMDS() false). Returns previous value of newVal. Note that this method can be used in several ways: 1. Only use it in your application when something is not working as expected: just add a call at the beginning of your app when you have a problem with topic messages not being received (for instance), and remove it when you have fixed the problem. 2. Use it from the beginning of your app and never use newVal=False: add a call at the beginning of your app and you leave it in (forever), and use Topic Definition Providers to provide the listener specifications. These are easy to use via the pub.addTopicDefnProvider(). 3. Use it as in #1 during app development, and once stable, use #2. This is easiest to do in combination with pub.exportTopicTreeSpec(). """ oldVal = self.__treeConfig.raiseOnTopicUnspecified self.__treeConfig.raiseOnTopicUnspecified = newVal if newVal and checkExisting: self.__topicMgr.checkAllTopicsHaveMDS() return oldVal def subscribe(self, listener: UserListener, topicName: str, **curriedArgs) -> Listener: """ Subscribe listener to named topic. Raises ListenerMismatchError if listener isn't compatible with the topic's MDS. Returns (pubsub.core.Listener, success), where success is False if listener was already subscribed. The pub.core.Listener wraps the callable subscribed and provides introspection-based info about the callable. Extra keyword arguments are treated as currying of listener arguments. Example: pub.subscribe(listener1, 'some_topic') pub.subscribe(listener2, 'some_other_topic', a=2, b=3) In the second example, the listener2 will always receive a=2 and b=3 and pubsub treats it as though a and b were curried, i.e. as if the actual listener subscribed were a callable that did not have a or b parameters. Hence if some_other_topic has a or b as message data, subscription will raise a ListenerInadequate error. Note that if 'subscribe' notification is on, the handler's 'notifySubscribe' method is called after subscription. """ topicObj = self.__topicMgr.getOrCreateTopic(topicName) subscribedListener, success = topicObj.subscribe(listener, **curriedArgs) return subscribedListener, success def unsubscribe(self, listener: UserListener, topicName: str): """ Unsubscribe from given topic. Returns the pubsub.core.Listener instance that was used to wrap listener at subscription time. Raises an TopicNameError if topicName doesn't exist. Note that if 'unsubscribe' notification is on, the handler's notifyUnsubscribe() method will be called after unsubscribing. """ topicObj = self.__topicMgr.getTopic(topicName) unsubdLisnr = topicObj.unsubscribe(listener) return unsubdLisnr def unsubAll(self, topicName: str = None, listenerFilter: ListenerFilter = None, topicFilter: Union[str, TopicFilter] = None) -> List[Listener]: """ Unsubscribe all listeners of a topic. :param topicName: if none given, unsub from all topics. :param listenerFilter: filter function to apply to listeners, unsubscribe only the listeners that satisfy listenerFilter(listener: Listener) == True :param topicFilter: topic name, or a filter function to apply to topics; in latter case, only topics that satisfy topicFilter(topic name) == True will be affected :returns: list of all listeners (instances of pub.Listener) that were unsubscribed from the topic tree Note: this method will generate one 'unsubcribe' notification message (see pub.setNotificationFlags()) for each listener unsubscribed. """ unsubdListeners = [] if topicName is None: # unsubscribe all listeners from all topics topicsMap = self.__topicMgr._topicsMap for topicName, topicObj in topicsMap.items(): if topicFilter is None or topicFilter(topicName): tmp = topicObj.unsubscribeAllListeners(listenerFilter) unsubdListeners.extend(tmp) else: topicObj = self.__topicMgr.getTopic(topicName) unsubdListeners = topicObj.unsubscribeAllListeners(listenerFilter) return unsubdListeners def sendMessage(self, topicName: str, **msgData): """ Send a message. :param topicName: name of message topic (dotted or tuple format) :param msgData: message data (must satisfy the topic's MDS) """ topicMgr = self.getTopicMgr() topicObj = topicMgr.getOrCreateTopic(topicName) topicObj.publish(**msgData)