ryuslash/4grab
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fixed in windows

Browse source 
Windows was giving trouble with the os.getenv(HOME) which can be executed in the python shell and IDLE and will return C:\Documents and Settings\username\, but when running a script it will return None.
If os.getenv(HOME) is None, then now the exec dir will be used instead
This commit is contained in:
unknown 2010-02-08 01:24:18 +01:00
parent 53c7802ae2
commit 7707d5d3c6
2 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
.gitignore vendored
View file

@ -1,2 +1,3 @@
*~ *~
*.pyc *.pyc
.*

9
config.py
View file

@ -1,9 +1,14 @@
import os import os
import ConfigParser import ConfigParser
import sys
homedir = os.getenv("HOME")
if homedir is None:
homedir = os.path.dirname(sys.argv[0])
class _Configuration(object): class _Configuration(object):
def __init__(self): def __init__(self):
self.filename = os.path.join(os.path.join(os.getenv("HOME"), ".4grab"), "config.cfg") self.filename = os.path.join(os.path.join(homedir, ".4grab"), "config.cfg")
self.configparser = ConfigParser.RawConfigParser() self.configparser = ConfigParser.RawConfigParser()
if not os.path.exists(self.filename): if not os.path.exists(self.filename):
self.create_new() self.create_new()
@ -15,7 +20,7 @@ class _Configuration(object):
self.configparser.set("settings", "category", self.raw_input_with_default("w", "Please enter which category you would like to download from: ")) self.configparser.set("settings", "category", self.raw_input_with_default("w", "Please enter which category you would like to download from: "))
self.configparser.add_section("locations") self.configparser.add_section("locations")
self.configparser.set("locations", "download", self.raw_input_with_default(os.path.join(os.getenv("HOME"), "Pictures"), "Please enter where you would like the downloads to go: ")) self.configparser.set("locations", "download", self.raw_input_with_default(os.path.join(homedir, "Pictures"), "Please enter where you would like the downloads to go: "))
#with open(self.filename) as configfile: #with open(self.filename) as configfile:
# self.configparser.write(configfile) # self.configparser.write(configfile)