the first commit

cache_entry.py

@@ -0,0 +1,108 @@
+# Copyright 2019 Novartis Institutes for BioMedical Research Inc. Licensed
+# under the Apache License, Version 2.0 (the "License"); you may not use
+# this file except in compliance with the License. You may obtain a copy
+# of the License at http://www.apache.org/licenses/LICENSE-2.0. Unless
+# required by applicable law or agreed to in writing, software distributed
+# under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES
+# OR CONDITIONS OF ANY KIND, either express or implied. See the License for
+# the specific language governing permissions and limitations under the License.
+
+from flask import Response, request
+from requests import get
+
+import env
+from util import current_time_stamp
+
+
+class CacheEntry:
+    def __init__(
+        self,
+        pid,
+        dataset,
+        file_path,
+        port,
+        launchtime,
+        timestamp,
+        status,
+        message,
+        all_output,
+        stderr,
+        http_status,
+    ):
+        self.pid = pid
+        self.dataset = dataset
+        self.file_path = file_path
+        self.port = port
+        self.launchtime = launchtime
+        self.timestamp = timestamp
+        self.status = status
+        self.message = message
+        self.all_output = all_output
+        self.stderr = stderr
+        self.http_status = http_status
+
+    @classmethod
+    def for_dataset(cls, dataset, file_path, port):
+        return cls(
+            "",
+            dataset,
+            file_path,
+            port,
+            current_time_stamp(),
+            current_time_stamp(),
+            "loading",
+            "",
+            "",
+            "",
+            "",
+        )
+
+    def set_loaded(self, pid):
+        self.pid = pid
+        self.status = "loaded"
+
+    def set_error(self, message, stderr, http_status):
+        self.message = message
+        self.stderr = stderr
+        self.http_status = http_status
+        self.status = "error"
+
+    def serve_content(self, path):
+        dataset = self.dataset
+
+        gateway_basepath = (
+            f"{env.gateway_protocol}://{env.gateway_host}/view/{dataset}/"
+        )
+        subpath = path[len(dataset) :]  # noqa: E203
+
+        if len(subpath) == 0:
+            r = Response(f"Redirect to {gateway_basepath}\n", status=301)
+            r.headers["location"] = gateway_basepath
+            return r
+
+        port = self.port
+        cellxgene_basepath = f"http://127.0.0.1:{port}"
+
+        headers = (
+            {"accept": request.headers["accept"]} if "accept" in request.headers else {}
+        )
+
+        cellxgene_response = get(cellxgene_basepath + subpath, headers=headers)
+
+        content_type = cellxgene_response.headers["content-type"]
+
+        if "text" in content_type:
+            cellxgene_content = cellxgene_response.content.decode()
+            gateway_content = cellxgene_content.replace(
+                "http://fonts.gstatic.com", "https://fonts.gstatic.com"
+            ).replace(cellxgene_basepath, gateway_basepath)
+        else:
+            gateway_content = cellxgene_response.content
+
+        gateway_response = Response(
+            gateway_content, status=cellxgene_response.status_code
+        )
+
+        gateway_response.headers["content-type"] = content_type
+
+        return gateway_response