mirror of
https://github.com/chanzuckerberg/cellxgene.git
synced 2026-09-19 02:48:30 +08:00
50 lines
1.0 KiB
Python
50 lines
1.0 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)
|
|
else:
|
|
return pd.DataFrame()
|
|
|
|
|
|
def write_labels(fname, df):
|
|
rotate_fname(fname)
|
|
if not df.empty:
|
|
df.to_csv(fname)
|
|
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)
|