defaults: cleanup

Signed-off-by: Niv Sardi <[email protected]>

python/common/defaults.py

@@ -1,16 +1,26 @@
-DATA_PATH='./data'
+#!/usr/bin/env python
+import os
 
-LABELS_PATH=f'{DATA_PATH}/labels'
-IMAGES_PATH=f'{DATA_PATH}/images'
-CERTS_PATH=f'{DATA_PATH}/certs'
+def D(n, d):
+    return os.environ.get(f'SPOOF_{n}') or d
 
-SQUARES_DATA_PATH=f'{DATA_PATH}/squares'
-SQUARES_LABELS_PATH=f'{SQUARES_DATA_PATH}/labels'
-SQUARES_IMAGES_PATH=f'{SQUARES_DATA_PATH}/images'
+DATA_PATH = D('DATA_PATH', './data')
 
-DEBUG_PATH=f'{DATA_PATH}/debug'
-DEBUG_SQUARES_PATH=f'{DEBUG_PATH}/squares'
+LABELS_PATH = D('LABELS_PATH', f'{DATA_PATH}/labels')
+IMAGES_PATH = D('IMAGES_PATH', f'{DATA_PATH}/images')
+CERTS_PATH = D('CERTS_PATH', f'{DATA_PATH}/certs')
 
-LOGOS_DATA_PATH=f'{DATA_PATH}/logos'
+SQUARES_DATA_PATH = D('SQUARES_DATA_PATH', f'{DATA_PATH}/squares')
+SQUARES_LABELS_PATH = D('SQUARES_LABELS_PATH', f'{SQUARES_DATA_PATH}/labels')
+SQUARES_IMAGES_PATH = D('SQUARES_IMAGES_PATH', f'{SQUARES_DATA_PATH}/images')
 
-MAIN_CSV_PATH=f'{DATA_PATH}/entities.csv'
+DEBUG_PATH = D('DEBUG_PATH', f'{DATA_PATH}/debug')
+DEBUG_SQUARES_PATH = D('DEBUG_SQUARES_PATH', f'{DEBUG_PATH}/squares')
+
+LOGOS_DATA_PATH = D('LOGOS_DATA_PATH', f'{DATA_PATH}/logos')
+
+AUGMENTED_DATA_PATH = D('AUGMENTED_DATA_PATH', f'{DATA_PATH}/augmented')
+AUGMENTED_LABELS_PATH = D('AUGMENTED_LABELS_PATH', f'{AUGMENTED_DATA_PATH}/labels')
+AUGMENTED_IMAGES_PATH = D('AUGMENTED_IMAGES_PATH', f'{AUGMENTED_DATA_PATH}/images')
+
+MAIN_CSV_PATH = D('MAIN_CSV_PATH', f'{DATA_PATH}/entities.csv')