summaryrefslogtreecommitdiff
path: root/templates/python/context.py
blob: fac49de7716bb7fc31ca465b1c2934b5633d1e55 (plain)
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
import logging


# Store context between steps.
class Context:
    def __init__(self):
        self._vars = {}

    def as_dict(self):
        return dict(self._vars)

    def get(self, key, default=None):
        return self._vars.get(key, default)

    def __getitem__(self, key):
        return self._vars[key]

    def __setitem__(self, key, value):
        logging.debug("Context: key {!r} set to {!r}".format(key, value))
        self._vars[key] = value

    def __contains__(self, key):
        return key in self._vars

    def __delitem__(self, key):
        del self._vars[key]

    def __repr__(self):
        return repr(self._vars)