This commit is contained in:
2025-06-19 09:25:07 +00:00
parent b28ce3fabd
commit 68bfd117b4

View File

@@ -109,49 +109,42 @@ PATTERNS_FILE_TXT = "technician_patterns.txt" # Alt (Optional beibehalten)
PATTERNS_FILE_JSON = "technician_patterns.json" # Neu (Empfohlen)
def load_branch_mapping(file_path='Branchen.csv'):
"""
Lädt das Mapping von Detail-Branche zu Branchen-Gruppe aus einer CSV-Datei.
Behandelt UTF-8-BOM-Kodierung und normalisiert die Spaltennamen und Keys.
"""
logger = logging.getLogger(__name__)
try:
# +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
# +++ KORREKTUR: encoding='utf-8-sig' verwenden, um BOM zu behandeln +++
# +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
# Wir lesen jetzt alle Spalten ein und benennen sie dann um, was robuster ist
# als sich auf exakte Header-Namen in `usecols` zu verlassen.
df_mapping = pd.read_csv(file_path, sep=';', encoding='utf-8-sig')
# Spaltennamen bereinigen (entfernt BOM und andere unsichtbare Zeichen)
df_mapping.columns = [str(col).strip() for col in df_mapping.columns]
# Erwartete Spaltennamen (jetzt ohne BOM)
expected_cols = ['Branch Group', 'Branch']
# Prüfen, ob die erwarteten Spalten nach der Bereinigung vorhanden sind
expected_cols = ['Branch Group', 'Branch']
if not all(col in df_mapping.columns for col in expected_cols):
logger.error(f"FEHLER: Die erwarteten Spalten {expected_cols} wurden in '{file_path}' nicht gefunden. "
f"Gefundene Spalten: {list(df_mapping.columns)}")
logger.error(f"FEHLER: Erwartete Spalten {expected_cols} in '{file_path}' nicht gefunden.")
return {}
# Dictionary erstellen: { 'branch-detail-lowercase': 'Branch-Group' }
# Verwende die neue Normalisierungsfunktion für die Keys
branch_map_dict = pd.Series(
df_mapping['Branch Group'].str.strip().values,
index=df_mapping['Branch'].str.strip().str.lower() # Keys werden kleingeschrieben und bereinigt
index=df_mapping['Branch'].apply(normalize_for_mapping) # HIER ANWENDEN
).to_dict()
logger.info(f"Branchen-Mapping aus '{file_path}' erfolgreich geladen ({len(branch_map_dict)} Einträge).")
logger.info(f"Branchen-Mapping aus '{file_path}' geladen ({len(branch_map_dict)} Einträge).")
return branch_map_dict
except FileNotFoundError:
logger.error(f"FEHLER: Branchen-Mapping-Datei '{file_path}' nicht gefunden.")
return {}
except Exception as e:
logger.error(f"FEHLER beim Laden oder Verarbeiten der Branchen-Mapping-Datei '{file_path}': {e}")
logger.debug(traceback.format_exc())
logger.error(f"FEHLER beim Laden der Branchen-Mapping-Datei '{file_path}': {e}")
return {}
def normalize_for_mapping(text):
"""
Normalisiert einen String aggressiv für Mapping-Zwecke.
Entfernt Leerzeichen, macht alles klein, ersetzt gängige Sonderzeichen.
"""
if not isinstance(text, str):
return ""
text = text.lower()
text = text.strip()
# Ersetze gängige Trennzeichen und Varianten durch nichts, um sie zu ignorieren
text = re.sub(r'[\s/,-]+', '', text)
return text
# In Config-Klasse oder global aufrufen:
BRANCH_MAPPING = load_branch_mapping()
@@ -9149,7 +9142,9 @@ class DataProcessor:
# --- Kategoriale Features vorbereiten (Branche) ---
branche_col_internal = "branche_ki" # Name der Spalte mit den Detail-Branchen
self.logger.info(f"Verarbeite kategoriales Feature '{branche_col_internal}' (aus Spalte AL) und mappe es zu 'Branchen_Gruppe'...")
self.logger.info(f"Verarbeite kategoriales Feature '{branche_col_internal}' und mappe es zu 'Branchen_Gruppe'...")
normalized_sheet_branches = df_filtered[branche_col_internal].apply(normalize_for_mapping)
if branche_col_internal not in df_filtered.columns:
self.logger.critical(f"FEHLER: Spalte '{branche_col_internal}' nicht im DataFrame gefunden.")
@@ -9170,7 +9165,7 @@ class DataProcessor:
# Erstelle eine normalisierte Version des Mapping-Dictionaries
mapping_lower = {k.lower(): v for k, v in BRANCH_MAPPING.items()}
# Wende dieses normalisierte Mapping auf die normalisierte Branchenspalte an
df_filtered.loc[:, 'Branchen_Gruppe'] = cleaned_branches.str.lower().map(mapping_lower).fillna('Sonstige')
df_filtered.loc[:, 'Branchen_Gruppe'] = normalized_sheet_branches.map(BRANCH_MAPPING).fillna('Sonstige')
self.logger.info("Mapping zu 'Branchen_Gruppe' durchgeführt.")