summaryrefslogtreecommitdiff
path: root/templates/python/context.py
blob: 8e891953a358a864bae0be2c88184e284646b4fa (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.info("Context: {}={!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)