import urllib from twisted.web import http from twisted.internet import defer from nevow import rend, url, tags as T from allmydata.immutable.upload import FileHandle from allmydata.web.common import getxmlfile, get_arg, boolean_of_arg, \ convert_initial_children_json from allmydata.web import status def PUTUnlinkedCHK(req, client): # "PUT /uri", to create an unlinked file. uploadable = FileHandle(req.content, client.convergence) d = client.upload(uploadable) d.addCallback(lambda results: results.uri) # that fires with the URI of the new file return d def PUTUnlinkedSSK(req, client): # SDMF: files are small, and we can only upload data req.content.seek(0) data = req.content.read() d = client.create_mutable_file(data) d.addCallback(lambda n: n.get_uri()) return d def PUTUnlinkedCreateDirectory(req, client): # "PUT /uri?t=mkdir", to create an unlinked directory. req.content.seek(0) kids_json = req.content.read() kids = convert_initial_children_json(client.nodemaker, kids_json) d = client.create_dirnode(initial_children=kids) d.addCallback(lambda dirnode: dirnode.get_uri()) # XXX add redirect_to_result return d def POSTUnlinkedCHK(req, client): fileobj = req.fields["file"].file uploadable = FileHandle(fileobj, client.convergence) d = client.upload(uploadable) when_done = get_arg(req, "when_done", None) if when_done: # if when_done= is provided, return a redirect instead of our # usual upload-results page def _done(upload_results, redir_to): if "%(uri)s" in redir_to: redir_to = redir_to % {"uri": urllib.quote(upload_results.uri) } return url.URL.fromString(redir_to) d.addCallback(_done, when_done) else: # return the Upload Results page, which includes the URI d.addCallback(UploadResultsPage) return d class UploadResultsPage(status.UploadResultsRendererMixin, rend.Page): """'POST /uri', to create an unlinked file.""" docFactory = getxmlfile("upload-results.xhtml") def __init__(self, upload_results): rend.Page.__init__(self) self.results = upload_results def upload_results(self): return defer.succeed(self.results) def data_done(self, ctx, data): d = self.upload_results() d.addCallback(lambda res: "done!") return d def data_uri(self, ctx, data): d = self.upload_results() d.addCallback(lambda res: res.uri) return d def render_download_link(self, ctx, data): d = self.upload_results() d.addCallback(lambda res: T.a(href="/uri/" + urllib.quote(res.uri)) ["/uri/" + res.uri]) return d def POSTUnlinkedSSK(req, client): # "POST /uri", to create an unlinked file. # SDMF: files are small, and we can only upload data contents = req.fields["file"] contents.file.seek(0) data = contents.file.read() d = client.create_mutable_file(data) d.addCallback(lambda n: n.get_uri()) return d def POSTUnlinkedCreateDirectory(req, client): # "POST /uri?t=mkdir", to create an unlinked directory. kids_json = get_arg(req, "children", "") kids = convert_initial_children_json(client.nodemaker, kids_json) d = client.create_dirnode(initial_children=kids) redirect = get_arg(req, "redirect_to_result", "false") if boolean_of_arg(redirect): def _then_redir(res): new_url = "uri/" + urllib.quote(res.get_uri()) req.setResponseCode(http.SEE_OTHER) # 303 req.setHeader('location', new_url) req.finish() return '' d.addCallback(_then_redir) else: d.addCallback(lambda dirnode: dirnode.get_uri()) return d