From 16584b7fab23c01c2c48eda9efad513dc4addcec Mon Sep 17 00:00:00 2001
From: "Ira W. Snyder" <devel@irasnyder.com>
Date: Tue, 26 Dec 2006 14:43:55 -0800
Subject: [PATCH] [CONFIG] Add RarslaveConfig class
Add the RarslaveConfig class, which is almost entirely imported from the
original rarslave program.
Signed-off-by: Ira W. Snyder <devel@irasnyder.com>
---
RarslaveConfig.py | 129 ++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 129 insertions(+)
create mode 100644 RarslaveConfig.py
diff --git a/RarslaveConfig.py b/RarslaveConfig.py
new file mode 100644
index 0000000..a3defe5
--- /dev/null
+++ b/RarslaveConfig.py
@@ -0,0 +1,129 @@
+#!/usr/bin/env python
+# vim: set ts=4 sts=4 sw=4 textwidth=92:
+
+import os, ConfigParser
+
+class RarslaveConfig (object):
+ """A simple class to hold the default configs for the whole program"""
+
+ DEFAULT_CONFIG='~/.config/rarslave2/rarslave2.conf'
+
+ def __read_config(self, filename=DEFAULT_CONFIG):
+ """Attempt to open and read the rarslave config file"""
+
+ # Make sure the filename is corrected
+ filename = os.path.abspath(os.path.expanduser(filename))
+
+ user_config = {}
+
+ # Write the default config if it doesn't exist
+ if not os.path.isfile(filename):
+ self.write_config(default=True)
+
+ config = ConfigParser.ConfigParser()
+ config.read(filename)
+
+ for section in config.sections():
+ for option in config.options(section):
+ user_config[(section, option)] = config.get(section, option)
+
+ return user_config
+
+ def write_config(self, filename=DEFAULT_CONFIG, default=False):
+ """Write out the current config to the config file. If you set default=True, then
+ the default config file will be written."""
+
+ config = ConfigParser.ConfigParser()
+
+ # Correct filename
+ filename = os.path.abspath(os.path.expanduser(filename))
+
+ # Reset all config to make sure we write the default one, if necessary
+ if default:
+ self.__user_config = {}
+ print 'Writing default config to %s' % (filename, )
+
+ # [directories] section
+ config.add_section('directories')
+ for (s, k) in self.__defaults.keys():
+ if s == 'directories':
+ config.set(s, k, self.get_value(s, k))
+
+ # [options] section
+ config.add_section('options')
+ for (s, k) in self.__defaults.keys():
+ if s == 'options':
+ config.set(s, k, self.get_value(s, k))
+
+ # [regular_expressions] section
+ config.add_section('regular expressions')
+ for (s, k) in self.__defaults.keys():
+ if s == 'regular expressions':
+ config.set(s, k, self.get_value(s, k))
+
+ # Try to make the ~/.config/rarslave/ directory
+ if not os.path.isdir(os.path.split(filename)[0]):
+ try:
+ os.makedirs(os.path.split(filename)[0])
+ except:
+ print 'Could not make directory: %s' % (os.path.split(filename)[0], )
+ sys.exit()
+
+ # Try to write the config file to disk
+ try:
+ fsock = open(filename, 'w')
+ try:
+ config.write(fsock)
+ finally:
+ fsock.close()
+ except:
+ print 'Could not open: %s for writing' % (filename, )
+ sys.exit()
+
+ def __get_default_val(self, section, key):
+ return self.__defaults[(section, key)]
+
+ def get_value(self, section, key):
+ """Get a config value. Attempts to get the value from the user's
+ config first, and then uses the default."""
+
+ try:
+ value = self.__user_config[(section, key)]
+ except:
+ # This should work, unless you write something stupid
+ # into the code, so DON'T DO IT
+ value = self.__get_default_val(section, key)
+
+ # Convert config options to booleans for easier use
+ if value == 'True':
+ value = True
+
+ if value == 'False':
+ value = False
+
+ return value
+
+ def __init__(self):
+ self.__defaults = {
+ ('directories', 'working_directory') : '~/downloads/usenet',
+ ('options', 'recursive') : True,
+ ('options', 'check_required_programs') : False,
+ ('options', 'extract_with_full_path') : False,
+ ('options', 'interactive') : False,
+ ('options', 'output_loglevel') : 1,
+ ('regular expressions', 'par2_regex') : '.*\.par2$',
+ ('regular expressions', 'video_file_regex') : '.*\.(avi|ogm|mkv|mp4)$',
+ ('regular expressions', 'temp_repair_regex') : '.*\.1$',
+ ('regular expressions', 'remove_regex') : '^.*\.(rar|r\d\d)$' }
+
+ self.__user_config = self.__read_config()
+
+
+
+
+def main ():
+ pass
+
+if __name__ == '__main__':
+ main ()
+
--
2.34.1