Fixed in windows

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

1
.gitignore vendored
View file

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

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)