import subprocess

def run(cmd):
    ret = subprocess.run(["powershell.exe", cmd], stdout=subprocess.PIPE)
    s = ret.stdout
    return s.decode("utf-8")


def get_busid():
    info = run(r"usbipd list")
    begin = info.find("STM32 STLink") 
    busid = info[begin - 18: begin - 15]
    return busid


def attach_stlink():
    busid = get_busid()
    cmd = "usbipd attach --wsl --busid " + busid
    run(cmd)


def detach_stlink():
    busid = get_busid()
    cmd = "usbipd detach --busid " + busid
    run(cmd)


if __name__ == "__main__":
    # detach_stlink()
    attach_stlink()