Mercurial > pylearn

comparison pylearn/datasets/config.py @ 655:14d22ca1c8b5

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
if PYLEARN_DATA_ROOT don't exist try DBPATH.
author Frederic Bastien <bastienf@iro.umontreal.ca>
date Fri, 13 Feb 2009 13:22:26 -0500
parents d3d8f5a17909
children f4729745bb58
comparison
equal deleted inserted replaced
654:2704c8688ced 655:14d22ca1c8b5
3 3
4 Especially, the locations of data files. 4 Especially, the locations of data files.
5 """ 5 """
6 6
7 import os, sys 7 import os, sys
8 def env_get(key, default): 8 def env_get(key, default, key2 = None):
9 if key2 and os.getenv(key) is None:
10 key=key2
9 if os.getenv(key) is None: 11 if os.getenv(key) is None:
10 print >> sys.stderr, "WARNING: Environment variable", key, 12 print >> sys.stderr, "WARNING: Environment variable", key,
11 print >> sys.stderr, "is not set. Using default of", default 13 print >> sys.stderr, "is not set. Using default of", default
12 return default if os.getenv(key) is None else os.getenv(key) 14 return default if os.getenv(key) is None else os.getenv(key)
13 15
14 def data_root(): 16 def data_root():
15 return env_get('PYLEARN_DATA_ROOT', os.getenv('HOME')+'/data') 17 return env_get('PYLEARN_DATA_ROOT', os.getenv('HOME')+'/data', 'DBPATH')
16 18