|
@@ -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()
|