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