From 7f78b331facea96451d85f4453fa1da342ca3a83 Mon Sep 17 00:00:00 2001 From: Salandora Date: Thu, 8 Oct 2015 15:29:14 +0200 Subject: [PATCH] Added possibility to send a cookie back by the LargeResponseHandler --- src/octoprint/server/util/tornado.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/octoprint/server/util/tornado.py b/src/octoprint/server/util/tornado.py index ce1bdd00..69e245f9 100644 --- a/src/octoprint/server/util/tornado.py +++ b/src/octoprint/server/util/tornado.py @@ -789,12 +789,16 @@ class LargeResponseHandler(tornado.web.StaticFileHandler): self._access_validation(self.request) if self._path_validation is not None: self._path_validation(path) + + if "cookie" in self.request.arguments: + self.set_cookie(self.request.arguments["cookie"][0], "true", path="/") + result = tornado.web.StaticFileHandler.get(self, path, include_body=include_body) return result def set_extra_headers(self, path): if self._as_attachment: - self.set_header("Content-Disposition", "attachment") + self.set_header("Content-Disposition", "attachment; filename=%s" % os.path.basename(path)) @classmethod def get_content_version(cls, abspath):