mirror of
https://github.com/chanzuckerberg/cellxgene.git
synced 2026-09-19 02:48:30 +08:00
5ea9ac8a9e
* add last mod time lookup to data locator * store data locator in Driver * save metadata header in labels csv * lint * fix tests * change datetime format to second precision
53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
"""
|
|
Helpers for user annotations / label_file parameter
|
|
"""
|
|
from os.path import exists, splitext, getsize
|
|
from os import remove, rename
|
|
import pandas as pd
|
|
|
|
|
|
def read_labels(fname):
|
|
if exists(fname) and getsize(fname) > 0:
|
|
return pd.read_csv(fname, dtype='category', index_col=0, header=0, comment='#')
|
|
else:
|
|
return pd.DataFrame()
|
|
|
|
|
|
def write_labels(fname, df, header=None):
|
|
rotate_fname(fname)
|
|
if not df.empty:
|
|
f = open(fname, 'a', newline="")
|
|
if header is not None:
|
|
f.write(header)
|
|
df.to_csv(f)
|
|
else:
|
|
open(fname, 'a').close()
|
|
|
|
|
|
def rotate_fname(fname):
|
|
"""
|
|
save N backups of file.
|
|
fname -> fname-0
|
|
fname-0 -> fname->1
|
|
...
|
|
fname-(N-1) -> fname-N
|
|
"""
|
|
|
|
def rotate(src, dst):
|
|
if exists(src):
|
|
if exists(dst):
|
|
remove(dst)
|
|
rename(src, dst)
|
|
|
|
rotation_size = 9 # rotation size
|
|
name, ext = splitext(fname)
|
|
|
|
# rotate existing files
|
|
for i in range(rotation_size - 1, 0, -1):
|
|
src = f"{name}-{i}{ext}"
|
|
tgt = f"{name}-{i+1}{ext}"
|
|
rotate(src, tgt)
|
|
|
|
tgt = f"{name}-1{ext}"
|
|
rotate(fname, tgt)
|