diff options
| -rwxr-xr-x | Demo/scripts/from.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/Demo/scripts/from.py b/Demo/scripts/from.py index 8e663f1..394af8d 100755 --- a/Demo/scripts/from.py +++ b/Demo/scripts/from.py @@ -4,15 +4,14 @@ # Extension to multiple mailboxes and other bells & whistles are left # as exercises for the reader. -import sys, posix +import sys, os # Open mailbox file. Exits with exception when this fails. try: - mailbox = posix.environ['MAIL'] -except RuntimeError: - sys.stderr.write \ - ('Please set environment variable MAIL to your mailbox\n') + mailbox = os.environ['MAIL'] +except (AttributeError, KeyError): + sys.stderr.write('No environment variable $MAIL\n') sys.exit(2) try: |