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
# 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)