pymin/services/proxy/__init__.py

   1 # vim: set encoding=utf-8 et sw=4 sts=4 :
   2
   3 from os import path
   4
   5 from pymin.seqtools import Sequence
   6 from pymin.dispatcher import Handler, handler, HandlerError
   7 from pymin.services.util import Restorable, ConfigWriter, InitdHandler, \
   8                                 TransactionalHandler, ParametersHandler
   9
  10 import crypt
  11
  12 __ALL__ = ('ProxyHandler', 'Error', 'HostError', 'HostAlreadyExistsError',
  13             'HostNotFoundError')
  14
  15 class Error(HandlerError):
  16     r"""
  17     Error(command) -> Error instance :: Base DnsHandler exception class.
  18
  19     All exceptions raised by the DnsHandler inherits from this one, so you can
  20     easily catch any DnsHandler exception.
  21
  22     message - A descriptive error message.
  23     """
  24
  25     def __init__(self, message):
  26         r"Initialize the Error object. See class documentation for more info."
  27         self.message = message
  28
  29     def __str__(self):
  30         return self.message
  31
  32 class HostError(Error, KeyError):
  33     r"""
  34     HostError(hostname) -> HostError instance
  35
  36     This is the base exception for all host related errors.
  37     """
  38
  39     def __init__(self, hostname):
  40         r"Initialize the object. See class documentation for more info."
  41         self.message = 'Host error: "%s"' % hostname
  42
  43 class HostAlreadyExistsError(HostError):
  44     r"""
  45     HostAlreadyExistsError(hostname) -> HostAlreadyExistsError instance
  46
  47     This exception is raised when trying to add a hostname that already exists.
  48     """
  49
  50     def __init__(self, hostname):
  51         r"Initialize the object. See class documentation for more info."
  52         self.message = 'Host already exists: "%s"' % hostname
  53
  54 class HostNotFoundError(HostError):
  55     r"""
  56     HostNotFoundError(hostname) -> HostNotFoundError instance
  57
  58     This exception is raised when trying to operate on a hostname that doesn't
  59     exists.
  60     """
  61
  62     def __init__(self, hostname):
  63         r"Initialize the object. See class documentation for more info."
  64         self.message = 'Host not found: "%s"' % hostname
  65
  66
  67 class Host(Sequence):
  68
  69     def __init__(self,ip):
  70         self.ip = ip
  71
  72     def as_tuple(self):
  73         return (self.ip)
  74
  75 class HostHandler(Handler):
  76
  77     def __init__(self, hosts):
  78         self.hosts = hosts
  79
  80     @handler(u'Adds a host')
  81     def add(self, ip):
  82         if ip in self.hosts:
  83             raise HostAlreadyExistsError(ip)
  84         self.hosts[ip] = Host(ip)
  85
  86     @handler(u'Deletes a host')
  87     def delete(self, ip):
  88         if not ip in self.hosts:
  89             raise HostNotFoundError(ip)
  90         del self.hosts[ip]
  91
  92     @handler(u'Shows all hosts')
  93     def list(self):
  94         return self.hosts.keys()
  95
  96     @handler(u'Get information about all hosts')
  97     def show(self):
  98         return self.hosts.items()
  99
 100
 101 class UserHandler(Handler):
 102
 103     def __init__(self, users):
 104         self.users = users
 105
 106     def add(self, user, password):
 107         if user in self.users:
 108             raise UserAlreadyExistsError(user)
 109         self.users[user] = crypt.crypt(password,'BA')
 110
 111     def delete(self, user):
 112         if not user in self.users:
 113             raise UserNotFound(user)
 114         del self.users[user]
 115
 116 class ProxyHandler(Restorable, ConfigWriter, InitdHandler,
 117                    TransactionalHandler, ParametersHandler):
 118
 119     _initd_name = 'squid'
 120
 121     _persistent_attrs = ('params', 'hosts', 'users')
 122
 123     _restorable_defaults = dict(
 124             hosts = dict(),
 125             params  = dict(
 126                 ip   = '192.168.0.1',
 127                 port = '8080',
 128             ),
 129             users = dict(),
 130     )
 131
 132     _config_writer_files = ('squid.conf','users.conf')
 133     _config_writer_tpl_dir = path.join(path.dirname(__file__), 'templates')
 134
 135     def __init__(self, pickle_dir='.', config_dir='.'):
 136         r"Initialize DhcpHandler object, see class documentation for details."
 137         self._persistent_dir = pickle_dir
 138         self._config_writer_cfg_dir = config_dir
 139         self._config_build_templates()
 140         self._restore()
 141         self.host = HostHandler(self.hosts)
 142         self.user = UserHandler(self.users)
 143
 144     def _get_config_vars(self, config_file):
 145         if config_file == 'squid.conf':
 146             return dict(hosts=self.hosts.values(), **self.params)
 147         return dict(users=self.users)
 148
 149
 150 if __name__ == '__main__':
 151
 152     px = ProxyHandler()
 153     px.set('ip','192.66.66.66')
 154     px.set('port','666')
 155     px.host.add('192.168.0.25.25')
 156     px.host.add('192.168.0.25.26')
 157     px.host.add('192.168.0.25.27')
 158     px.host.delete('192.168.0.25.27')
 159     px.user.add('lala','soronga')
 160     px.user.add('culo','sarasa')
 161     px.commit()