summaryrefslogtreecommitdiff
path: root/templates/python/context.py
blob: 9bb8056a80afe1521706bb831bb388b0628d1f01 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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