diff options
author | Jelmer Vernooij <jelmer@samba.org> | 2009-06-16 02:05:18 +0200 |
---|---|---|
committer | Jelmer Vernooij <jelmer@samba.org> | 2009-06-16 02:28:10 +0200 |
commit | f7ada51c29a517e841843c2a869ac9d16c7057d4 (patch) | |
tree | df089aeacbe84dff11c7e8813381870e399a343e | |
parent | 537aa4cf51def606de6c73c429f97d12c76333b1 (diff) | |
download | samba-f7ada51c29a517e841843c2a869ac9d16c7057d4.tar.gz samba-f7ada51c29a517e841843c2a869ac9d16c7057d4.tar.bz2 samba-f7ada51c29a517e841843c2a869ac9d16c7057d4.zip |
python: Add API for accessing available shares.
-rw-r--r-- | source4/bin/python/samba/tests/shares.py | 42 | ||||
-rw-r--r-- | source4/scripting/python/samba/shares.py | 58 | ||||
-rwxr-xr-x | source4/selftest/tests.sh | 1 |
3 files changed, 101 insertions, 0 deletions
diff --git a/source4/bin/python/samba/tests/shares.py b/source4/bin/python/samba/tests/shares.py new file mode 100644 index 0000000000..0a771b1754 --- /dev/null +++ b/source4/bin/python/samba/tests/shares.py @@ -0,0 +1,42 @@ +#!/usr/bin/python + +# Unix SMB/CIFS implementation. Tests for shares +# Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2009 +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. +# +from samba.shares import SharesContainer +from unittest import TestCase + + +class ShareTests(TestCase): + + def _get_shares(self, conf): + return SharesContainer(conf) + + def test_len_no_global(self): + shares = self._get_shares({}) + self.assertEquals(0, len(shares)) + + def test_len(self): + shares = self._get_shares({"global": {}}) + self.assertEquals(0, len(shares)) + + def test_getitem_nonexistant(self): + shares = self._get_shares({"global": {}}) + self.assertRaises(KeyError, shares.__getitem__, "bla") + + def test_getitem_global(self): + shares = self._get_shares({"global": {}}) + self.assertRaises(KeyError, shares.__getitem__, "global") diff --git a/source4/scripting/python/samba/shares.py b/source4/scripting/python/samba/shares.py new file mode 100644 index 0000000000..622c3b6486 --- /dev/null +++ b/source4/scripting/python/samba/shares.py @@ -0,0 +1,58 @@ +#!/usr/bin/python + +# Unix SMB/CIFS implementation. +# Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2009 +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. +# + +"""Share management.""" + + +# TODO: Rather than accessing Loadparm directly here, we should really +# have bindings to the param/shares.c and use those. + + +class SharesContainer(object): + """A shares container.""" + + def __init__(self, lp): + self._lp = lp + + def __getitem__(self, name): + if name == "global": + # [global] is not a share + raise KeyError + return Share(self._lp[name]) + + def __len__(self): + if "global" in self._lp: + return len(self._lp)-1 + return len(self._lp) + + def __iter__(self): + return self.lp.services() + + +class Share(object): + """A file share.""" + + def __init__(self, service): + self._service = service + + def __getitem__(self, name): + return self._service[name] + + def __setitem__(self, name, value): + self._service[name] = value diff --git a/source4/selftest/tests.sh b/source4/selftest/tests.sh index 04f8d2b0c8..56462ac097 100755 --- a/source4/selftest/tests.sh +++ b/source4/selftest/tests.sh @@ -418,6 +418,7 @@ plantest "samr.python" dc:local $SUBUNITRUN samba.tests.dcerpc.sam plantest "dcerpc.bare.python" dc:local $SUBUNITRUN samba.tests.dcerpc.bare plantest "unixinfo.python" dc:local $SUBUNITRUN samba.tests.dcerpc.unix plantest "samdb.python" none $SUBUNITRUN samba.tests.samdb +plantest "shares.python" none $SUBUNITRUN samba.tests.shares plantest "messaging.python" none PYTHONPATH="$PYTHONPATH:$samba4srcdir/lib/messaging/tests" $SUBUNITRUN bindings plantest "samba3sam.python" none PYTHONPATH="$PYTHONPATH:$samba4srcdir/dsdb/samdb/ldb_modules/tests" $SUBUNITRUN samba3sam plantest "subunit.python" none $SUBUNITRUN subunit |