From 53c7802ae2ef25ebed4e130813ce95e2816c919b Mon Sep 17 00:00:00 2001 From: ryuslash Date: Tue, 26 Jan 2010 20:20:39 +0100 Subject: [PATCH] Configuration * A configuration file is made if one hasn't been found * Certain values (currently download category and location) are stored in config file. --- config.py | 9 +++++++-- download.py | 4 ++-- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/config.py b/config.py index 7810458..7110ce5 100644 --- a/config.py +++ b/config.py @@ -17,8 +17,13 @@ class _Configuration(object): 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: ")) - with open(self.filename) as configfile: - self.configparser.write(configfile) + #with open(self.filename) as configfile: + # self.configparser.write(configfile) + dirname = os.path.dirname(self.filename) + if not os.path.exists(dirname): + os.mkdir(dirname) + configfile = open(self.filename, "w") + self.configparser.write(configfile) def raw_input_with_default(self, default, prompt): inp = raw_input("%s (default=%s): " % (prompt, default)) diff --git a/download.py b/download.py index 28b316e..e448fa0 100644 --- a/download.py +++ b/download.py @@ -5,7 +5,7 @@ import htmlparser import progressbar import config -savedir = Configuration().get_download_location() +savedir = config.Configuration().get_download_location() def get_thread_links(baseurl): myparser = htmlparser.MyParser() @@ -89,7 +89,7 @@ def get_images(t = []): if __name__ == "__main__": # Get a file-like object for the 4chan.org w/imgboard - base_url = "http://boards.4chan.org/" + Configuration().get_category() + "/" + base_url = "http://boards.4chan.org/" + config.Configuration().get_category() + "/" # Get the hyperlinks. t = get_thread_links(base_url)