From 40833582658d96aa5cf67c0dcba1bf45b837b716 Mon Sep 17 00:00:00 2001 From: Floke Date: Thu, 19 Jun 2025 18:24:41 +0000 Subject: [PATCH] bugfix --- brancheneinstufung.py | 21 ++++++++++++++++++--- 1 file changed, 18 insertions(+), 3 deletions(-) diff --git a/brancheneinstufung.py b/brancheneinstufung.py index cb27ab1a..6bd26a01 100644 --- a/brancheneinstufung.py +++ b/brancheneinstufung.py @@ -205,6 +205,8 @@ def load_target_schema(csv_filepath=Config.SCHEMA_FILE): def parse_arguments(): """Parst und validiert Kommandozeilen-Argumente.""" parser = argparse.ArgumentParser(description=f"Unternehmensbewertung {Config.VERSION}", formatter_class=argparse.RawTextHelpFormatter) + + # --- Modus-Definitionen --- mode_categories = { "Sequentielle Verarbeitung (Zeilenweise)": ["full_run"], "Re-Evaluate Markierte Zeilen (Spalte A='x')": ["reeval"], @@ -222,17 +224,30 @@ def parse_arguments(): for mode in modes: mode_help_text += f" - {mode}\n" parser.add_argument("--mode", type=str, help=mode_help_text) + + # --- Allgemeine Argumente --- parser.add_argument("--limit", type=int, help="Maximale Anzahl zu verarbeitender Zeilen.", default=None) parser.add_argument("--start_sheet_row", type=int, help="Startzeile im Sheet (1-basiert).", default=None) parser.add_argument("--end_sheet_row", type=int, help="Endzeile im Sheet (1-basiert).", default=None) + + # --- Argumente für spezifische Modi --- valid_single_row_steps = ['wiki', 'chat', 'web', 'ml_predict'] default_steps_arg = ','.join(valid_single_row_steps) parser.add_argument("--steps", type=str, help=f"Schritte im 'reeval'/'full_run' Modus. Möglich: {valid_single_row_steps}", default=default_steps_arg) + parser.add_argument("--min_umsatz", type=float, help="Mindestumsatz in MIO € für find_wiki_serp.", default=Config.PLAUSI_UMSATZ_MIN_SCHWELLE_FUER_MA_CHECK / 1_000_000) parser.add_argument("--min_employees", type=int, help="Mindestmitarbeiterzahl für find_wiki_serp.", default=500) - parser.add_argument("--model_out", type=str, default=Config.MODEL_FILE, help="Pfad für trainiertes Modell.") - parser.add_argument("--imputer_out", type=str, default=Config.IMPUTER_FILE, help="Pfad für Imputer.") - parser.add_argument("--patterns_out", type=str, default=Config.PATTERNS_FILE_JSON, help="Pfad für Feature-Patterns.") + + # +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ + # +++ KORREKTUR HIER: Greife auf Config-Klasse zu +++++++++++++++++++++++++ + # +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ + parser.add_argument("--model_out", type=str, default=Config.MODEL_FILE, help=f"Pfad für trainiertes Modell. Standard: {Config.MODEL_FILE}") + parser.add_argument("--imputer_out", type=str, default=Config.IMPUTER_FILE, help=f"Pfad für Imputer. Standard: {Config.IMPUTER_FILE}") + parser.add_argument("--patterns_out", type=str, default=Config.PATTERNS_FILE_JSON, help=f"Pfad für Feature-Patterns. Standard: {Config.PATTERNS_FILE_JSON}") + # +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ + # +++ ENDE KORREKTUR ++++++++++++++++++++++++++++++++++++++++++++++++++++++ + # +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ + return parser.parse_args() # --- Globale Spalten-Mapping (WICHTIG: MUSS ZU IHREM SHEET PASSEN!) ---