Added .upload(url) to FileRef, does a HTTP POST

Author: Elmo Todurov

spinoff/contrib/filetransfer/fileref.py

@@ -99,6 +99,9 @@ def fetch(self, dst_path=None, on_progress=None):
             assert reasonable_get_mtime(ret) == self.mtime
             return ret
 
+    def upload(self, url):
+        return self.server.ask(('upload', self.file_id, url))
+
     def _transfer(self, fh, on_progress):
         request = get_context().spawn(Request.using(server=self.server, file_id=self.file_id, size=self.size, abstract_path=self.abstract_path))
         more = True

spinoff/contrib/filetransfer/server.py

@@ -2,6 +2,7 @@
 import os
 
 from gevent.threadpool import ThreadPool
+import requests
 
 from spinoff.actor import Actor
 from spinoff.util.logging import dbg, err
@@ -51,6 +52,15 @@ def receive(self, msg):
         elif ('terminated', IN(self.responses)) == msg:
             _, sender = msg
             self._touch_file(file_id=self.responses.pop(sender))
+        elif ('upload', ANY, ANY) == msg:
+            _, file_id, url = msg
+            if file_id not in self.published:
+                self.reply((False, "Attempted to upload a file with ID %r which has not been published or is not available anymore" % (file_id,)))
+            else:
+                self._touch_file(file_id)
+                file_path, _ = self.published[file_id]
+                r = requests.post(url, files={'file': open(file_path, 'rb')})
+                self.reply((True, r.text))
 
     def _touch_file(self, file_id):
         file_path, time_added = self.published[file_id]