source4/scripting/python/samba/hostconfig.py

   1 #!/usr/bin/env python
   2
   3 # Unix SMB/CIFS implementation.
   4 # Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2008
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 #
  19
  20 """Local host configuration."""
  21
  22 from samdb import SamDB
  23
  24 class Hostconfig(object):
  25     """Aggregate object that contains all information about the configuration
  26     of a Samba host."""
  27
  28     def __init__(self, lp):
  29         self.lp = lp
  30
  31     def get_shares(self):
  32         return SharesContainer(self.lp)
  33
  34     def get_samdb(self, session_info, credentials):
  35         """Access the SamDB host.
  36
  37         :param session_info: Session info to use
  38         :param credentials: Credentials to access the SamDB with
  39         """
  40         return SamDB(url=self.lp.samdb_url(),
  41                      session_info=session_info, credentials=credentials,
  42                      lp=self.lp)
  43
  44
  45 # TODO: Rather than accessing Loadparm directly here, we should really
  46 # have bindings to the param/shares.c and use those.
  47
  48
  49 class SharesContainer(object):
  50     """A shares container."""
  51
  52     def __init__(self, lp):
  53         self._lp = lp
  54
  55     def __getitem__(self, name):
  56         if name == "global":
  57             # [global] is not a share
  58             raise KeyError
  59         return Share(self._lp[name])
  60
  61     def __len__(self):
  62         if "global" in self._lp.services():
  63             return len(self._lp)-1
  64         return len(self._lp)
  65
  66     def keys(self):
  67         return [name for name in self._lp.services() if name != "global"]
  68
  69     def __iter__(self):
  70         return iter(self.keys())
  71
  72
  73 class Share(object):
  74     """A file share."""
  75
  76     def __init__(self, service):
  77         self._service = service
  78
  79     def __getitem__(self, name):
  80         return self._service[name]
  81
  82     def __setitem__(self, name, value):
  83         self._service[name] = value