Use a context manager as suggested by a comment.