def __enter__(self):
        if self.db.transaction_depth() == 0:
            self._helper = self.db.transaction()
        else:
            self._helper = self.db.savepoint()
        return self._helper.__enter__()
dataset/ETHPy150Open coleifer/peewee/peewee.py/_atomic.__enter__/original