Source code for pubsub.core.topicobj

"""
Provide the Topic class.

:copyright: Copyright since 2006 by Oliver Schoenborn, all rights reserved.
:license: BSD, see LICENSE_BSD_Simple.txt for details.
"""

from weakref import ref as weakref
import sys
from typing import Tuple, List, Sequence, Mapping, Dict, Callable, Any, Optional, Union, TextIO, MutableMapping, \
    Iterator, ValuesView

from .listener import (
    Listener,
    ListenerValidator,
    CallArgsInfo,
    UserListener,
)

from .topicutils import (
    ALL_TOPICS,
    stringize,
    tupleize,
    validateName,
    smartDedent,
)

from .topicexc import (
    TopicDefnError,
    TopicNameError,
    ExcHandlerError,
)

from .topicargspec import (
    ArgsInfo,
    ArgSpecGiven,
    MsgData,
    ArgsDocs,
    topicArgsFromCallable,
    MessageDataSpecError,
    SenderUnknownMsgDataError,
    SenderMissingReqdMsgDataError,
)

from .annotations import annotationType


@annotationType
class Topic:
    pass


@annotationType
class TreeConfig:
    pass


ListenerFilter = Callable[[Listener], bool]


[docs] class Topic: """ Represent topics in pubsub. Contains information about a topic, including topic's message data specification (MDS), the list of subscribed listeners, docstring for the topic. It allows Python-like access to subtopics (e.g. A.B is subtopic B of topic A). """ def __init__(self, treeConfig: TreeConfig, nameTuple: Tuple[str, ...], description: str, msgArgsInfo: ArgsInfo, parent: Topic = None): """ Create a topic. Should only be called by TopicManager via its getOrCreateTopic() method (which gets called in several places in pubsub, such as sendMessage, subscribe, and newTopic). :param treeConfig: topic tree configuration settings :param nameTuple: topic name, in tuple format (no dots) :param description: "docstring" for topic :param ArgsInfo msgArgsInfo: object that defines MDS for topic :param parent: parent of topic :raises ValueError: invalid topic name """ if parent is None: if nameTuple != (ALL_TOPICS,): msg = 'Only one topic, named %s, can be root of topic tree' raise ValueError(msg % 'pub.ALL_TOPICS') else: validateName(nameTuple) self.__tupleName = nameTuple self.__handlingUncaughtListenerExc = False self._treeConfig = treeConfig self.__validator = None # Registered listeners were originally kept in a Python list; however # a few methods require lookup of the Listener for the given callable, # which is an O(n) operation. A set() could have been more suitable but # there is no way of retrieving an element from a set without iterating # over the set, again an O(n) operation. A dict() is ok too. Because # Listener.__eq__(callable) returns true if the Listener instance wraps # the given callable, and because Listener.__hash__ produces the hash # value of the wrapped callable, calling dict[callable] on a # dict(Listener -> Listener) mapping will be O(1) in most cases: # the dict will take the callables hash, find the list of Listeners that # have that hash, and then iterate over that inner list to find the # Listener instance which satisfies Listener == callable, and will return # the Listener. self.__listeners = dict() # specification: self.__description = None self.setDescription(description) self.__msgArgs = msgArgsInfo if msgArgsInfo.isComplete(): self.__finalize() else: assert not self._treeConfig.raiseOnTopicUnspecified # now that we know the args are fine, we can link to parent self.__parentTopic = None self.__subTopics = {} if parent is None: assert self.hasMDS() else: self.__parentTopic = weakref(parent) assert self.__msgArgs.parentAI() is parent.__msgArgs parent.__adoptSubtopic(self)
[docs] def setDescription(self, desc: str): """Set the 'docstring' of topic""" self.__description = desc
[docs] def getDescription(self) -> str: """Return the 'docstring' of topic""" if self.__description is None: return None return smartDedent(self.__description)
[docs] def setMsgArgSpec(self, argsDocs: ArgsDocs, required: Sequence[str] = ()): """ Specify the message data for topic messages. :param argsDocs: a dictionary of keyword names (message data name) and data 'docstring'; cannot be None :param required: a list of those keyword names, appearing in argsDocs, which are required (all others are assumed optional) Can only be called if this info has not been already set at construction or in a previous call. :raise RuntimeError: if MDS already set at construction or previous call. """ assert self.__parentTopic is not None # for root of tree, this method never called! if argsDocs is None: raise ValueError('Cannot set listener spec to None') if self.__msgArgs is None or not self.__msgArgs.isComplete(): try: specGiven = ArgSpecGiven(argsDocs, required) self.__msgArgs = ArgsInfo(self.__tupleName, specGiven, self.__parentTopic().__msgArgs) except MessageDataSpecError: # discard the lower part of the stack trace exc = sys.exc_info()[1] raise exc self.__finalize() else: raise RuntimeError('Not allowed to call this: msg spec already set!')
[docs] def getArgs(self) -> Tuple[Sequence[str], Sequence[str]]: """ Returns a pair (reqdArgs, optArgs) where reqdArgs is tuple of names of required message arguments, optArgs is tuple of names for optional arguments. If topic args not specified yet, returns (None, None). """ sendable = self.__msgArgs.isComplete() assert sendable == self.hasMDS() if sendable: return (self.__msgArgs.allRequired, self.__msgArgs.allOptional) return None, None
[docs] def getArgDescriptions(self) -> ArgsDocs: """Get a map of keyword names to docstrings: documents each MDS element. """ return self.__msgArgs.getArgsDocs()
[docs] def setArgDescriptions(self, **docs: ArgsDocs): """Set the docstring for each MDS datum.""" self.__msgArgs.setArgsDocs(docs)
[docs] def hasMDS(self) -> bool: """Return true if this topic has a message data specification (MDS).""" return self.__validator is not None
[docs] def filterMsgArgs(self, msgData: MsgData, check: bool = False) -> MsgData: """Get the MDS docstrings for each of the spedified kwargs.""" filteredArgs = self.__msgArgs.filterArgs(msgData) # if no check of args yet, do it now: if check: self.__msgArgs.check(filteredArgs) return filteredArgs
[docs] def isAll(self) -> bool: """ Returns true if this topic is the 'all topics' topic. All root topics behave as though they are child of that topic. """ return self.__tupleName == (ALL_TOPICS,)
[docs] def isRoot(self) -> bool: """ Returns true if this is a "root" topic, false otherwise. A root topic is a topic whose name contains no dots and which has pub.ALL_TOPICS as parent. """ parent = self.getParent() if parent: return parent.isAll() assert self.isAll() return False
[docs] def getName(self) -> str: """Return dotted form of full topic name""" return stringize(self.__tupleName)
[docs] def getNameTuple(self) -> Tuple[str, ...]: """Return tuple form of full topic name""" return self.__tupleName
[docs] def getNodeName(self) -> str: """Return the last part of the topic name (has no dots)""" name = self.__tupleName[-1] return name
[docs] def getParent(self) -> Topic: """ Get Topic object that is parent of self (i.e. self is a subtopic of parent). Return none if self is the "all topics" topic. """ if self.__parentTopic is None: return None return self.__parentTopic()
[docs] def hasSubtopic(self, name: str = None) -> bool: """ Return true only if name is a subtopic of self. If name not specified, return true only if self has at least one subtopic. """ if name is None: return len(self.__subTopics) > 0 return name in self.__subTopics
[docs] def getSubtopic(self, relName: Union[str, Tuple[str, ...]]) -> Topic: """ Get the specified subtopic object. The relName can be a valid subtopic name, a dotted-name string, or a tuple. """ if not relName: raise ValueError("getSubtopic() arg can't be empty") topicTuple = tupleize(relName) assert topicTuple topicObj = self for topicName in topicTuple: child = topicObj.__subTopics.get(topicName) if child is None: msg = 'Topic "%s" doesn\'t have "%s" as subtopic' % (topicObj.getName(), topicName) raise TopicNameError(relName, msg) topicObj = child return topicObj
[docs] def getSubtopics(self) -> ValuesView[Topic]: """Get a list of Topic instances that are subtopics of self.""" return self.__subTopics.values()
[docs] def getNumListeners(self) -> int: """ Return number of listeners currently subscribed to topic. This is different from number of listeners that will get notified since more general topics up the topic tree may have listeners. """ return len(self.__listeners)
[docs] def hasListener(self, listener: UserListener) -> bool: """Return true if listener is subscribed to this topic.""" return listener in self.__listeners
[docs] def hasListeners(self) -> bool: """ Return true if there are any listeners subscribed to this topic, false otherwise. """ return bool(self.__listeners)
[docs] def getListeners(self) -> List[Listener]: """ Get a copy of list of listeners subscribed to this topic. Safe to iterate over while listeners get un/subscribed from this topics (such as while sending a message). """ return list(self.__listeners.keys())
[docs] def getListenersIter(self) -> Iterator[Listener]: """ Get an iterator over listeners subscribed to this topic. Do not use if listeners can be un/subscribed while iterating. """ return self.__listeners.keys()
[docs] def validate(self, listener: UserListener, curriedArgNames: Sequence[str] = None) -> CallArgsInfo: """ Checks whether listener could be subscribed to this topic: if yes, just returns; if not, raises ListenerMismatchError. Note that method raises TopicDefnError if self not hasMDS(). """ if not self.hasMDS(): raise TopicDefnError(self.__tupleName) return self.__validator.validate(listener, curriedArgNames=curriedArgNames)
[docs] def isValid(self, listener: UserListener, curriedArgNames: Sequence[str] = None) -> bool: """ Return True only if listener could be subscribed to this topic, otherwise returns False. Note that method raises TopicDefnError if self not hasMDS(). """ if not self.hasMDS(): raise TopicDefnError(self.__tupleName) return self.__validator.isValid(listener, curriedArgNames=curriedArgNames)
[docs] def subscribe(self, listener: UserListener, **curriedArgs) -> Tuple[Listener, bool]: """ Subscribe listener to this topic. Returns a pair (pub.Listener, success). :param curriedArgs: keyword argument to curry the listener arguments at message time; the listener(args) is treated essentially as ``listener(**(args - curriedArgs))``. If the listener was already subscribed, the pure curried args names (curriendArgs.keys() - _overrides_) must be unchanged. :return: True only if listener was not already subscribed; False if it was already subscribed. """ if listener in self.__listeners: assert self.hasMDS() newSub = False subdLisnr = self.__listeners[listener] # subscribe with different curried args; only ok if the keys are the same! if curriedArgs: subdLisnr.setCurriedArgs(**curriedArgs) else: newSub = True if self.__validator is None: args, reqd = topicArgsFromCallable(listener, ignoreArgs=curriedArgs) self.setMsgArgSpec(args, reqd) assert self.__validator is not None argsInfo = self.__validator.validate(listener, curriedArgNames=curriedArgs) weakListener = Listener( listener, argsInfo, curriedArgs=curriedArgs, onDead=self.__onDeadListener) self.__listeners[weakListener] = weakListener subdLisnr = weakListener # notify of subscription self._treeConfig.notificationMgr.notifySubscribe(subdLisnr, self, newSub) return subdLisnr, newSub
[docs] def unsubscribe(self, listener: UserListener) -> Listener: """ Unsubscribe the specified listener from this topic. Returns the pub.Listener object associated with the listener that was unsubscribed, or None if the specified listener was not subscribed to this topic. Note that this method calls ``notifyUnsubscribe(listener, self)`` on all registered notification handlers (see pub.addNotificationHandler). """ unsubdLisnr = self.__listeners.pop(listener, None) if unsubdLisnr is None: return None unsubdLisnr._unlinkFromTopic_() assert listener == unsubdLisnr.getCallable() # notify of unsubscription self._treeConfig.notificationMgr.notifyUnsubscribe(unsubdLisnr, self) return unsubdLisnr
[docs] def unsubscribeAllListeners(self, filter: ListenerFilter = None) -> List[Listener]: """ Clears list of subscribed listeners. If filter is given, it must be a function that takes a listener and returns true if the listener should be unsubscribed. Returns the list of Listener for listeners that were unsubscribed. """ unsubd = [] if filter is None: for listener in self.__listeners: listener._unlinkFromTopic_() unsubd = self.__listeners.keys() self.__listeners = {} else: unsubd = [] for listener in list(self.__listeners): if filter(listener): unsubd.append(listener) listener._unlinkFromTopic_() del self.__listeners[listener] # send notification regarding all listeners actually unsubscribed notificationMgr = self._treeConfig.notificationMgr for unsubdLisnr in unsubd: notificationMgr.notifyUnsubscribe(unsubdLisnr, self) return unsubd
[docs] def publish(self, **msgData): """ This sends message to listeners of parent topics as well. If an exception is raised in a listener, the publish is aborted, except if there is a handler (see pub.setListenerExcHandler). Note that it is important that the PublisherMixin NOT modify any state data during message sending, because in principle it could happen that a listener causes another message of same topic to be sent (presumably, the listener has a way of preventing infinite loop). """ self._treeConfig.notificationMgr.notifySend('pre', self) # check the message data: if self.__validator is not None: self._getListenerSpec().check(msgData) else: assert not self.hasListeners() # get the list of topics from self to root (ALL_TOPICS) topicStack = [self] parent = self.__parentTopic while parent is not None: parent = parent() topicStack.append(parent) parent = parent.__parentTopic # deref weakref # for each topic, send to listeners: msgDataSubset = {} for topicObj in reversed(topicStack): added_args = topicObj.__msgArgs.argsAddedToParent add_to_parent = {k: msgData[k] for k in added_args if k in msgData} msgDataSubset.update(add_to_parent) if topicObj.hasListeners(): self.__sendMessage(msgData, topicObj, msgDataSubset) self._treeConfig.notificationMgr.notifySend('post', self)
name = property(getName) parent = property(getParent) subtopics = property(getSubtopics) description = property(getDescription, setDescription) listeners = property(getListeners) numListeners = property(getNumListeners) args = property(getArgs) argDescriptions = property(getArgDescriptions, setArgDescriptions) ############################################################# # # Impementation # ############################################################# def _getListenerSpec(self) -> ArgsInfo: """Only to be called by pubsub package""" return self.__msgArgs def __sendMessage(self, allData: MsgData, topicObj: Topic, data: MsgData): # now send message data to each listener for current topic; # use list of listeners rather than iterator, so that if listeners added/removed during # send loop, no runtime exception (performance hit is marginal): for listener in topicObj.getListeners(): try: self._treeConfig.notificationMgr.notifySend('in', topicObj, pubListener=listener) listener(data, self, allData) except Exception: # if exception handling is on, handle, otherwise re-raise handler = self._treeConfig.listenerExcHandler if handler is None or self.__handlingUncaughtListenerExc: raise # try handling the exception so we can continue the send: try: self.__handlingUncaughtListenerExc = True handler(listener.name(), topicObj) self.__handlingUncaughtListenerExc = False except Exception: exc = sys.exc_info()[1] # print 'exception raised', exc self.__handlingUncaughtListenerExc = False raise ExcHandlerError(listener.name(), topicObj, exc) def __finalize(self): """ Finalize the topic specification, which currently means creating the listener validator for this topic. This allows calls to subscribe() to validate that listener adheres to topic's message data specification (MDS). """ assert self.__msgArgs.isComplete() assert not self.hasMDS() # must make sure can adopt a validator required = self.__msgArgs.allRequired optional = self.__msgArgs.allOptional self.__validator = ListenerValidator(required, list(optional)) assert not self.__listeners def _undefineSelf_(self, topicsMap: MutableMapping[str, Topic]): """Called by topic manager when deleting a topic.""" if self.__parentTopic is not None: self.__parentTopic().__abandonSubtopic(self.__tupleName[-1]) self.__undefineBranch(topicsMap) def __undefineBranch(self, topicsMap: MutableMapping[str, Topic]): """ Unsubscribe all our listeners, remove all subtopics from self, then detach from parent. Parent is not notified, because method assumes it has been called by parent """ # print 'Remove %s listeners (%s)' % (self.getName(), self.getNumListeners()) self.unsubscribeAllListeners() self.__parentTopic = None for subName, subObj in self.__subTopics.items(): assert isinstance(subObj, Topic) # print 'Unlinking %s from parent' % subObj.getName() subObj.__undefineBranch(topicsMap) self.__subTopics = {} del topicsMap[self.getName()] def __adoptSubtopic(self, topicObj: Topic): """Add topicObj as child topic.""" assert topicObj.__parentTopic() is self attrName = topicObj.getNodeName() self.__subTopics[attrName] = topicObj def __abandonSubtopic(self, name: str): """The given subtopic becomes orphan (no parent).""" topicObj = self.__subTopics.pop(name) assert topicObj.__parentTopic() is self def __onDeadListener(self, listener: Listener): """One of our subscribed listeners has died, so remove it and notify""" pubListener = self.__listeners.pop(listener) self._treeConfig.notificationMgr.notifyDeadListener(pubListener, self) def __str__(self): return "%s(%s)" % (self.getName(), self.getNumListeners())