Merge pull request #3 from ConvertAPI/feature/result_file_io

Allow getting result file io

Author: laurynas-convertapi

convertapi/client.py

@@ -1,6 +1,7 @@
 import requests
 import convertapi
 
+from io import BytesIO
 from .exceptions import *
 
 class Client:
@@ -36,6 +37,10 @@ def download(self, url, path):
 
 		return path
 
+	def download_io(self, url):
+		response = requests.get(url, timeout = convertapi.download_timeout)
+		return BytesIO(response.content)
+
 	def handle_response(self, r):
 		try:
 			r.raise_for_status()

convertapi/result_file.py

@@ -17,6 +17,13 @@ def filename(self):
     def size(self):
         return self.info['FileSize']
 
+    @property
+    def io(self):
+        if not hasattr(self, '_io'):
+            self._io = convertapi.client.download_io(self.url)
+
+        return self._io
+
     def save(self, path):
         if os.path.isdir(path):
             path = os.path.join(path, self.filename)

tests/test_convertapi.py

@@ -37,6 +37,10 @@ def test_upload_io(self):
 		result = convertapi.convert('pdf', { 'File': upload_io })
 		assert result.conversion_cost > 0
 
+	def test_download_io(self):
+		result = convertapi.convert('pdf', { 'Url': 'https://www.w3.org/TR/PNG/iso_8859-1.txt' })
+		assert len(result.file.io.getvalue()) > 0
+
 	def test_zip_files(self):
 		files = ['examples/files/test.docx', 'examples/files/test.docx']
 		result = convertapi.convert('zip', { 'Files': files })