Small fixes, still need to change the right config file

        name = '{}.desktop'.format(name)

        if os.path.exists(name):

            return name

    output = subprocess.run(['locate', name], stdout=subprocess.PIPE).stdout.decode()

    return output.replace("\n", "")

    output = subprocess.run(['find', '/usr/share', '-name', name], stdout=subprocess.PIPE).stdout.decode()

    return output.split("\n")[0]

def parse_desktop(desktop):

    out = {}

    with open(desktop) as f:

        text = "\n".join(f.readlines())

        p = re.compile(r'((Exec|Icon|Name\[fr\]|GenericName\[fr\]).*)')

        p = re.compile(r'((Exec|Icon|Name=|Name\[fr\]|GenericName\[fr\]).*)')

        matches = p.findall(text)

        return {k: m.split("=")[1] for m, k in matches}

def add_to_handymenu(user, cmd, category):

    path = '/home/{}/.config/handymenu/conf.default.yml'.format(user)

    # TODO : change path

    path = '/home/{}/.config/handymenu/conf.default.yaml'.format(user)

    subprocess.run(['sed', '-i', '-e', 's/!!python\/str//g', path])

    print(path)

    with open(path) as stream:

        try:

            yml = yaml.safe_load(stream)

    desktop = find_desktop(args.app[0])

    values = parse_desktop(desktop)

    cmd = {

        'cmd': values['Exec'],

        'generic': values['GenericName[fr]'],

        'name': values['Name[fr]'],

        'icon': values['Icon'],

    links = {

            'Exec' : 'cmd',

            'GenericName': 'generic',

            'GenericName[fr]': 'generic',

            'Name[fr]': 'name',

            'Name': 'name',

            'Icon': 'icon',

            }

    cmd = {}

    for k,v in links.items():

        try:

            cmd[v] = values[k]

        except KeyError:

            pass

    for user in args.users:

        add_to_handymenu(user, cmd, args.category[0])