mirror of
https://github.com/Novartis/cellxgene-gateway.git
synced 2026-09-15 12:47:59 +08:00
85 lines
3.5 KiB
Python
85 lines
3.5 KiB
Python
# 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.
|
|
|
|
import os
|
|
from cellxgene_gateway import env
|
|
from cellxgene_gateway.dir_util import make_h5ad, make_annotations, annotations_suffix
|
|
|
|
def recurse_dir(path):
|
|
if not os.path.exists(path):
|
|
raise CellxgeneException(
|
|
"The given path does not exist.", status.HTTP_400_BAD_REQUEST
|
|
)
|
|
|
|
all_entries = os.listdir(path)
|
|
def is_h5ad(el):
|
|
return el.endswith('.h5ad') and os.path.isfile(os.path.join(path, el))
|
|
h5ad_entries = [x for x in all_entries if is_h5ad(x)]
|
|
annotation_dir_entries = [x for x in all_entries if x.endswith(annotations_suffix) and make_h5ad(x) in h5ad_entries]
|
|
def list_annotations(el):
|
|
full_path = os.path.join(path, el)
|
|
if not os.path.isdir(full_path):
|
|
entries = []
|
|
else:
|
|
entries = [{
|
|
"name": x[:-13] if (len(x) > 13 and x[-13] in ['-','_']) else (
|
|
x[:-4] if x.endswith('.csv') else x),
|
|
"path": os.path.join(full_path, x).replace(env.cellxgene_data, ""),
|
|
} for x in os.listdir(full_path) if x.endswith('.csv') and os.path.isfile(os.path.join(full_path, x))]
|
|
return [{"name":'new', "class":'new', "path":full_path.replace(env.cellxgene_data, "")}] + entries
|
|
|
|
def make_entry(el):
|
|
full_path = os.path.join(path, el)
|
|
if el in h5ad_entries:
|
|
return {
|
|
"path": full_path.replace(env.cellxgene_data, ""),
|
|
"name": el,
|
|
"type": "file",
|
|
"annotations": list_annotations(make_annotations(el)),
|
|
}
|
|
elif os.path.isdir(full_path) and el not in annotation_dir_entries:
|
|
return {
|
|
"path": full_path.replace(env.cellxgene_data, ""),
|
|
"name": el,
|
|
"type": "directory",
|
|
"children": recurse_dir(full_path),
|
|
}
|
|
else:
|
|
return {
|
|
"path": full_path,
|
|
"name": el,
|
|
"type": "neither",
|
|
}
|
|
|
|
return [make_entry(x) for x in os.listdir(path)]
|
|
|
|
|
|
def render_entries(entries):
|
|
return "<ul>" + "\n".join([render_entry(e) for e in entries]) + "</ul>"
|
|
|
|
def get_url(entry):
|
|
return f"/view/{ entry['path'].lstrip('/') }"
|
|
def get_class(entry):
|
|
return f" class='{entry['class']}'" if 'class' in entry else ''
|
|
|
|
def render_annotations(entry):
|
|
if len(entry['annotations']) > 0:
|
|
return ' | annotations: ' + ", ".join([f"<a href='{get_url(a)}'{get_class(a)}>{a['name']}</a>" for a in entry['annotations']])
|
|
else:
|
|
return ''
|
|
|
|
def render_entry(entry):
|
|
if entry["type"] == "file":
|
|
return f"<li> <a href='{ get_url(entry) }'>{entry['name']}</a> {render_annotations(entry)}</li>"
|
|
elif entry["type"] == "directory":
|
|
url = f"/filecrawl/{entry['path'].lstrip('/')}"
|
|
return f"<li><a href='{url}'>{entry['name']}</a>{render_entries(entry['children'])}</li>"
|
|
else:
|
|
return ""
|