From de52de5059e02b0f419e8db657d92e51ade06fc0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pierre-Elliott=20B=C3=A9cue?= Date: Thu, 10 Sep 2015 16:22:13 +0200 Subject: [PATCH] =?UTF-8?q?On=20continue=20le=20m=C3=A9nage.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/forward.py | 25 ++++++++++++++++++++----- 1 file changed, 20 insertions(+), 5 deletions(-) diff --git a/utils/forward.py b/utils/forward.py index b7362597..575af1ba 100755 --- a/utils/forward.py +++ b/utils/forward.py @@ -5,20 +5,35 @@ # detraz@crans.org import argparse +import os def getforward(user): """Bonjour, j'aurais pu être une docstring""" + + homedir = os.path.expanduser("~%s" % (user,)) + if homedir == "~%s" % (user,): + raise IOError("Le home de %r n'existe probablement pas." % (user,)) + + fwd = os.path.join(homedir, ".forward") + try: - mailredirect = open('/home/%s/.forward' % (user,)).readlines()[0] + with open(fwd, 'r') as forwardfile: + mailredirect = forwardfile.readline().strip() except IOError: mailredirect = "" print mailredirect - return -def writeforward(user,txt): +def writeforward(user, txt): """Bonjour, j'aurais pu être une docstring""" - open('/home/%s/.forward' % (user,), 'w').write(txt) - return + + homedir = os.path.expanduser("~%s" % (user,)) + if homedir == "~%s" % (user,): + raise IOError("Le home de %r n'existe probablement pas." % (user,)) + + fwd = os.path.join(homedir, ".forward") + + with open(fwd, 'w') as forwardfile: + forwardfile.write(txt) if __name__=="__main__": parser = argparse.ArgumentParser(description="Script qui, éventuellement appelé avec sudo, permet d'écrire dans le .forward de quelqu'un.", add_help=False)