etm-terrain/utils/produit_detector.py

34 lines
962 B
Python

PRODUITS = {
"fronius": "Onduleur Fronius",
"symo": "Onduleur Fronius Symo",
"primo": "Onduleur Fronius Primo",
"sma": "Onduleur SMA",
"huawei": "Onduleur Huawei",
"enphase": "Micro-onduleur Enphase",
"byd": "Batterie BYD",
"pylontech": "Batterie Pylontech",
"soltaro": "Batterie Soltaro",
"daikin": "PAC Daikin Altherma",
"viessmann": "PAC Viessmann",
"atlantic": "PAC Atlantic",
"keba": "Borne IRVE Keba",
"powercharger": "ETM-PowerCharger",
"wallbox": "Borne IRVE Wallbox",
"hems": "Système HEMS",
"onduleur": "Onduleur",
"batterie": "Batterie",
"pac ": "Pompe à Chaleur",
"irve": "Borne IRVE",
"borne": "Borne IRVE",
"panneau": "Panneaux PV",
"optimiseur": "Optimiseur",
}
def detecter_produit(texte: str) -> str:
t = texte.lower()
for mot, produit in PRODUITS.items():
if mot in t:
return produit
return "Produit non identifié"