import itertools
su_m = (lambda f: f(f))(lambda rec: sum(itertools.takewhile(lambda x: x != 0, (int(input()) for _ in itertools.count()))))
print(su_m)
Если будет просить код в формате PEP8, то используйте - https://formatter.org/python-formatter