crans/scripts
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

PEP8 + arrêter de jouer à la roulette russe avec sys.argv

Browse source
This commit is contained in:
Pierre-Elliott Bécue 2015-09-10 04:17:27 +02:00
parent 1262b47fb3
commit 062114d4ea
1 changed files with 18 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

30
utils/forward.py
View file

@ -4,27 +4,33 @@
# Executé avec un sudo # Executé avec un sudo
# detraz@crans.org # detraz@crans.org
import sys import argparse
def getforward(user): def getforward(user):
"""Bonjour, j'aurais pu être une docstring"""
try: try:
mailredirect = open('/home/%s/.forward'%user).readlines()[0] mailredirect = open('/home/%s/.forward' % (user,)).readlines()[0]
except IOError: except IOError:
mailredirect = "Inconnu" mailredirect = "Inconnu"
print mailredirect print mailredirect
return return
def writeforward(user,txt): 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 return
if __name__=="__main__": if __name__=="__main__":
for arg in sys.argv: parser = argparse.ArgumentParser(description="Script qui, éventuellement appelé avec sudo, permet d'écrire dans le .forward de quelqu'un.", add_help=False)
if "--name" in arg: meh = parser.add_mutually_exclusive_group()
name=arg.split("=")[1] parser.add_argument("-h", "--help", help="Affiche cette aide et quitte.", action="store_true")
if "--mail" in arg: parser.add_argument("-m", "--mail", help="Mail d'utilisateur", type=str, action="store")
mail=arg.split("=")[1] parser.add_argument("-n", "--name", help="Nom d'utilisateur", type=str, action="store")
if "--read" in sys.argv: meh.add_argument("-r", "--read", help="Lire", action="store_true")
getforward(name) meh.add_argument("-w", "--write", help="Écrire", action="store_true")
elif "--write" in sys.argv:
writeforward(name,mail) args = parser.parse_args()
if args.read:
getforward(args.name)
elif args.write:
writeforward(args.name, args.mail)