save figures to directory

2026-01-27 18:35:09 +01:00
parent fd4cb4b596
commit 23136b5c2e
6 changed files with 277 additions and 49 deletions
--- a/utils.py
+++ b/utils.py
@@ -107,6 +107,11 @@ class JPMCSurvey:
        self.qsf_filepath = qsf_path
        self.qid_descr_map = self._extract_qid_descr_map()
        self.qsf:dict = self._load_qsf()
+        
+        # get export directory name for saving figures ie if data_path='data/exports/OneDrive_2026-01-21/...' should be 'figures/OneDrive_2026-01-21'
+        self.fig_save_dir = Path('figures') / self.data_filepath.parts[2]
+        if not self.fig_save_dir.exists():
+            self.fig_save_dir.mkdir(parents=True, exist_ok=True)
    
    
    def _extract_qid_descr_map(self) -> dict: