68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
import yaml
|
|
import streamlit as st
|
|
|
|
def get_seuil(seuils_dict, key):
|
|
try:
|
|
if key in seuils_dict:
|
|
data = seuils_dict[key]
|
|
for niveau in ["rouge", "orange", "vert"]:
|
|
if niveau in data:
|
|
seuil = data[niveau]
|
|
if "min" in seuil and seuil["min"] is not None:
|
|
return seuil["min"]
|
|
if "max" in seuil and seuil["max"] is not None:
|
|
return seuil["max"]
|
|
except:
|
|
pass
|
|
return None
|
|
|
|
def set_vulnerability(v1, v2, t1, t2, seuils):
|
|
v1_poids = 1
|
|
v1_couleur = "Vert"
|
|
if v1 > seuils[t1]["rouge"]["min"]:
|
|
v1_poids = 3
|
|
v1_couleur = "Rouge"
|
|
elif v1 > seuils[t1]["vert"]["max"]:
|
|
v1_poids = 2
|
|
v1_couleur = "Orange"
|
|
|
|
v2_poids = 1
|
|
v2_couleur = "Vert"
|
|
if v2 > seuils[t2]["rouge"]["min"]:
|
|
v2_poids = 3
|
|
v2_couleur = "Rouge"
|
|
elif v2 > seuils[t2]["vert"]["max"]:
|
|
v2_poids = 2
|
|
v2_couleur = "Orange"
|
|
|
|
poids = v1_poids * v2_poids
|
|
couleur = "Rouge"
|
|
if poids <= 2:
|
|
couleur = "Vert"
|
|
elif poids <= 4:
|
|
couleur = "Orange"
|
|
|
|
return poids, couleur, v1_couleur, v2_couleur
|
|
|
|
def colorer_couleurs(la_couleur):
|
|
t = la_couleur.lower()
|
|
if t == "rouge" or t == "difficile":
|
|
return f":red-badge[{la_couleur}]"
|
|
if t == "orange" or t == "modérée":
|
|
return f":orange-badge[{la_couleur}]"
|
|
if t == "vert" or t == "facile":
|
|
return f":green-badge[{la_couleur}]"
|
|
return la_couleur
|
|
|
|
def initialiser_seuils(config_path):
|
|
seuils = {}
|
|
|
|
try:
|
|
with open(config_path, "r", encoding="utf-8") as f:
|
|
config = yaml.safe_load(f)
|
|
seuils = config.get("seuils", seuils)
|
|
except FileNotFoundError:
|
|
st.warning(f"Fichier de configuration {config_path} non trouvé.")
|
|
|
|
return seuils
|