summaryrefslogtreecommitdiff
path: root/megapixels/app/utils/path_utils.py
blob: ddbc28641525252abad0c0d91f7174cc8f58fb86 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import os
from os.path import join
import logging

from app.settings import app_cfg as cfg
from app.settings import types

  
# class properties
DATA_STORE = cfg.DATA_STORE
DIR_DATSETs = cfg.DIR_DATSET_NAS
DIR_DATASETS = join(DATA_STORE, 'datasets')
DIR_PEOPLE = 'datasets/people'

# -------------------------------------------------------------------------
# Metadata
# -------------------------------------------------------------------------

def data_store_dir(opt_data_store):
  return f'/data_store_{opt_data_store.name.lower()}'

def dataset_dir(opt_dataset):
  return f'{opt_dataset.name.lower()}'

def metadata_dir(opt_data_store, opt_dataset):
  return join(data_store_dir(opt_data_store), DIR_PEOPLE, dataset_dir(opt_dataset), 'metadata')

def metadata(opt_data_store, opt_dataset, opt_metadata):
  return join(metadata_dir(opt_data_store, opt_dataset), f'{opt_metadata.name.lower()}.csv')


# -------------------------------------------------------------------------
# Media
# -------------------------------------------------------------------------