Add configuration and command-line option framework.

[software/pymin.git] / services / firewall / __init__.py

diff --git a/services/firewall/__init__.py b/services/firewall/__init__.py
index 2c8e091124646f381685ea936e78e73841463dd8..6c1cbe30aaf2793b1a83e0a1cfc00c919cdcdcd4 100644 (file)
--- a/services/firewall/__init__.py
+++ b/services/firewall/__init__.py
@@ -1,7 +1,17 @@
 # vim: set encoding=utf-8 et sw=4 sts=4 :
 
 # vim: set encoding=utf-8 et sw=4 sts=4 :
 

+from formencode import validators as V
+from pymin.config import Option

 from handler import FirewallHandler
 
 from handler import FirewallHandler
 

+def setup_service(options, config):
+    options.add_group('firewall', 'Firewall service', [
+        Option('pickle_dir', V.String, metavar='DIR',
+               help='store persistent data in DIR directory'),
+        Option('config_dir', V.String, metavar='DIR',
+               help='write config file in DIR directory'),
+    ])
+

 def get_service(config):
     return FirewallHandler(config.firewall.pickle_dir, config.firewall.config_dir)
 
 def get_service(config):
     return FirewallHandler(config.firewall.pickle_dir, config.firewall.config_dir)