Compare commits

...
7 Commits
Author SHA1 Message Date
Alok Saldanha d5246d4b4b prepared for 0.2.2 release 2020-09-08 21:11:59 -04:00
Alok Saldanha 465373ca7e #30 black formatting 2020-09-08 21:11:59 -04:00
Alok Saldanha fc83523c84 #30 added missing js asset 2020-09-08 21:07:18 -04:00
Alok Saldanha 4e367b0ec9 update travis.yml to new conda env 2020-08-30 13:53:57 -04:00
Alok Saldanha ce3eb0eedf Merge branch 'gmerge' into gmaster
# Conflicts:
#	cellxgene_gateway/backend_cache.py
#	cellxgene_gateway/cache_entry.py
#	cellxgene_gateway/env.py
#	cellxgene_gateway/gateway.py
#	cellxgene_gateway/subprocess_backend.py
#	tests/test_cache_entry.py
#	tests/test_dir_util.py
2020-08-30 13:47:22 -04:00
Alok Saldanha 9219336356 blackened code 2020-08-30 13:38:35 -04:00
Alok Saldanha b689357f29 Applied "Formatting and new enumeration for Cellxgene-gateway" patch 2020-08-30 13:36:39 -04:00
19 changed files with 133 additions and 45 deletions
+2 -2
View File
@@ -7,7 +7,7 @@
language: python language: python
matrix: matrix:
include: include:
- python: 3.5 # we don't actually use this - python: 3.5 # we don't actually use this
env: PYTHON_VERSION=3.7 env: PYTHON_VERSION=3.7
install: install:
@@ -26,7 +26,7 @@ install:
# Install Python, py.test, and required packages. # Install Python, py.test, and required packages.
- conda env create -f environment.yml - conda env create -f environment.yml
- source activate cellxgene-dev - source activate cellxgene-gateway
- python setup.py install - python setup.py install
script: script:
+4
View File
@@ -1,3 +1,7 @@
# 0.2.2
* Fixed bug with annotations (missing annotation.js asset)
# 0.2.1 # 0.2.1
* Minor fixes to enable cellxgene 0.16.0 * Minor fixes to enable cellxgene 0.16.0
+1 -1
View File
@@ -7,4 +7,4 @@
# OR CONDITIONS OF ANY KIND, either express or implied. See the License for # OR CONDITIONS OF ANY KIND, either express or implied. See the License for
# the specific language governing permissions and limitations under the License. # the specific language governing permissions and limitations under the License.
__version__ = "0.2.1" __version__ = "0.2.2"
+2 -2
View File
@@ -13,7 +13,7 @@ from threading import Thread
from flask_api import status from flask_api import status
from cellxgene_gateway import env from cellxgene_gateway import env
from cellxgene_gateway.cache_entry import CacheEntry from cellxgene_gateway.cache_entry import CacheEntry, CacheEntryStatus
from cellxgene_gateway.cellxgene_exception import CellxgeneException from cellxgene_gateway.cellxgene_exception import CellxgeneException
from cellxgene_gateway.subprocess_backend import SubprocessBackend from cellxgene_gateway.subprocess_backend import SubprocessBackend
@@ -42,7 +42,7 @@ class BackendCache:
for c in contents for c in contents
if c.key.dataset == key.dataset if c.key.dataset == key.dataset
and c.key.annotation_file == key.annotation_file and c.key.annotation_file == key.annotation_file
and c.status != "terminated" and c.status != CacheEntryStatus.terminated
] ]
if len(matches) == 0: if len(matches) == 0:
+28 -14
View File
@@ -6,18 +6,26 @@
# under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES # 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 # OR CONDITIONS OF ANY KIND, either express or implied. See the License for
# the specific language governing permissions and limitations under the License. # the specific language governing permissions and limitations under the License.
import psutil
import logging
import datetime import datetime
import logging
from flask import make_response, request, render_template import psutil
from enum import Enum
from flask import make_response, render_template, request
from requests import get, post, put from requests import get, post, put
import re import re
from cellxgene_gateway import env from cellxgene_gateway import env
from cellxgene_gateway.cellxgene_exception import CellxgeneException from cellxgene_gateway.cellxgene_exception import CellxgeneException
from cellxgene_gateway.util import current_time_stamp
from cellxgene_gateway.flask_util import querystring from cellxgene_gateway.flask_util import querystring
from cellxgene_gateway.util import current_time_stamp
class CacheEntryStatus(Enum):
loaded = "loaded"
loading = "loading"
error = "error"
terminated = "terminated"
class CacheEntry: class CacheEntry:
@@ -28,7 +36,7 @@ class CacheEntry:
port, port,
launchtime, launchtime,
timestamp, timestamp,
status, status: CacheEntryStatus,
message, message,
all_output, all_output,
stderr, stderr,
@@ -54,7 +62,7 @@ class CacheEntry:
port, port,
current_time_stamp(), current_time_stamp(),
current_time_stamp(), current_time_stamp(),
"loading", CacheEntryStatus.loading,
None, None,
None, None,
None, None,
@@ -63,13 +71,13 @@ class CacheEntry:
def set_loaded(self, pid): def set_loaded(self, pid):
self.pid = pid self.pid = pid
self.status = "loaded" self.status = CacheEntryStatus.loaded
def set_error(self, message, stderr, http_status): def set_error(self, message, stderr, http_status):
self.message = message self.message = message
self.stderr = stderr self.stderr = stderr
self.http_status = http_status self.http_status = http_status
self.status = "error" self.status = CacheEntryStatus.error
def append_output(self, output): def append_output(self, output):
if self.all_output == None: if self.all_output == None:
@@ -79,7 +87,7 @@ class CacheEntry:
def terminate(self): def terminate(self):
pid = self.pid pid = self.pid
if pid != None and self.status != "terminated": if pid != None and self.status != CacheEntryStatus.terminated:
terminated = [] terminated = []
def on_terminate(p): def on_terminate(p):
@@ -96,7 +104,7 @@ class CacheEntry:
logging.getLogger("cellxgene_gateway").info( logging.getLogger("cellxgene_gateway").info(
f"terminated {terminated}" f"terminated {terminated}"
) )
self.status = "terminated" self.status = CacheEntryStatus.terminated
def rewrite_text_content(self, cellxgene_content): def rewrite_text_content(self, cellxgene_content):
# for v0.16.0 compatibility, see issue #24 # for v0.16.0 compatibility, see issue #24
@@ -125,7 +133,7 @@ class CacheEntry:
r = make_response(f"Redirect to {gateway_basepath}\n", 301) r = make_response(f"Redirect to {gateway_basepath}\n", 301)
r.headers["location"] = gateway_basepath + querystring() r.headers["location"] = gateway_basepath + querystring()
return r return r
elif self.status == "loading": elif self.status == CacheEntryStatus.loading:
launch_time = datetime.datetime.fromtimestamp(self.launchtime) launch_time = datetime.datetime.fromtimestamp(self.launchtime)
return render_template( return render_template(
"loading.html", "loading.html",
@@ -161,11 +169,15 @@ class CacheEntry:
cellxgene_response = get(full_path, headers=headers) cellxgene_response = get(full_path, headers=headers)
elif request.method == "PUT": elif request.method == "PUT":
cellxgene_response = put( cellxgene_response = put(
full_path, headers=headers, data=request.data, full_path,
headers=headers,
data=request.data,
) )
elif request.method == "POST": elif request.method == "POST":
cellxgene_response = post( cellxgene_response = post(
full_path, headers=headers, data=request.data, full_path,
headers=headers,
data=request.data,
) )
else: else:
raise CellxgeneException( raise CellxgeneException(
@@ -185,7 +197,9 @@ class CacheEntry:
resp_headers[h] = cellxgene_response.headers[h] resp_headers[h] = cellxgene_response.headers[h]
gateway_response = make_response( gateway_response = make_response(
gateway_content, cellxgene_response.status_code, resp_headers, gateway_content,
cellxgene_response.status_code,
resp_headers,
) )
return gateway_response return gateway_response
+2 -2
View File
@@ -10,8 +10,8 @@
# There are three kinds of CacheKey: # There are three kinds of CacheKey:
# 1) somedir/dataset.h5ad: a dataset # 1) somedir/dataset.h5ad: a dataset
# in this case, pathpart == dataset == 'somedir/dataset.h5ad' # in this case, pathpart == dataset == 'somedir/dataset.h5ad'
# 2) somedir/dataset_annotations/saldaal1-T5HMVBNV.csv : an actual annotaitons file. # 2) somedir/dataset_annotations/my_annotations.csv : an actual annotaitons file.
# in this case, pathpart == 'dataset_annotations/saldaal1-T5HMVBNV.csv', dataset == 'somedir/dataset.h5ad' # in this case, pathpart == 'dataset_annotations/my_annotations.csv', dataset == 'somedir/dataset.h5ad'
# 3) somedir/dataset_annotations: an annotation directory. The corresponding h5ad must exist, but the directory may not. # 3) somedir/dataset_annotations: an annotation directory. The corresponding h5ad must exist, but the directory may not.
# in this case, pathpart == 'dataset_annotations', dataset == 'somedir/dataset.h5ad' # in this case, pathpart == 'dataset_annotations', dataset == 'somedir/dataset.h5ad'
+9 -3
View File
@@ -7,8 +7,8 @@
# OR CONDITIONS OF ANY KIND, either express or implied. See the License for # OR CONDITIONS OF ANY KIND, either express or implied. See the License for
# the specific language governing permissions and limitations under the License. # the specific language governing permissions and limitations under the License.
import os
import logging import logging
import os
import socket import socket
cellxgene_location = os.environ.get("CELLXGENE_LOCATION") cellxgene_location = os.environ.get("CELLXGENE_LOCATION")
@@ -31,10 +31,16 @@ enable_upload = os.environ.get("GATEWAY_ENABLE_UPLOAD", "").lower() in [
] ]
enable_annotations = os.environ.get( enable_annotations = os.environ.get(
"GATEWAY_ENABLE_ANNOTATIONS", "" "GATEWAY_ENABLE_ANNOTATIONS", ""
).lower() in ["true", "1"] ).lower() in [
"true",
"1",
]
enable_backed_mode = os.environ.get( enable_backed_mode = os.environ.get(
"GATEWAY_ENABLE_BACKED_MODE", "" "GATEWAY_ENABLE_BACKED_MODE", ""
).lower() in ["true", "1"] ).lower() in [
"true",
"1",
]
env_vars = { env_vars = {
"CELLXGENE_LOCATION": cellxgene_location, "CELLXGENE_LOCATION": cellxgene_location,
+2 -1
View File
@@ -7,9 +7,10 @@
# OR CONDITIONS OF ANY KIND, either express or implied. See the License for # OR CONDITIONS OF ANY KIND, either express or implied. See the License for
# the specific language governing permissions and limitations under the License. # the specific language governing permissions and limitations under the License.
from cellxgene_gateway import env
from json import loads from json import loads
from cellxgene_gateway import env
def get_extra_scripts(): def get_extra_scripts():
# can be array of script tags to inject on every page, e.g. for google analytics could be # can be array of script tags to inject on every page, e.g. for google analytics could be
+13 -8
View File
@@ -7,16 +7,17 @@
# OR CONDITIONS OF ANY KIND, either express or implied. See the License for # OR CONDITIONS OF ANY KIND, either express or implied. See the License for
# the specific language governing permissions and limitations under the License. # the specific language governing permissions and limitations under the License.
import json
import logging
# import BaseHTTPServer # import BaseHTTPServer
import os import os
import logging from threading import Lock, Thread
from threading import Thread, Lock
import json
from flask import ( from flask import (
Flask, Flask,
redirect,
make_response, make_response,
redirect,
render_template, render_template,
request, request,
send_from_directory, send_from_directory,
@@ -27,14 +28,15 @@ from werkzeug.utils import secure_filename
from cellxgene_gateway import env from cellxgene_gateway import env
from cellxgene_gateway.backend_cache import BackendCache from cellxgene_gateway.backend_cache import BackendCache
from cellxgene_gateway.cache_entry import CacheEntryStatus
from cellxgene_gateway.cellxgene_exception import CellxgeneException from cellxgene_gateway.cellxgene_exception import CellxgeneException
from cellxgene_gateway.dir_util import create_dir, is_subdir from cellxgene_gateway.dir_util import create_dir, is_subdir
from cellxgene_gateway.filecrawl import recurse_dir, render_entries
from cellxgene_gateway.extra_scripts import get_extra_scripts from cellxgene_gateway.extra_scripts import get_extra_scripts
from cellxgene_gateway.filecrawl import recurse_dir, render_entries
from cellxgene_gateway.path_util import get_key
from cellxgene_gateway.process_exception import ProcessException from cellxgene_gateway.process_exception import ProcessException
from cellxgene_gateway.prune_process_cache import PruneProcessCache from cellxgene_gateway.prune_process_cache import PruneProcessCache
from cellxgene_gateway.util import current_time_stamp from cellxgene_gateway.util import current_time_stamp
from cellxgene_gateway.path_util import get_key
app = Flask(__name__) app = Flask(__name__)
@@ -238,9 +240,12 @@ def do_view(path):
match.timestamp = current_time_stamp() match.timestamp = current_time_stamp()
if match.status == "loaded" or match.status == "loading": if (
match.status == CacheEntryStatus.loaded
or match.status == CacheEntryStatus.loading
):
return match.serve_content(path) return match.serve_content(path)
elif match.status == "error": elif match.status == CacheEntryStatus.error:
raise ProcessException.from_cache_entry(match) raise ProcessException.from_cache_entry(match)
+2 -2
View File
@@ -12,9 +12,9 @@ import os
from flask_api import status from flask_api import status
from cellxgene_gateway import env from cellxgene_gateway import env
from cellxgene_gateway.cache_key import CacheKey
from cellxgene_gateway.cellxgene_exception import CellxgeneException from cellxgene_gateway.cellxgene_exception import CellxgeneException
from cellxgene_gateway.dir_util import make_h5ad from cellxgene_gateway.dir_util import make_h5ad
from cellxgene_gateway.cache_key import CacheKey
def get_key(path): def get_key(path):
@@ -31,7 +31,7 @@ def get_key(path):
return CacheKey(trimmed, trimmed, None) return CacheKey(trimmed, trimmed, None)
elif trimmed.endswith(".csv"): elif trimmed.endswith(".csv"):
# 2) somedir/dataset_annotations/saldaal1-T5HMVBNV.csv : an actual annotations file. # 2) somedir/dataset_annotations/my_annotations.csv : an actual annotations file.
annotations_dir = os.path.split(trimmed)[0] annotations_dir = os.path.split(trimmed)[0]
dataset = make_h5ad(annotations_dir) dataset = make_h5ad(annotations_dir)
if data_file_exists(dataset): if data_file_exists(dataset):
+2 -2
View File
@@ -7,11 +7,11 @@
# OR CONDITIONS OF ANY KIND, either express or implied. See the License for # OR CONDITIONS OF ANY KIND, either express or implied. See the License for
# the specific language governing permissions and limitations under the License. # the specific language governing permissions and limitations under the License.
import time
import logging import logging
import time
from cellxgene_gateway.util import current_time_stamp
from cellxgene_gateway.env import ttl from cellxgene_gateway.env import ttl
from cellxgene_gateway.util import current_time_stamp
class PruneProcessCache: class PruneProcessCache:
+4 -3
View File
@@ -11,14 +11,15 @@ import logging
import subprocess import subprocess
from flask_api import status from flask_api import status
from cellxgene_gateway.cache_entry import CacheEntryStatus
from cellxgene_gateway.dir_util import make_annotations
from cellxgene_gateway.path_util import get_annotation_file_path, get_file_path
from cellxgene_gateway.env import ( from cellxgene_gateway.env import (
enable_annotations, enable_annotations,
enable_backed_mode, enable_backed_mode,
cellxgene_args, cellxgene_args,
) )
from cellxgene_gateway.process_exception import ProcessException from cellxgene_gateway.process_exception import ProcessException
from cellxgene_gateway.dir_util import make_annotations
from cellxgene_gateway.path_util import get_file_path, get_annotation_file_path
class SubprocessBackend: class SubprocessBackend:
@@ -85,7 +86,7 @@ class SubprocessBackend:
message = "Cellxgene failed to launch dataset." message = "Cellxgene failed to launch dataset."
http_status = status.HTTP_500_INTERNAL_SERVER_ERROR http_status = status.HTTP_500_INTERNAL_SERVER_ERROR
cache_entry.status = "error" cache_entry.status = CacheEntryStatus.error
cache_entry.set_error(message, stderr, http_status) cache_entry.set_error(message, stderr, http_status)
raise ProcessException.from_cache_entry(cache_entry) raise ProcessException.from_cache_entry(cache_entry)
@@ -48,11 +48,11 @@
<td>{{ entry.port }}</td> <td>{{ entry.port }}</td>
<td class="timestamp">{{ entry.launchtime }}</td> <td class="timestamp">{{ entry.launchtime }}</td>
<td class="timestamp">{{ entry.timestamp }}</td> <td class="timestamp">{{ entry.timestamp }}</td>
<td>{{ entry.status }}</td> <td>{{ entry.status.name }}</td>
<td>{{ entry.message }}</td> <td>{{ entry.message }}</td>
<td>{{ entry.http_status }}</td> <td>{{ entry.http_status }}</td>
<td> <td>
{% if entry.status == 'loaded' %} {% if entry.status.name == 'loaded' %}
<a href="{{ url_for('do_terminate', path=entry.key.pathpart) }}"> terminate </a> <a href="{{ url_for('do_terminate', path=entry.key.pathpart) }}"> terminate </a>
{% endif %} {% endif %}
</td> </td>
+1 -1
View File
@@ -1,4 +1,4 @@
name: cellxgene-dev name: cellxgene-gateway
channels: channels:
- conda-forge - conda-forge
dependencies: dependencies:
+1
View File
@@ -53,6 +53,7 @@ setup(
package_data={ package_data={
"cellxgene_gateway": [ "cellxgene_gateway": [
"static/css/homepagestyle.css", "static/css/homepagestyle.css",
"static/js/annotation.js",
"static/nibr.ico", "static/nibr.ico",
"templates/*.html", "templates/*.html",
] ]
+9 -1
View File
@@ -1,11 +1,15 @@
import unittest import unittest
from cellxgene_gateway.cache_entry import CacheEntry from cellxgene_gateway.cache_entry import CacheEntry, CacheEntryStatus
from cellxgene_gateway.cache_key import CacheKey from cellxgene_gateway.cache_key import CacheKey
key = CacheKey("czi/pbmc3k.h5ad", "pbmc3k.h5ad", "tmp.csv") key = CacheKey("czi/pbmc3k.h5ad", "pbmc3k.h5ad", "tmp.csv")
class TestRenderEntry(unittest.TestCase): class TestRenderEntry(unittest.TestCase):
def test_GIVEN_key_and_port_THEN_returns_loading_CacheEntry(self):
entry = CacheEntry.for_key("some-key", 1)
self.assertEqual(entry.status, CacheEntryStatus.loading)
def test_GIVEN_absolute_static_url_THEN_include_path(self): def test_GIVEN_absolute_static_url_THEN_include_path(self):
actual = CacheEntry.for_key(key, 8000).rewrite_text_content( actual = CacheEntry.for_key(key, 8000).rewrite_text_content(
"src:url(/static/assets/" "src:url(/static/assets/"
@@ -21,3 +25,7 @@ class TestRenderEntry(unittest.TestCase):
) )
expected = '<link rel="shortcut icon" href="http://localhost:5005/view/czi/pbmc3k.h5ad/static/assets/favicon.ico">' expected = '<link rel="shortcut icon" href="http://localhost:5005/view/czi/pbmc3k.h5ad/static/assets/favicon.ico">'
self.assertEqual(actual, expected) self.assertEqual(actual, expected)
if __name__ == "__main__":
unittest.main()
+1
View File
@@ -1,5 +1,6 @@
import unittest import unittest
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
from cellxgene_gateway.extra_scripts import get_extra_scripts from cellxgene_gateway.extra_scripts import get_extra_scripts
+46
View File
@@ -0,0 +1,46 @@
import unittest
from unittest.mock import MagicMock, patch
from cellxgene_gateway.filecrawl import render_entry
class TestRenderEntry(unittest.TestCase):
def test_GIVEN_path_both_slash_THEN_view_has_single_slash(self):
entry = {
"path": "/somepath/",
"name": "entry",
"type": "file",
"annotations": [],
}
rendered = render_entry(entry)
self.assertIn("view/somepath", rendered)
def test_GIVEN_path_starts_slash_THEN_view_has_single_slash(self):
entry = {
"path": "/somepath",
"name": "entry",
"type": "file",
"annotations": [],
}
rendered = render_entry(entry)
self.assertIn("view/somepath", rendered)
def test_GIVEN_path_ends_slash_THEN_view_has_single_slash(self):
entry = {
"path": "somepath/",
"name": "entry",
"type": "file",
"annotations": [],
}
rendered = render_entry(entry)
self.assertIn("view/somepath", rendered)
def test_GIVEN_path_no_slash_THEN_view_has_single_slash(self):
entry = {
"path": "somepath",
"name": "entry",
"type": "file",
"annotations": [],
}
rendered = render_entry(entry)
self.assertIn("view/somepath", rendered)
+2 -1
View File
@@ -1,7 +1,8 @@
import unittest import unittest
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
from cellxgene_gateway.cache_entry import CacheEntry
from cellxgene_gateway.backend_cache import BackendCache from cellxgene_gateway.backend_cache import BackendCache
from cellxgene_gateway.cache_entry import CacheEntry
class TestPruneProcessCache(unittest.TestCase): class TestPruneProcessCache(unittest.TestCase):