Change get_default_download_dir to use expanduser as fallback

This commit is contained in:
Calum Lind 2014-02-03 12:26:57 +00:00
parent dc0000059b
commit bddfb2a5c6
1 changed files with 9 additions and 9 deletions

View File

@ -165,21 +165,21 @@ def get_default_download_dir():
:rtype: string
"""
if windows_check():
return os.path.join(os.path.expanduser("~"), 'Downloads')
else:
download_dir = ""
if not windows_check():
from xdg.BaseDirectory import xdg_config_home
userdir_file = os.path.join(xdg_config_home, 'user-dirs.dirs')
try:
for line in open(userdir_file, 'r'):
if not line.startswith('#') and 'XDG_DOWNLOAD_DIR' in line:
with open(os.path.join(xdg_config_home, 'user-dirs.dirs'), 'r') as _file:
for line in _file:
if not line.startswith('#') and line.startswith('XDG_DOWNLOAD_DIR'):
download_dir = os.path.expandvars(line.partition("=")[2].rstrip().strip('"'))
if os.path.isdir(download_dir):
return download_dir
break
except IOError:
pass
return os.environ.get("HOME")
if not download_dir:
download_dir = os.path.join(os.path.expanduser("~"), 'Downloads')
return download_dir
def windows_check():