diff --git a/wifi/bornes.py b/wifi/bornes.py index ee3e202a..1e8f71ba 100755 --- a/wifi/bornes.py +++ b/wifi/bornes.py @@ -69,6 +69,9 @@ def main(): action='store_const', const='uname -v', dest='cmd', default='uname -v') parser.add_option('-u', '--uptime', help=u"renvoie l'uptime", action='store_const', const='uptime', dest='cmd') + parser.add_option('-c', '--custom', help=u'exécute une commande custom', + action='store', dest='cmd') + (options, args) = parser.parse_args() names = bornes() @@ -112,8 +115,9 @@ if __name__ == "__main__": usage = u"""usage: %prog [OPTION] Récupère la liste des bornes et récupère le résultat d'une commande sur toutes les bornes""" - if len(sys.argv[0].split('_', 1)) > 1: - plugin = sys.argv[0].split('_', 1)[0] + prog = os.path.basename(sys.argv[0]) + if len(prog.split('_', 1)) > 1: + plugin = prog.split('_', 1)[0] if plugin == 'uptime': config = """graph_title Uptime Bornes graph_args --base 1000 -l 0