user.py 1.49 KB
Newer Older
1 2 3 4
"""Hook to allow user-specified customization code to run.

As a policy, Python doesn't run user-specified code on startup of
Python programs (interactive sessions execute the script specified in
5
the PYTHONSTARTUP environment variable if it exists).
6 7 8 9

However, some programs or sites may find it convenient to allow users
to have a standard customization file, which gets run when a program
requests it.  This module implements such a mechanism.  A program
Guido van Rossum's avatar
Guido van Rossum committed
10
that wishes to use the mechanism must execute the statement
11 12 13 14 15 16 17 18 19 20 21 22 23 24 25

    import user

The user module looks for a file .pythonrc.py in the user's home
directory and if it can be opened, execfile()s it in its own global
namespace.  Errors during this phase are not caught; that's up to the
program that imports the user module, if it wishes.

The user's .pythonrc.py could conceivably test for sys.version if it
wishes to do different things depending on the Python version.

"""

import os

26
home = os.curdir                        # Default
27
if os.environ.has_key('HOME'):
28
    home = os.environ['HOME']
Raymond Hettinger's avatar
Raymond Hettinger committed
29 30
elif os.name == 'posix':
    home = os.path.expanduser("~/")
31
elif os.name == 'nt':                   # Contributed by Jeff Bauer
32
    if os.environ.has_key('HOMEPATH'):
33 34 35 36
        if os.environ.has_key('HOMEDRIVE'):
            home = os.environ['HOMEDRIVE'] + os.environ['HOMEPATH']
        else:
            home = os.environ['HOMEPATH']
37 38 39 40 41 42 43 44 45

pythonrc = os.path.join(home, ".pythonrc.py")
try:
    f = open(pythonrc)
except IOError:
    pass
else:
    f.close()
    execfile(pythonrc)