feat cache

.gitignore

@@ -1 +1,2 @@
-.venv/
+.venv/
+__pycache__/

src/ingest/load_data.py

@@ -0,0 +1,10 @@
+import streamlit as st
+import polars as pl
+import requests
+
+GRIST_URL = "https://grist.numerique.gouv.fr/o/dim-chu-brest/api/docs/5ZXqTKBiHTD2qqUj6GVPtp"
+
+@st.cache_data
+def load_grist(table: str)->pl.DataFrame:
+    records = requests.get(f"{GRIST_URL}/tables/{table}/records").json()["records"]
+    return pl.DataFrame(records).drop("id").unnest("fields")

src/page/acte.py

@@ -1,15 +1,10 @@
-import requests
-import unidecode
-import polars as pl
 import streamlit as st
+import polars as pl
+import unidecode
+from ingest.load_data import load_grist
 
 
-GRIST_URL = "https://grist.numerique.gouv.fr/o/dim-chu-brest/api/docs/5ZXqTKBiHTD2qqUj6GVPtp"   # ou ton serveur
-TABLE     = "Actes_classants"
-
-records = requests.get(f"{GRIST_URL}/tables/{TABLE}/records").json()["records"]
-
-acte = pl.DataFrame(records).drop("id").unnest("fields")
+data = load_grist("Actes_classants")
 
 names = {
     "CODE_ACTE": "Code acte",
@@ -19,7 +14,7 @@ names = {
 }
 
 items = (
-    acte.filter(pl.col("LIBELLE_CARACTERISTIQUE").is_not_null())["LIBELLE_CARACTERISTIQUE"]
+    data.filter(pl.col("LIBELLE_CARACTERISTIQUE").is_not_null())["LIBELLE_CARACTERISTIQUE"]
     .unique()
     .str.to_titlecase()
     .sort()
@@ -37,7 +32,7 @@ exp = unidecode.unidecode(exp)
 selection = col3.multiselect("Filtre(s) :", options=items, default=[])
 
 _temp = (
-    acte.filter(
+    data.filter(
         (pl.col("CODE_ACTE").str.contains(r"(?i)" + exp))
         | (pl.col("LIBELLE_ACTE").str.contains(r"(?i)" + exp))
         | (pl.col("LIBELLE_CARACTERISTIQUE").str.contains(r"(?i)" + exp))
@@ -48,7 +43,7 @@ _temp = (
 )
 
 df = (
-    acte.join(_temp, on="CODE_ACTE", how="inner")
+    data.join(_temp, on="CODE_ACTE", how="inner")
     .with_columns(
         pl.lit(1).alias("flag"),
     )
@@ -62,7 +57,7 @@ df = (
 )
 
 if selection:
-    _temp = acte.filter((pl.col("LIBELLE_CARACTERISTIQUE").is_in(selection))).select("CODE_ACTE").unique()
+    _temp = data.filter((pl.col("LIBELLE_CARACTERISTIQUE").is_in(selection))).select("CODE_ACTE").unique()
     df = df.join(_temp, on="CODE_ACTE", how="inner").rename(names)
 
 st.dataframe(df.to_pandas(), hide_index=True, width='stretch')

src/page/bacteries.py

@@ -1,15 +1,11 @@
-import requests
-import unidecode
+from json import load
 import polars as pl
 import streamlit as st
+import unidecode
+from ingest.load_data import load_grist
 
 
-GRIST_URL = "https://grist.numerique.gouv.fr/o/dim-chu-brest/api/docs/5ZXqTKBiHTD2qqUj6GVPtp"   # ou ton serveur
-TABLE     = "Bacteries"
-
-records = requests.get(f"{GRIST_URL}/tables/{TABLE}/records").json()["records"]
-
-data = pl.DataFrame(records).drop("id").unnest("fields").select("Bacterie", "Code_bacterie", "Code_si_sepsis").sort("Bacterie")
+data = load_grist("Bacteries").select("Bacterie", "Code_bacterie", "Code_si_sepsis").sort("Bacterie")
 
 st.title("Bactéries CIM-10")
 

src/page/ctxt.py

@@ -1,18 +1,11 @@
-import requests
-import unidecode
-import polars as pl
 import streamlit as st
+from ingest.load_data import load_grist
 
 
-GRIST_URL = "https://grist.numerique.gouv.fr/o/dim-chu-brest/api/docs/5ZXqTKBiHTD2qqUj6GVPtp"   # ou ton serveur
-TABLE     = "Ctxt"
-
-records = requests.get(f"{GRIST_URL}/tables/{TABLE}/records").json()["records"]
-
-ctxt = pl.DataFrame(records).drop("id").unnest("fields").sort(pl.col("Code_CIM_10"))
+data = load_grist("Ctxt").sort("Code_CIM_10")
 
 st.title("Contexte patient")
 
-st.dataframe(ctxt.to_pandas(), hide_index=True, width='stretch')
+st.dataframe(data.to_pandas(), hide_index=True, width='stretch')
 
 st.divider()

src/page/intox.py

@@ -1,18 +1,11 @@
-import requests
-import unidecode
-import polars as pl
 import streamlit as st
+import polars as pl
+import unidecode
+from ingest.load_data import load_grist
 
 
-GRIST_URL = "https://grist.numerique.gouv.fr/o/dim-chu-brest/api/docs/5ZXqTKBiHTD2qqUj6GVPtp"   # ou ton serveur
-TABLE1 = "Intox_data"
-TABLE2 = "Intox_data2"
-
-records1 = requests.get(f"{GRIST_URL}/tables/{TABLE1}/records").json()["records"]
-records2 = requests.get(f"{GRIST_URL}/tables/{TABLE2}/records").json()["records"]
-
-cim10 = pl.DataFrame(records1).drop("id").unnest("fields").sort("SUBSTANCE")
-data = pl.DataFrame(records2).drop("id").unnest("fields").sort("Nom")
+cim10 = load_grist("Intox_data").sort("SUBSTANCE")
+data = load_grist("Intox_data2").sort("Nom")
 
 data = data.with_columns(
     pl.col("Nom")