Skip to content

Commit 686a5aa

Browse files
committed
TR updates, second round
1 parent 2c5c2db commit 686a5aa

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

python-async-iterators/async_csv.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,20 @@
77
class AsyncCSVIterator:
88
def __init__(self, path):
99
self.path = path
10-
self.file = None
10+
self.file_was_read = False
1111

1212
def __aiter__(self):
1313
return self
1414

1515
async def __anext__(self):
16-
if self.file is None:
17-
self.file = await aiofiles.open(self.path, mode="r")
18-
lines = await self.file.readlines()
19-
self.reader = csv.reader(lines)
16+
if not self.file_was_read:
17+
async with aiofiles.open(self.path, mode="r") as file:
18+
lines = await file.readlines()
19+
self.reader = csv.reader(lines)
20+
self.file_was_read = True
2021
try:
2122
return next(self.reader)
2223
except StopIteration:
23-
await self.file.close()
2424
raise StopAsyncIteration
2525

2626

0 commit comments

Comments
 (0)