Use local cache dir for resource processor

whiletrue-industries · Jun 13, 2024 · 5d8d8e7 · 5d8d8e7
1 parent dffcbfe
commit 5d8d8e7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -10,4 +10,6 @@ COPY setup.py ./
 RUN pip install -e .
 COPY utils ./utils
 ENV PYTHONUNBUFFERED=1
+RUN mkdir /temp-cache/ && chmod 777 /temp-cache/
+ENV RESOURCE_PROCESSOR_CACHE_DIR=/temp-cache/
 ENTRYPOINT ["/bin/bash", "-c"]
diff --git a/odds/backend/processor/resource_processor.py b/odds/backend/processor/resource_processor.py
@@ -14,7 +14,7 @@
 from ...common.realtime_status import realtime_status as rts
 
 
-TMP_DIR = CACHE_DIR / 'resource-processor-temp'
+TMP_DIR = os.environ.get('RESOURCE_PROCESSOR_CACHE_DIR') or CACHE_DIR / 'resource-processor-temp'
 TMP_DIR.mkdir(exist_ok=True, parents=True)
 TMP_DIR = str(TMP_DIR)