Changeset 1152

Timestamp:

11/28/07 01:17:25 (1 year ago)

Author:

alban

Message:

Add configuration file support via --config option. Fixes #1

Files:

• trunk/justrecord.py (modified) (4 diffs)

trunk/justrecord.py

@@ -11 +11 @@
-import time
-import wave
+import optparse
-from optparse import OptionParser
-import pwd, grp
-import syslog
-import subprocess
+import ConfigParser
-
-class Recorder:
-
-options
-options.channels
-int(options.rate
-options.basename
-options.time_limit
-options.basedir
-options.after_script
+config
+config.int('channels')
+config.int('rate'
+config['basename']
+config.int('time_limit')
+config['basedir']
+config['after_script']
-
-    self.output_file_index = 0
@@ -144 +146 @@
-class Daemon:
-
-options
-options
-options
-options
-options
-options
+config
+config
+config
+config
+config
+config
-    self.main = main
-
@@ -239 +241 @@
-      syslog.closelog()
-
+class Config:
+
+  def __init__(self, default_options):
+    self.options = {}
+    self.update(default_options)
+
+  def update(self, options):
+    if isinstance(options, optparse.Values):
+      for entry in options.__dict__.iteritems():
+        if entry[1] is not None:
+          self.options[entry[0]] = entry[1]
+    else:
+      self.options.update(options)
+
+    return self
+
+  def load_ini(self, inifile):
+    config_parser = ConfigParser.ConfigParser()
+    if config_parser.read([inifile]) is []:
+      return False
+
+    ini_options = {}
+    for config_section in config_parser.sections():
+      for config_entry in config_parser.items(config_section):
+        ini_options[config_entry[0]] = config_entry[1]
+    self.update(ini_options)
+    return True
+
+  def __getitem__(self, key):
+    return self.options[key]
+
+  def string(self, key):
+    return self.options[key]
+
+  def int(self, key):
+    return int(self.options[key])
+
+  def bool(self, key):
+    value = self.options[key]
+    if isinstance(value, basestring):
+      return value.lower() in [ 'true', 'on', '1' ]
+    return value is True
+
-if __name__ == "__main__":
+
+  config = Config({
+    'basename': 'record',
+    'channels': 2,
+    'after_script': None,
+    'rate': 44100,
+    'time_limit': 30 * 60,
+    'basedir': os.getcwd(),
+    'daemon': False,
+    'console': False,
+    'syslog': False,
+    'uid': os.getuid(),
+    'gid': os.getgid(),
+    'pidfile': None,
+    'logfile': None
+  })
+
-  parser = OptionParser()
- default='record',
+
-                    help="The output basename. The default is 'record'.")
-
-
-
+
+
+
+
+
-                    choices=["48000", "44100", "22000"],
-44100 Hertz"
- default=30 * 60,
+%d Hertz" % config['rate']
+
-                    help="The time length of recording file in seconds. The default time length is 1800 seconds (30 minutes).")
- default=os.getcwd(),
+
-                    help="The directory where the recording files are created. The default directory is the current one.")
- default=False,
+
-                    help="Starts as daemon. Disabled by default")
- default=False,
+
-                    help="Logs messages to the console even if a logfile is specified.")
- default=False,
+
-                    help="Logs messages to syslog.")
- default=os.getuid(),
+
-                    help="Change to this username/uid before starting the recorder in daemon mode. The default is the current user.")
- default=os.getgid(),
+
-                    help="Change to this group/gid before starting the recorder in daemon mode. The default is the current group.")
- default=None,
+
-                    help="The file used to store the pid in daemon mode.")
- default=None,
+
-                    help="The file used to log messages.")
- default=None,
+
-                    help="Specifies a script to be executed each time a recording file is finished.")
-
-
-
-
+
+
+
+
+
+
+
+
+
+
+
-
-  def log(message):
-    logger.log(message)
+
+  log("config: %s" % config.options)
-
-  def main():
@@ -280 +353 @@
-    logger.close()
-
-options.daemon
+config.bool('daemon')
-  if daemon:
-
-
-
-
-
+
-  else:
-    main()