80 lines
3.4 KiB
Python
80 lines
3.4 KiB
Python
import json
|
|
import time
|
|
import os
|
|
from company_explorer_connector import handle_company_workflow
|
|
|
|
def run_trading_twins_process(target_company_name: str):
|
|
"""
|
|
Startet den Trading Twins Prozess für ein Zielunternehmen.
|
|
Ruft den Company Explorer Workflow auf, um das Unternehmen zu finden,
|
|
zu erstellen oder anzureichern.
|
|
"""
|
|
print(f"\n{'='*50}")
|
|
print(f"Starte Trading Twins Analyse für: {target_company_name}")
|
|
print(f"{'='*50}\n")
|
|
|
|
# Aufruf des Company Explorer Workflows
|
|
# Diese Funktion prüft, ob die Firma existiert.
|
|
# Wenn nicht, erstellt sie die Firma und startet die Anreicherung.
|
|
# Sie gibt am Ende die Daten aus dem Company Explorer zurück.
|
|
company_data_result = handle_company_workflow(target_company_name)
|
|
|
|
# Verarbeitung der Rückgabe (für den POC genügt eine Ausgabe)
|
|
print("\n--- Ergebnis vom Company Explorer Connector (für Trading Twins) ---")
|
|
|
|
status = company_data_result.get("status")
|
|
data = company_data_result.get("data")
|
|
|
|
if status == "error":
|
|
print(f"Ein Fehler ist aufgetreten: {company_data_result.get('message')}")
|
|
elif status == "found":
|
|
print(f"Unternehmen gefunden. ID: {data.get('id')}, Name: {data.get('name')}")
|
|
print(json.dumps(data, indent=2, ensure_ascii=False))
|
|
elif status == "created_and_enriched":
|
|
print(f"Unternehmen erstellt und Enrichment angestoßen. ID: {data.get('id')}, Name: {data.get('name')}")
|
|
print("Hinweis: Enrichment-Prozesse laufen im Hintergrund und können einige Zeit dauern, bis alle Daten verfügbar sind.")
|
|
print(json.dumps(data, indent=2, ensure_ascii=False))
|
|
elif status == "created_discovery_timeout":
|
|
print(f"Unternehmen erstellt, aber Discovery konnte keine Website finden (ID: {data.get('id')}, Name: {data.get('name')}).")
|
|
print("Der Analyse-Prozess wurde daher nicht gestartet.")
|
|
print(json.dumps(data, indent=2, ensure_ascii=False))
|
|
else:
|
|
print("Ein unerwarteter Status ist aufgetreten.")
|
|
print(json.dumps(company_data_result, indent=2, ensure_ascii=False))
|
|
|
|
print(f"\n{'='*50}")
|
|
print(f"Trading Twins Analyse für {target_company_name} abgeschlossen.")
|
|
print(f"{'='*50}\n")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Simulieren der Umgebungsvariablen für diesen Testlauf, falls nicht gesetzt
|
|
if "COMPANY_EXPLORER_API_USER" not in os.environ:
|
|
os.environ["COMPANY_EXPLORER_API_USER"] = "admin"
|
|
if "COMPANY_EXPLORER_API_PASSWORD" not in os.environ:
|
|
os.environ["COMPANY_EXPLORER_API_PASSWORD"] = "gemini"
|
|
|
|
# Testfall 1: Ein Unternehmen, das wahrscheinlich bereits existiert
|
|
# Da 'Robo-Planet GmbH' bei den vorherigen Läufen erstellt wurde, sollte es jetzt gefunden werden.
|
|
run_trading_twins_process("Robo-Planet GmbH")
|
|
|
|
# Kurze Pause zwischen den Testläufen
|
|
time.sleep(5)
|
|
|
|
# Testfall 1b: Ein bekanntes, real existierendes Unternehmen
|
|
run_trading_twins_process("Klinikum Landkreis Erding")
|
|
|
|
# Kurze Pause zwischen den Testläufen
|
|
time.sleep(5)
|
|
|
|
# Testfall 2: Ein neues, eindeutiges Unternehmen
|
|
new_unique_company_name = f"Trading Twins New Target {int(time.time())}"
|
|
run_trading_twins_process(new_unique_company_name)
|
|
|
|
# Kurze Pause
|
|
time.sleep(5)
|
|
|
|
# Testfall 3: Ein weiteres neues Unternehmen, um die Erstellung zu prüfen
|
|
another_new_company_name = f"Another Demo Corp {int(time.time())}"
|
|
run_trading_twins_process(another_new_company_name)
|