Browse Source

Making readFileToBuffer to download to io.BytesIO

Andrej Studen/Merlin 4 years ago
parent
commit
19f4171b12
1 changed files with 9 additions and 2 deletions
  1. 9 2
      labkeyFileBrowser.py

+ 9 - 2
labkeyFileBrowser.py

@@ -1,6 +1,7 @@
 import urllib3
 import xml.etree.ElementTree as ET
 import shutil
+import io
 
 class labkeyFileBrowser:
 
@@ -71,10 +72,16 @@ class labkeyFileBrowser:
 
 
     def readFileToBuffer(self, url):
-        return self.net.get(url,binary=True)
+        response=self.net.get(url,binary=True)
+        buffer=io.BytesIO()
+        for chunk in resp.stream(1024):
+            buffer.write(chunk)
+        resp.release_conn()
+        buffer.seek(0,0)
+        return buffer
 
     def readFileToFile(self,url,path):
-        response=self.readFileToBuffer(url)
+        response=self.net.get(url,binary=True)
         with open(path,'wb') as out_file:
             shutil.copyfileobj(response,out_file)
         response.release_conn()