from well_routing.classes import Graph


start_time = "01-Dec-2025 11:00"
stop_time = "03-Dec-2025 13:00"
chosen_time = "01-Dec-2025 12:00"
resolution_hourly = int(3600 / 3600)

print("=" * 60)
print("WELL ROUTING ANALYSIS")
print("=" * 60)
print("\nTime window configuration:")
print(f"  Start time: {start_time}")
print(f"  Stop time:  {stop_time}")
print(f"  Chosen time for analysis: {chosen_time}")
print(f"  Resolution: {resolution_hourly} hour(s)")

# === Create Asset Graph ===
print("\n" + "-" * 60)
print("Creating asset graph for ASGA...")
routing_graph = Graph.get_asset_graph("ASGA", start_time, stop_time)
# g = routing_graph.convert_to_networkx(timestamp=chosen_time)
routing_graph.plot(timestamp=chosen_time)
