Merge pull request #3 from tumb1er/content_from_response_iter

Record content via iterating over response
just-work · Apr 6, 2020 · 8b49386 · 8b49386
2 parents fe8aa27 + aa6c931
commit 8b49386
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sitemap_generate/generator.py b/sitemap_generate/generator.py
@@ -56,10 +56,10 @@ def record(self, url: str) -> bytes:
             'QUERY_STRING': url.query,
             'HTTP_X_FORWARDED_PROTO': defaults.SITEMAP_PROTO
         }
-        response = self.wsgi(environ, self._start_response)
+        content = b''.join(self.wsgi(environ, self._start_response))
         if self.status != "200 OK":
-            raise SitemapError(self.status, response.content)
-        return response.content
+            raise SitemapError(self.status, content)
+        return content
 
     def _start_response(self, status, _):
         """ WSGI headers callback func."""