parent
fe345409bb
commit
fc5f0b14e0
@ -0,0 +1,33 @@
|
||||
import os
|
||||
|
||||
|
||||
def _get_user_home():
|
||||
return os.path.expanduser('~')
|
||||
|
||||
|
||||
def _get_paddlespcceh_home():
|
||||
if 'PPSPEECH_HOME' in os.environ:
|
||||
home_path = os.environ['PPSPEECH_HOME']
|
||||
if os.path.exists(home_path):
|
||||
if os.path.isdir(home_path):
|
||||
return home_path
|
||||
else:
|
||||
raise RuntimeError(
|
||||
'The environment variable PPSPEECH_HOME {} is not a directory.'.
|
||||
format(home_path))
|
||||
else:
|
||||
return home_path
|
||||
return os.path.join(_get_user_home(), '.paddlespeech')
|
||||
|
||||
|
||||
def _get_sub_home(directory):
|
||||
home = os.path.join(_get_paddlespcceh_home(), directory)
|
||||
if not os.path.exists(home):
|
||||
os.makedirs(home)
|
||||
return home
|
||||
|
||||
|
||||
PPSPEECH_HOME = _get_paddlespcceh_home()
|
||||
MODEL_HOME = _get_sub_home('models')
|
||||
CONF_HOME = _get_sub_home('conf')
|
||||
DATA_HOME = _get_sub_home('datasets')
|
Loading…
Reference in new issue