Merge pull request #356 from stripe/ob-file-upload-stringio

Allow uploading in-memory files

stripe/api_resources/file_upload.py

@@ -1,4 +1,5 @@
-from stripe import api_requestor, upload_api_base, util
+import stripe
+from stripe import api_requestor, util
 from stripe.api_resources.abstract import ListableAPIResource
 
 
@@ -7,7 +8,7 @@ class FileUpload(ListableAPIResource):
 
     @classmethod
     def api_base(cls):
-        return upload_api_base
+        return stripe.upload_api_base
 
     @classmethod
     def class_name(cls):

stripe/multipart_data_generator.py

@@ -18,10 +18,12 @@ def add_params(self, params):
             self._write(self.param_header())
             self._write(self.line_break)
             if hasattr(value, 'read'):
+                filename = value.name if hasattr(value, 'name') else "blob"
+
                 self._write("Content-Disposition: form-data; name=\"")
                 self._write(key)
                 self._write("\"; filename=\"")
-                self._write(value.name)
+                self._write(filename)
                 self._write("\"")
                 self._write(self.line_break)
                 self._write("Content-Type: application/octet-stream")

stripe/test/test_multipart_data_generator.py

@@ -2,6 +2,7 @@
 
 import re
 import sys
+import StringIO
 
 from stripe.multipart_data_generator import MultipartDataGenerator
 from stripe.test.helper import StripeTestCase
@@ -50,3 +51,7 @@ def test_multipart_data_file_text(self):
     def test_multipart_data_file_binary(self):
         with open(__file__, mode='rb') as test_file:
             self.run_test_multipart_data_with_file(test_file)
+
+    def test_multipart_data_stringio(self):
+        string = StringIO.StringIO("foo")
+        self.run_test_multipart_data_with_file(string)