ipython/IPython/kernel/core/message_cache.py at bug-tests · ivanov/ipython · GitHub

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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# encoding: utf-8

"""Storage for the responses from the interpreter."""

__docformat__ = "restructuredtext en"

#-------------------------------------------------------------------------------
#  Copyright (C) 2008  The IPython Development Team
#
#  Distributed under the terms of the BSD License.  The full license is in
#  the file COPYING, distributed as part of this software.
#-------------------------------------------------------------------------------

#-------------------------------------------------------------------------------
# Imports
#-------------------------------------------------------------------------------


class IMessageCache(object):
    """ Storage for the response from the interpreter.
    """

    def add_message(self, i, message):
        """ Add a message dictionary to the cache.

        Parameters
        ----------
        i : int
        message : dict
        """

    def get_message(self, i=None):
        """ Get the message from the cache.

        Parameters
        ----------
        i : int, optional
            The number of the message. If not provided, return the
            highest-numbered message.

        Returns
        -------
        message : dict

        Raises
        ------
        IndexError if the message does not exist in the cache.
        """


class SimpleMessageCache(object):
    """ Simple dictionary-based, in-memory storage of the responses from the
    interpreter.
    """

    def __init__(self):
        self.cache = {}

    def add_message(self, i, message):
        """ Add a message dictionary to the cache.

        Parameters
        ----------
        i : int
        message : dict
        """

        self.cache[i] = message

    def get_message(self, i=None):
        """ Get the message from the cache.

        Parameters
        ----------
        i : int, optional
            The number of the message. If not provided, return the
            highest-numbered message.

        Returns
        -------
        message : dict

        Raises
        ------
        IndexError if the message does not exist in the cache.
        """
        if i is None:
            keys = self.cache.keys()
            if len(keys) == 0:
                raise IndexError("index %r out of range" % i)
            else:
                i = max(self.cache.keys())
        try:
            return self.cache[i]
        except KeyError:
            # IndexError is more appropriate, here.
            raise IndexError("index %r out of range" % i)