fix: directory and plot bugs

COMP0016-IFRC-Team5 · Mar 22, 2023 · a26e8af · a26e8af
1 parent 47dc079
commit a26e8af
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/visualiser/_config.py b/visualiser/_config.py
@@ -4,7 +4,7 @@
 __all__ = ['set_data_folder', 'get_data_folder', 'get_selected_folder']
 
 __DATA_FOLDER: Directory | None = None
-__SELECTED_FOLDER = Folders.unsliced
+__SELECTED_FOLDER = Folders.sliced
 
 
 def set_data_folder(path):

diff --git a/visualiser/_main.py b/visualiser/_main.py
@@ -129,9 +129,8 @@ def __plot_one(metric, country, event, df, years_required):
 
     """
     rpc = ReturnPeriodCalculator(country, event, df, metric, years_required)
-    if _config.get_selected_folder() == Folders.sliced:
-        rpc.plot(sliced=True)
-    rpc.plot()
+    is_slice = _config.get_selected_folder() == Folders.sliced
+    rpc.plot(sliced=is_slice)
 
 
 def get_exceedance_table(countries, events, years_required: int = -1):

diff --git a/visualiser/_utils/_directory.py b/visualiser/_utils/_directory.py
@@ -64,6 +64,8 @@ def __scan_directory(self):
                     self.__directories.append(Directory(entry.path))
 
     def __update(self):
+        self.__files = []
+        self.__directories = []
         self.__scan_directory()
         self.__files.sort(key=lambda f: f.get_filename())
         self.__directories.sort(key=lambda d: d.get_dirname())