- Add commands to add and remove user's subscriptions to/from folders

2 files changed, 192 insertions, 0 deletions

diff --git a/pykolab/cli/cmd_add_user_subscription.py b/pykolab/cli/cmd_add_user_subscription.py
new file mode 100644
index 0000000..38dc2d7
--- /dev/null
+++ b/pykolab/cli/cmd_add_user_subscription.py
@@ -0,0 +1,96 @@
+# -*- coding: utf-8 -*-
+# Copyright 2010-2012 Kolab Systems AG (http://www.kolabsys.com)
+#
+# Jeroen van Meeuwen (Kolab Systems) <vanmeeuwen a kolabsys.com>
+#
+# 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; version 3 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 Library General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+#
+
+import sys
+
+import commands
+
+import pykolab
+
+from pykolab.imap import IMAP
+from pykolab.translate import _
+from pykolab import utils
+
+log = pykolab.getLogger('pykolab.cli')
+conf = pykolab.getConf()
+
+def __init__():
+    commands.register('add_user_subscription', execute, description=description())
+
+def description():
+    return _("Subscribe a user to a folder.")
+
+def execute(*args, **kw):
+    folder_pattern = "*"
+
+    try:
+        user = conf.cli_args.pop(0)
+        try:
+            folder_pattern = conf.cli_args.pop(0)
+        except IndexError, errmsg:
+            folder_pattern = utils.ask_question(_("Folder pattern"))
+
+    except IndexError, errmsg:
+        user = utils.ask_question(_("User ID"))
+        folder_pattern = utils.ask_question(_("Folder pattern"))
+
+    if len(user.split('@')) > 1:
+        domain = user.split('@')[1]
+    else:
+        domain = conf.get('kolab', 'primary_domain')
+
+    imap = IMAP()
+    imap.connect(domain=domain, login=False)
+
+    backend = conf.get(domain, 'imap_backend')
+    if backend == None:
+        backend = conf.get('kolab', 'imap_backend')
+
+    admin_login = conf.get(backend, 'admin_login')
+    admin_password = conf.get(backend, 'admin_password')
+
+    imap.login_plain(admin_login, admin_password, user)
+
+    if not imap.has_folder(folder_pattern):
+        print >> sys.stderr, \
+                _("Cannot subscribe user to folder %r:") % (folder_pattern), \
+                _("No such folder")
+        sys.exit(1)
+
+    _folders = imap.lm(folder_pattern)
+    _subscribed_folders = imap.lsub(folder_pattern)
+    subscribed_folders = []
+
+    for _folder in _folders:
+        if not _folder in _subscribed_folders:
+            imap.subscribe(_folder)
+            subscribed_folders.append(_folder)
+
+    if len(subscribed_folders) > 0:
+        print _("Successfully subscribed user %s to the following folders:") % (
+                user
+            )
+
+        print "\n".join(subscribed_folders)
+    else:
+        print >> sys.stderr, _("User %s not subscribed to any folders.") % (
+                user
+            )
+
+        sys.exit(1)
\ No newline at end of file
diff --git a/pykolab/cli/cmd_remove_user_subscription.py b/pykolab/cli/cmd_remove_user_subscription.py
new file mode 100644
index 0000000..79e9d09
--- /dev/null
+++ b/pykolab/cli/cmd_remove_user_subscription.py
@@ -0,0 +1,96 @@
+# -*- coding: utf-8 -*-
+# Copyright 2010-2012 Kolab Systems AG (http://www.kolabsys.com)
+#
+# Jeroen van Meeuwen (Kolab Systems) <vanmeeuwen a kolabsys.com>
+#
+# 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; version 3 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 Library General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+#
+
+import sys
+
+import commands
+
+import pykolab
+
+from pykolab.imap import IMAP
+from pykolab.translate import _
+from pykolab import utils
+
+log = pykolab.getLogger('pykolab.cli')
+conf = pykolab.getConf()
+
+def __init__():
+    commands.register('remove_user_subscription', execute, description=description())
+
+def description():
+    return _("Unsubscribe a user from a folder.")
+
+def execute(*args, **kw):
+    folder_pattern = "*"
+
+    try:
+        user = conf.cli_args.pop(0)
+        try:
+            folder_pattern = conf.cli_args.pop(0)
+        except IndexError, errmsg:
+            folder_pattern = utils.ask_question(_("Folder pattern"))
+
+    except IndexError, errmsg:
+        user = utils.ask_question(_("User ID"))
+        folder_pattern = utils.ask_question(_("Folder pattern"))
+
+    if len(user.split('@')) > 1:
+        domain = user.split('@')[1]
+    else:
+        domain = conf.get('kolab', 'primary_domain')
+
+    imap = IMAP()
+    imap.connect(domain=domain, login=False)
+
+    backend = conf.get(domain, 'imap_backend')
+    if backend == None:
+        backend = conf.get('kolab', 'imap_backend')
+
+    admin_login = conf.get(backend, 'admin_login')
+    admin_password = conf.get(backend, 'admin_password')
+
+    imap.login_plain(admin_login, admin_password, user)
+
+    if not imap.has_folder(folder_pattern):
+        print >> sys.stderr, \
+                _("Cannot subscribe user to folder %r:") % (folder_pattern), \
+                _("No such folder")
+        sys.exit(1)
+
+    _folders = imap.lm(folder_pattern)
+    _subscribed_folders = imap.lsub(folder_pattern)
+    unsubscribed_folders = []
+
+    for _folder in _folders:
+        if _folder in _subscribed_folders:
+            imap.unsubscribe(_folder)
+            unsubscribed_folders.append(_folder)
+
+    if len(unsubscribed_folders) > 0:
+        print _("Successfully unsubscribed user %s from the following folders:") % (
+                user
+            )
+
+        print "\n".join(unsubscribed_folders)
+    else:
+        print >> sys.stderr, _("User %s not be unsubscribed from any folders.") % (
+                user
+            )
+
+        sys.exit(1)
\ No newline at end of file