From 32c4471273097523d64ad0f7704861f211e78f72 Mon Sep 17 00:00:00 2001
From: ryuslash
Date: Tue, 9 Feb 2010 01:32:42 +0100
Subject: working on the command-line arguments here

---
 config.py | 26 ++++++++++++++++++--------
 1 file changed, 18 insertions(+), 8 deletions(-)

(limited to 'config.py')

diff --git a/config.py b/config.py
index 7110ce5..519a0b3 100644
--- a/config.py
+++ b/config.py
@@ -12,18 +12,12 @@ class _Configuration(object):
 
     def create_new(self):
         self.configparser.add_section("settings")
-        self.configparser.set("settings", "category", self.raw_input_with_default("w", "Please enter which category you would like to download from: "))
+        self.set_category(self.raw_input_with_default("w", "Please enter which category you would like to download from: "))
         
         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)
-        dirname = os.path.dirname(self.filename)
-        if not os.path.exists(dirname):
-            os.mkdir(dirname)
-        configfile = open(self.filename, "w")
-        self.configparser.write(configfile)
+        self.save()
 
     def raw_input_with_default(self, default, prompt):
         inp = raw_input("%s (default=%s): " % (prompt, default))
@@ -37,5 +31,21 @@ class _Configuration(object):
     def get_category(self):
         return self.configparser.get("settings", "category")
 
+    def set_category(self, value):
+        self.configparser.set("settings", "category", value)
+
+    def option_exists(self, option):
+        for section in self.configparser.sections():
+            if self.configparser.has_option(section, option):
+                return True
+        return False
+
+    def save(self):
+        dirname = os.path.dirname(self.filename)
+        if not os.path.exists(dirname):
+            os.mkdir(dirname)
+        configfile = open(self.filename, "w")
+        self.configparser.write(configfile)
+
 _configuration = _Configuration()
 def Configuration(): return _configuration
-- 
cgit v1.2.3-54-g00ecf