def preprocess_stream(stream, charset): if six.PY2: # csv.py doesn't do Unicode; encode temporarily: return (chunk.encode(charset) for chunk in stream) else: return stream