[Python-ideas] Relax __exit__ method to be a generator