summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--module/obmc/wsgi/apps/rest_dbus.py36
1 files changed, 36 insertions, 0 deletions
diff --git a/module/obmc/wsgi/apps/rest_dbus.py b/module/obmc/wsgi/apps/rest_dbus.py
index e83d933..6ba2e28 100644
--- a/module/obmc/wsgi/apps/rest_dbus.py
+++ b/module/obmc/wsgi/apps/rest_dbus.py
@@ -26,6 +26,7 @@ import obmc.mapper
import spwd
import grp
import crypt
+import tempfile
DBUS_UNKNOWN_INTERFACE = 'org.freedesktop.UnknownInterface'
DBUS_UNKNOWN_INTERFACE_ERROR = 'org.freedesktop.DBus.Error.UnknownInterface'
@@ -531,6 +532,39 @@ class SessionHandler(MethodHandler):
pass
+class ImageUploadHandler(RouteHandler):
+ ''' Handles the /upload route. '''
+
+ verbs = ['POST']
+ rules = ['/upload/image']
+ content_type = 'application/octet-stream'
+ file_loc = '/tmp/images'
+ file_prefix = 'img'
+ file_suffix = ''
+
+ def __init__(self, app, bus):
+ super(ImageUploadHandler, self).__init__(
+ app, bus, self.verbs, self.rules, self.content_type)
+
+ def do_post(self, **kw):
+ self.do_upload(**kw)
+
+ def do_upload(self, **kw):
+ if not os.path.exists(self.file_loc):
+ os.makedirs(self.file_loc)
+ handle, filename = tempfile.mkstemp(self.file_suffix,
+ self.file_prefix, self.file_loc)
+
+ with os.fdopen(handle, "w") as fd:
+ fd.write(request.body.read())
+
+ def find(self, **kw):
+ pass
+
+ def setup(self, **kw):
+ pass
+
+
class AuthorizationPlugin(object):
''' Invokes an optional list of authorization callbacks. '''
@@ -836,6 +870,7 @@ class App(Bottle):
self.method_handler = MethodHandler(self, self.bus)
self.property_handler = PropertyHandler(self, self.bus)
self.schema_handler = SchemaHandler(self, self.bus)
+ self.image_upload_handler = ImageUploadHandler(self, self.bus)
self.instance_handler = InstanceHandler(self, self.bus)
def install_handlers(self):
@@ -846,6 +881,7 @@ class App(Bottle):
self.method_handler.install()
self.property_handler.install()
self.schema_handler.install()
+ self.image_upload_handler.install()
# this has to come last, since it matches everything
self.instance_handler.install()
OpenPOWER on IntegriCloud