This commit is contained in:
2025-06-19 18:24:41 +00:00
parent 3a95af017f
commit 4083358265

View File

@@ -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!) ---