# vim: set encoding=utf-8 et sw=4 sts=4 :
-from mako.template import Template
-from mako.runtime import Context
-from subprocess import Popen, PIPE
-from os import path
-try:
- import cPickle as pickle
-except ImportError:
- import pickle
+from formencode import validators as V
+from pymin.config import Option
+from handler import ProxyHandler
-try:
- from seqtools import Sequence
-except ImportError:
- # NOP for testing
- class Sequence: pass
-try:
- from dispatcher import handler, HandlerError, Handler
-except ImportError:
- # NOP for testing
- class HandlerError(RuntimeError): pass
- class Handler: pass
- def handler(help):
- def wrapper(f):
- return f
- return wrapper
+def setup_service(options, config):
+ options.add_group('proxy', 'Proxy 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 files in DIR directory'),
+ ])
-__ALL__ = ('ProxyHandler')
+def get_service(config):
+ return ProxyHandler(config.proxy.pickle_dir, config.proxy.config_dir)
-pickle_ext = '.pkl'
-pickle_vars = 'vars'
-pickle_hosts= 'hosts'
-pickle_users = 'users'
-config_filename = 'squid.conf'
-
-template_dir = path.join(path.dirname(__file__), 'templates')
-
-
-class Error(HandlerError):
- r"""
- Error(command) -> Error instance :: Base DnsHandler exception class.
-
- All exceptions raised by the DnsHandler inherits from this one, so you can
- easily catch any DnsHandler exception.
-
- message - A descriptive error message.
- """
-
- def __init__(self, message):
- r"Initialize the Error object. See class documentation for more info."
- self.message = message
-
- def __str__(self):
- return self.message
-
-class HostError(Error, KeyError):
- r"""
- HostError(hostname) -> HostError instance
-
- This is the base exception for all host related errors.
- """
-
- def __init__(self, hostname):
- r"Initialize the object. See class documentation for more info."
- self.message = 'Host error: "%s"' % hostname
-
-class HostAlreadyExistsError(HostError):
- r"""
- HostAlreadyExistsError(hostname) -> HostAlreadyExistsError instance
-
- This exception is raised when trying to add a hostname that already exists.
- """
-
- def __init__(self, hostname):
- r"Initialize the object. See class documentation for more info."
- self.message = 'Host already exists: "%s"' % hostname
-
-class HostNotFoundError(HostError):
- r"""
- HostNotFoundError(hostname) -> HostNotFoundError instance
-
- This exception is raised when trying to operate on a hostname that doesn't
- exists.
- """
-
- def __init__(self, hostname):
- r"Initialize the object. See class documentation for more info."
- self.message = 'Host not found: "%s"' % hostname
-
-class ParameterError(Error, KeyError):
- r"""
- ParameterError(paramname) -> ParameterError instance
-
- This is the base exception for all DhcpHandler parameters related errors.
- """
-
- def __init__(self, paramname):
- r"Initialize the object. See class documentation for more info."
- self.message = 'Parameter error: "%s"' % paramname
-
-class ParameterNotFoundError(ParameterError):
- r"""
- ParameterNotFoundError(hostname) -> ParameterNotFoundError instance
-
- This exception is raised when trying to operate on a parameter that doesn't
- exists.
- """
-
- def __init__(self, paramname):
- r"Initialize the object. See class documentation for more info."
- self.message = 'Parameter not found: "%s"' % paramname
-
-class Host(Sequence):
-
- def __init__(self,ip):
- self.ip = ip
-
- def as_tuple(self):
- return (self.ip)
-
-class HostHandler(Handler):
-
- def __init__(self, hosts):
- self.hosts = hosts
-
- @handler(u'Adds a host')
- def add(self, ip):
- if ip in self.hosts:
- raise HostAlreadyExistsError(ip)
- self.hosts[ip] = Host(ip)
-
- @handler(u'Deletes a host')
- def delete(self, ip):
- if not ip in self.hosts:
- raise HostNotFoundError(ip)
- del self.hosts[ip]
-
- @handler(u'Shows all hosts')
- def list(self):
- return self.hosts.keys()
-
- @handler(u'Get information about all hosts')
- def show(self):
- return self.hosts.items()
-
-
-class ProxyHandler(Handler):
-
- def __init__(self, pickle_dir='.', config_dir='.'):
- self.pickle_dir = pickle_dir
- self.config_dir = config_dir
- f = path.join(template_dir, config_filename)
- self.config_template = Template(filename=f)
- try:
- self._load()
- except IOError:
- self.hosts = dict()
- self.vars = dict(
- ip = '192.168.0.1',
- port = '8080',
- )
- self.host = HostHandler(self.hosts)
-
- @handler(u'Set a Proxy parameter')
- def set(self, param, value):
- r"set(param, value) -> None :: Set a Proxy parameter."
- if not param in self.vars:
- raise ParameterNotFoundError(param)
- self.vars[param] = value
-
- @handler(u'Get a DNS parameter')
- def get(self, param):
- r"get(param) -> None :: Get a Proxy parameter."
- if not param in self.vars:
- raise ParameterNotFoundError(param)
- return self.vars[param]
-
- @handler(u'List Proxy parameters')
- def list(self):
- return self.vars.keys()
-
- @handler(u'Get all Proxy parameters, with their values.')
- def show(self):
- return self.vars.values()
-
- @handler(u'Start the service.')
- def start(self):
- r"start() -> None :: Start the DNS service."
- #esto seria para poner en una interfaz
- #y seria el hook para arrancar el servicio
- pass
-
- @handler(u'Stop the service.')
- def stop(self):
- r"stop() -> None :: Stop the DNS service."
- #esto seria para poner en una interfaz
- #y seria el hook para arrancar el servicio
- pass
-
- @handler(u'Restart the service.')
- def restart(self):
- r"restart() -> None :: Restart the DNS service."
- #esto seria para poner en una interfaz
- #y seria el hook para arrancar el servicio
- pass
-
- @handler(u'Reload the service config (without restarting, if possible)')
- def reload(self):
- r"reload() -> None :: Reload the configuration of the DNS service."
- #esto seria para poner en una interfaz
- #y seria el hook para arrancar el servicio
- print('reloading configuration')
-
- @handler(u'Commit the changes (reloading the service, if necessary).')
- def commit(self):
- r"commit() -> None :: Commit the changes and reload the DNS service."
- #esto seria para poner en una interfaz
- #y seria que hace el pickle deberia llamarse
- #al hacerse un commit
- self._dump()
- self._write_config()
- self.reload()
-
- @handler(u'Discard all the uncommited changes.')
- def rollback(self):
- r"rollback() -> None :: Discard the changes not yet commited."
- self._load()
-
- def _dump(self):
- r"_dump() -> None :: Dump all persistent data to pickle files."
- # XXX podría ir en una clase base
- self._dump_var(self.vars, pickle_vars)
- self._dump_var(self.hosts, pickle_hosts)
-
- def _load(self):
- r"_load() -> None :: Load all persistent data from pickle files."
- # XXX podría ir en una clase base
- self.vars = self._load_var(pickle_vars)
- self.hosts = self._load_var(pickle_hosts)
-
- def _pickle_filename(self, name):
- r"_pickle_filename() -> string :: Construct a pickle filename."
- # XXX podría ir en una clase base
- return path.join(self.pickle_dir, name) + pickle_ext
-
- def _dump_var(self, var, name):
- r"_dump_var() -> None :: Dump a especific variable to a pickle file."
- # XXX podría ir en una clase base
- pkl_file = file(self._pickle_filename(name), 'wb')
- pickle.dump(var, pkl_file, 2)
- pkl_file.close()
-
- def _load_var(self, name):
- r"_load_var() -> object :: Load a especific pickle file."
- # XXX podría ir en una clase base
- return pickle.load(file(self._pickle_filename(name)))
-
- def _write_config(self):
- r"_write_config() -> None :: Generate all the configuration files."
- out_file = file(path.join(self.config_dir, config_filename), 'w')
- ctx = Context(out_file,
- ip = self.vars['ip'],
- port = self.vars['port'],
- hosts = self.hosts.values()
- )
- self.config_template.render_context(ctx)
- out_file.close()
-
-
-if __name__ == '__main__':
-
- px = ProxyHandler()
- px.set('ip','192.66.66.66')
- px.set('port','666')
- px.host.add('192.168.0.25.25')
- px.host.add('192.168.0.25.26')
- px.host.add('192.168.0.25.27')
- px.host.delete('192.168.0.25.27')
- px.commit()
\ No newline at end of file
projects / software / pymin.git / blobdiff