An asyncronous iterator which automatically closes the underlying iterator when done

aclosing.py

#
# Python's `async for` only iterates and won't call `aclose()` on an Async Generator when done.
# Although `aclose()` is automatically scheduled to happen when the iterator goes out of scope, this will
# happen on a different Task.
# If iteration over an Async Generator is interrupted somehow, or left unfinished, that can be problematic

import contextlib
@contextlib.asynccontextmanager
async def aclosingiterator(iterable):
    iter = iterable.__aiter__()
    try:
        yield iter
    finally:
        if hasattr(iter, "aclose"):
            await iter.aclose()