From 062114d4eacf75e6d82ca2c1741ab6110198ae80 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pierre-Elliott=20B=C3=A9cue?= Date: Thu, 10 Sep 2015 04:17:27 +0200 Subject: [PATCH] =?UTF-8?q?PEP8=20+=20arr=C3=AAter=20de=20jouer=20=C3=A0?= =?UTF-8?q?=20la=20roulette=20russe=20avec=20sys.argv?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/forward.py | 30 ++++++++++++++++++------------ 1 file changed, 18 insertions(+), 12 deletions(-) diff --git a/utils/forward.py b/utils/forward.py index 022831fd..e4c69e3b 100755 --- a/utils/forward.py +++ b/utils/forward.py @@ -4,27 +4,33 @@ # Executé avec un sudo # detraz@crans.org -import sys +import argparse def getforward(user): + """Bonjour, j'aurais pu être une docstring""" try: - mailredirect = open('/home/%s/.forward'%user).readlines()[0] + mailredirect = open('/home/%s/.forward' % (user,)).readlines()[0] except IOError: mailredirect = "Inconnu" print mailredirect return def writeforward(user,txt): - open('/home/%s/.forward'%user,'w').write(txt) + """Bonjour, j'aurais pu être une docstring""" + open('/home/%s/.forward' % (user,), 'w').write(txt) return if __name__=="__main__": - for arg in sys.argv: - if "--name" in arg: - name=arg.split("=")[1] - if "--mail" in arg: - mail=arg.split("=")[1] - if "--read" in sys.argv: - getforward(name) - elif "--write" in sys.argv: - writeforward(name,mail) + parser = argparse.ArgumentParser(description="Script qui, éventuellement appelé avec sudo, permet d'écrire dans le .forward de quelqu'un.", add_help=False) + meh = parser.add_mutually_exclusive_group() + parser.add_argument("-h", "--help", help="Affiche cette aide et quitte.", action="store_true") + parser.add_argument("-m", "--mail", help="Mail d'utilisateur", type=str, action="store") + parser.add_argument("-n", "--name", help="Nom d'utilisateur", type=str, action="store") + meh.add_argument("-r", "--read", help="Lire", action="store_true") + meh.add_argument("-w", "--write", help="Écrire", action="store_true") + + args = parser.parse_args() + if args.read: + getforward(args.name) + elif args.write: + writeforward(args.name, args.mail)