fix: Trawlwatcher loading data from settings.data_folder

dataforgoodfr · Mar 5, 2024 · b55ee29 · b55ee29
1 parent 39a39b2
commit b55ee29
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/bloom/bloom/infra/repositories/repository_vessel.py b/bloom/bloom/infra/repositories/repository_vessel.py
@@ -23,7 +23,7 @@ def __init__(
         session_factory: Callable,
     ) -> Callable[..., AbstractContextManager]:
         self.session_factory = session_factory
-        self.vessels_path = Path.joinpath(Path.cwd(), "data/chalutiers_pelagiques.csv")
+        self.vessels_path = Path(settings.data_folder).joinpath("./chalutiers_pelagiques.csv")
 
     def load_vessel_metadata(self) -> list[Vessel]:
         with self.session_factory() as session:

diff --git a/bloom/pages/1_Vessel_Exploration.py b/bloom/pages/1_Vessel_Exploration.py
@@ -2,6 +2,7 @@
 import streamlit as st
 from dotenv import load_dotenv
 from streamlit_folium import st_folium
+from pathlib import Path
 
 from bloom.config import settings
 from bloom.infra.database.database_manager import Database
@@ -31,7 +32,7 @@ def load_trajectory(mmsi, mpa):
 
 # @st.cache_data  # 👈 Add the caching decorator
 def load_trawlers():
-    trawlers = pd.read_csv("data/chalutiers_pelagiques.csv", sep=";")
+    trawlers = pd.read_csv(Path(settings.data_folder).joinpath("./chalutiers_pelagiques.csv"), sep=";")
     return trawlers