import yaml
import numpy
import pandas as pd

def parse_wireviz_yml(file_path):
    # Load the WireViz YML file
    with open(file_path, 'r') as file:
        data = yaml.safe_load(file)

    wires = data['cables']
    connectors = data['connectors']
    connections = data['connections']
    wire_info = []
    for connection in connections:
        connectorwpin = str(connection[0])
        connector = connectorwpin[2:connectorwpin.find("':")]
        connectorpins = connectorwpin[connectorwpin.find(" [")+2:connectorwpin.rfind("]}")].strip(" ").split(",")
        connectorpinsadjusted = []
        for pins in connectorpins:
            if pins.count("-") > 0:
                pinrange = pins.strip("' ").split("-")
                for x in range(int(pinrange[0]),int(pinrange[1])+1):
                    connectorpinsadjusted.append(x)
            else:
                connectorpinsadjusted = connectorpins

        connectorwires = str(connection[1])
        bundle = connectorwires[2:connectorwires.find("':")]
        bundlewires = connectorwires[connectorwires.find(" [")+2:connectorwires.rfind("]}")].split(",")
        if len(connection) > 2:
            connector2wpin = str(connection[2])
            connector2 = connector2wpin[2:connector2wpin.find("':")]
            connector2pins = connector2wpin[connector2wpin.find(" [")+2:connector2wpin.rfind("]}")].split(",")
            print(connector2pins)
        connector2pinsadjusted = []
        for pins in connector2pins:
            if pins.count("-") > 0:
                pinrange = pins.strip("' ").split("-")
                for x in range(int(pinrange[0]),int(pinrange[1])+1):
                    connector2pinsadjusted.append(x)
            else:
                connector2pinsadjusted = connector2pins
            print(connector2pinsadjusted)
            for x in range(0,len(connectorpinsadjusted)):
                wire_info.append({
                    'End 1': connector,
                    'Pin 1': connectorpinsadjusted[x],
                    'Length': wires[bundle]['length'],
                    'Color': wires[bundle]['colors'][x],
                    'Gauge': wires[bundle]['gauge'],
                    'End 2': connector2,
                    'Pin 2': connector2pinsadjusted[x],
                })




        #print(connection[1])


    for wire in wires:
        wire_name = wire
        length = wires[wire]['length']


    # Create a DataFrame to display the data as a table
    wire_df = pd.DataFrame(wire_info, columns=['End 1', 'Pin 1', 'Length', 'Color', 'Gauge', 'End 2', 'Pin 2'])
    return wire_df

# Path to your WireViz YML file
file_path = 'wireviz/Choppy_IB_PPS_main.yml'
wire_table = parse_wireviz_yml(file_path)

# Display the table
print(wire_table)