File size: 843 Bytes
d1ceb73
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
"""A PUB log handler."""
import warnings

from zmq.log.handlers import PUBHandler

warnings.warn(
    "ipykernel.log is deprecated. It has moved to ipyparallel.engine.log",
    DeprecationWarning,
    stacklevel=2,
)


class EnginePUBHandler(PUBHandler):
    """A simple PUBHandler subclass that sets root_topic"""

    engine = None

    def __init__(self, engine, *args, **kwargs):
        """Initialize the handler."""
        PUBHandler.__init__(self, *args, **kwargs)
        self.engine = engine

    @property  # type:ignore[misc]
    def root_topic(self):
        """this is a property, in case the handler is created
        before the engine gets registered with an id"""
        if isinstance(getattr(self.engine, "id", None), int):
            return "engine.%i" % self.engine.id  # type:ignore[union-attr]
        return "engine"