Merge pull request #22 from vlanse/master

Fix file handlers leak
mosquito · Dec 7, 2017 · b4bc59b · b4bc59b
2 parents f1924e1 + bce145a
commit b4bc59b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pypi_server/__init__.py b/pypi_server/__init__.py
@@ -9,7 +9,7 @@
     ("Dmitry Orlov", "me@mosquito.su")
 ]
 
-version_info = (0, 4, 6)
+version_info = (0, 4, 7)
 
 __version__ = ".".join(map(str, version_info))
 __author__ = ", ".join("{0} <{1}>".format(*author) for author in author_info)

diff --git a/pypi_server/handlers/pypi/package.py b/pypi_server/handlers/pypi/package.py
@@ -79,8 +79,10 @@ def get(self, package, version, filename):
                 self.write(data)
                 yield Task(self.flush)
 
-                while data:
-                    data = yield reader(self.CHUNK_SIZE)
+                for chunk in iter(lambda: reader(self.CHUNK_SIZE), None):
+                    data = yield chunk
+                    if not data:
+                        break
                     self.write(data)
                     yield Task(self.flush)