Final QA (#564)

realpython · Aug 7, 2024 · 5dd42f8 · 5dd42f8
1 parent da8e557
commit 5dd42f8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/python-async-iterators/async_csv.py b/python-async-iterators/async_csv.py
@@ -7,17 +7,16 @@
 class AsyncCSVIterator:
     def __init__(self, path):
         self.path = path
-        self.file_was_read = False
+        self.reader = None
 
     def __aiter__(self):
         return self
 
     async def __anext__(self):
-        if not self.file_was_read:
+        if self.reader is None:
             async with aiofiles.open(self.path, mode="r") as file:
                 lines = await file.readlines()
                 self.reader = csv.reader(lines)
-                self.file_was_read = True
         try:
             return next(self.reader)
         except StopIteration:

diff --git a/python-async-iterators/compress.py b/python-async-iterators/compress.py
@@ -18,5 +18,5 @@ async def main(directory, zip_name="output.zip"):
             await archive.write(chunk)
 
 
-directory = Path()
+directory = Path.cwd()
 asyncio.run(main(directory))