Mercurial > libervia-backend
annotate sat_frontends/jp/cmd_roster.py @ 3366:e09cb08166a3
plugin XEP-0329, core(xmpp): moved `_compParseJids` to `SatXMPPComponent`:
This method to retrieve owner and peer JID from an element is generally useful for
components, thus it has been moved. The part to retrieve owner JID from local JID has been
splitted in its own `getOwnerFromJid` method.
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Sun, 20 Sep 2020 14:04:11 +0200 |
| parents | 012e89fb2dd1 |
| children | f30b238d9c45 |
| rev | line source |
|---|---|
| 3137 | 1 #!/usr/bin/env python3 |
| 2 | |
|
3255
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
3 # jp: a SàT command line tool |
| 3136 | 4 # Copyright (C) 2009-2020 Jérôme Poisson (goffi@goffi.org) |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
5 # Copyright (C) 2003-2016 Adrien Cossa (souliane@mailoo.org) |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
6 |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
7 # This program is free software: you can redistribute it and/or modify |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
8 # it under the terms of the GNU Affero General Public License as published by |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
9 # the Free Software Foundation, either version 3 of the License, or |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
10 # (at your option) any later version. |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
11 |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
12 # This program is distributed in the hope that it will be useful, |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
15 # GNU Affero General Public License for more details. |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
16 |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
17 # You should have received a copy of the GNU Affero General Public License |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
18 # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
19 |
| 3028 | 20 from . import base |
|
2893
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
21 from collections import OrderedDict |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
22 from sat.core.i18n import _ |
|
2893
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
23 from sat_frontends.jp.constants import Const as C |
| 3040 | 24 from sat_frontends.tools import jid |
| 25 from sat.tools.common.ansi import ANSI as A | |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
26 |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
27 __commands__ = ["Roster"] |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
28 |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
29 |
| 3040 | 30 class Get(base.CommandBase): |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
31 |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
32 def __init__(self, host): |
| 3040 | 33 super().__init__( |
| 34 host, 'get', use_output=C.OUTPUT_DICT, use_verbose=True, | |
| 35 extra_outputs = {"default": self.default_output}, | |
| 36 help=_('retrieve the roster entities')) | |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
37 |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
38 def add_parser_options(self): |
| 3040 | 39 pass |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
40 |
| 3040 | 41 def default_output(self, data): |
| 42 for contact_jid, contact_data in data.items(): | |
| 43 all_keys = list(contact_data.keys()) | |
| 44 keys_to_show = [] | |
| 45 name = contact_data.get('name', contact_jid.node) | |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
46 |
| 3040 | 47 if self.verbosity >= 1: |
| 48 keys_to_show.append('groups') | |
| 49 all_keys.remove('groups') | |
| 50 if self.verbosity >= 2: | |
| 51 keys_to_show.extend(all_keys) | |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
52 |
| 3040 | 53 if name is None: |
| 54 self.disp(A.color(C.A_HEADER, contact_jid)) | |
| 55 else: | |
| 56 self.disp(A.color(C.A_HEADER, name, A.RESET, f" ({contact_jid})")) | |
| 57 for k in keys_to_show: | |
| 58 value = contact_data[k] | |
| 59 if value: | |
| 60 if isinstance(value, list): | |
| 61 value = ', '.join(value) | |
| 62 self.disp(A.color( | |
| 63 " ", C.A_SUBHEADER, f"{k}: ", A.RESET, str(value))) | |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
64 |
| 3040 | 65 async def start(self): |
| 66 try: | |
| 67 contacts = await self.host.bridge.getContacts(profile_key=self.host.profile) | |
| 68 except Exception as e: | |
| 69 self.disp(f"error while retrieving the contacts: {e}", error=True) | |
| 70 self.host.quit(C.EXIT_BRIDGE_ERRBACK) | |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
71 |
| 3040 | 72 contacts_dict = {} |
| 73 for contact_jid_s, data, groups in contacts: | |
| 74 # FIXME: we have to convert string to bool here for historical reason | |
| 75 # getContacts format should be changed and serialised properly | |
| 76 for key in ('from', 'to', 'ask'): | |
| 77 if key in data: | |
| 78 data[key] = C.bool(data[key]) | |
| 79 data['groups'] = list(groups) | |
| 80 contacts_dict[jid.JID(contact_jid_s)] = data | |
| 81 | |
| 82 await self.output(contacts_dict) | |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
83 self.host.quit() |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
84 |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
85 |
|
3255
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
86 class Set(base.CommandBase): |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
87 |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
88 def __init__(self, host): |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
89 super().__init__(host, 'set', help=_('set metadata for a roster entity')) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
90 |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
91 def add_parser_options(self): |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
92 self.parser.add_argument( |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
93 "-n", "--name", default="", help=_('name to use for this entity')) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
94 self.parser.add_argument( |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
95 "-g", "--group", dest='groups', action='append', metavar='GROUP', default=[], |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
96 help=_('groups for this entity')) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
97 self.parser.add_argument( |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
98 "-R", "--replace", action="store_true", |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
99 help=_("replace all metadata instead of adding them")) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
100 self.parser.add_argument( |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
101 "jid", help=_("jid of the roster entity")) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
102 |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
103 async def start(self): |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
104 |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
105 if self.args.replace: |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
106 name = self.args.name |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
107 groups = self.args.groups |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
108 else: |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
109 try: |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
110 entity_data = await self.host.bridge.contactGet( |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
111 self.args.jid, self.host.profile) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
112 except Exception as e: |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
113 self.disp(f"error while retrieving the contact: {e}", error=True) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
114 self.host.quit(C.EXIT_BRIDGE_ERRBACK) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
115 name = self.args.name or entity_data[0].get('name') or '' |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
116 groups = set(entity_data[1]) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
117 groups = list(groups.union(self.args.groups)) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
118 |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
119 try: |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
120 await self.host.bridge.updateContact( |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
121 self.args.jid, name, groups, self.host.profile) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
122 except Exception as e: |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
123 self.disp(f"error while updating the contact: {e}", error=True) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
124 self.host.quit(C.EXIT_BRIDGE_ERRBACK) |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
125 self.host.quit() |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
126 |
|
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
127 |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
128 class Stats(base.CommandBase): |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
129 |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
130 def __init__(self, host): |
|
1798
c5d58387d031
jp (roster stats): add some extra information
souliane <souliane@mailoo.org>
parents:
1797
diff
changeset
|
131 super(Stats, self).__init__(host, 'stats', help=_('Show statistics about a roster')) |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
132 |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
133 def add_parser_options(self): |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
134 pass |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
135 |
| 3040 | 136 async def start(self): |
| 137 try: | |
| 138 contacts = await self.host.bridge.getContacts(profile_key=self.host.profile) | |
| 139 except Exception as e: | |
| 140 self.disp(f"error while retrieving the contacts: {e}", error=True) | |
| 141 self.host.quit(C.EXIT_BRIDGE_ERRBACK) | |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
142 |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
143 hosts = {} |
|
1798
c5d58387d031
jp (roster stats): add some extra information
souliane <souliane@mailoo.org>
parents:
1797
diff
changeset
|
144 unique_groups = set() |
|
c5d58387d031
jp (roster stats): add some extra information
souliane <souliane@mailoo.org>
parents:
1797
diff
changeset
|
145 no_sub, no_from, no_to, no_group, total_group_subscription = 0, 0, 0, 0, 0 |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
146 for contact, attrs, groups in contacts: |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
147 from_, to = C.bool(attrs["from"]), C.bool(attrs["to"]) |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
148 if not from_: |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
149 if not to: |
|
1798
c5d58387d031
jp (roster stats): add some extra information
souliane <souliane@mailoo.org>
parents:
1797
diff
changeset
|
150 no_sub += 1 |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
151 else: |
|
1798
c5d58387d031
jp (roster stats): add some extra information
souliane <souliane@mailoo.org>
parents:
1797
diff
changeset
|
152 no_from += 1 |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
153 elif not to: |
|
1798
c5d58387d031
jp (roster stats): add some extra information
souliane <souliane@mailoo.org>
parents:
1797
diff
changeset
|
154 no_to += 1 |
| 3040 | 155 |
| 156 host = jid.JID(contact).domain | |
| 157 | |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
158 hosts.setdefault(host, 0) |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
159 hosts[host] += 1 |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
160 if groups: |
|
1798
c5d58387d031
jp (roster stats): add some extra information
souliane <souliane@mailoo.org>
parents:
1797
diff
changeset
|
161 unique_groups.update(groups) |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
162 total_group_subscription += len(groups) |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
163 if not groups: |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
164 no_group += 1 |
| 3028 | 165 hosts = OrderedDict(sorted(list(hosts.items()), key=lambda item:-item[1])) |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
166 |
| 3028 | 167 print() |
| 168 print("Total number of contacts: %d" % len(contacts)) | |
| 169 print("Number of different hosts: %d" % len(hosts)) | |
| 170 print() | |
| 171 for host, count in hosts.items(): | |
| 3040 | 172 print("Contacts on {host}: {count} ({rate:.1f}%)".format( |
| 173 host=host, count=count, rate=100 * float(count) / len(contacts))) | |
| 3028 | 174 print() |
| 175 print("Contacts with no 'from' subscription: %d" % no_from) | |
| 176 print("Contacts with no 'to' subscription: %d" % no_to) | |
| 177 print("Contacts with no subscription at all: %d" % no_sub) | |
| 178 print() | |
| 179 print("Total number of groups: %d" % len(unique_groups)) | |
|
1932
44342730df66
jp (cmd/roster): fixes division by zero exceptions
souliane <souliane@mailoo.org>
parents:
1864
diff
changeset
|
180 try: |
|
44342730df66
jp (cmd/roster): fixes division by zero exceptions
souliane <souliane@mailoo.org>
parents:
1864
diff
changeset
|
181 contacts_per_group = float(total_group_subscription) / len(unique_groups) |
|
44342730df66
jp (cmd/roster): fixes division by zero exceptions
souliane <souliane@mailoo.org>
parents:
1864
diff
changeset
|
182 except ZeroDivisionError: |
|
44342730df66
jp (cmd/roster): fixes division by zero exceptions
souliane <souliane@mailoo.org>
parents:
1864
diff
changeset
|
183 contacts_per_group = 0 |
| 3028 | 184 print("Average contacts per group: {:.1f}".format(contacts_per_group)) |
|
1932
44342730df66
jp (cmd/roster): fixes division by zero exceptions
souliane <souliane@mailoo.org>
parents:
1864
diff
changeset
|
185 try: |
|
44342730df66
jp (cmd/roster): fixes division by zero exceptions
souliane <souliane@mailoo.org>
parents:
1864
diff
changeset
|
186 groups_per_contact = float(total_group_subscription) / len(contacts) |
|
44342730df66
jp (cmd/roster): fixes division by zero exceptions
souliane <souliane@mailoo.org>
parents:
1864
diff
changeset
|
187 except ZeroDivisionError: |
|
44342730df66
jp (cmd/roster): fixes division by zero exceptions
souliane <souliane@mailoo.org>
parents:
1864
diff
changeset
|
188 groups_per_contact = 0 |
| 3040 | 189 print(f"Average groups' subscriptions per contact: {groups_per_contact:.1f}") |
| 3028 | 190 print("Contacts not assigned to any group: %d" % no_group) |
|
1796
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
191 self.host.quit() |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
192 |
|
314d2eb7fbaa
jp: add command "roster stats"
souliane <souliane@mailoo.org>
parents:
1793
diff
changeset
|
193 |
| 3040 | 194 class Purge(base.CommandBase): |
|
1799
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
195 |
|
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
196 def __init__(self, host): |
| 3040 | 197 super(Purge, self).__init__( |
| 198 host, 'purge', | |
| 199 help=_('purge the roster from its contacts with no subscription')) | |
|
1799
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
200 |
|
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
201 def add_parser_options(self): |
| 3040 | 202 self.parser.add_argument( |
| 203 "--no_from", action="store_true", | |
| 204 help=_("also purge contacts with no 'from' subscription")) | |
| 205 self.parser.add_argument( | |
| 206 "--no_to", action="store_true", | |
| 207 help=_("also purge contacts with no 'to' subscription")) | |
|
1799
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
208 |
| 3040 | 209 async def start(self): |
| 210 try: | |
| 211 contacts = await self.host.bridge.getContacts(self.host.profile) | |
| 212 except Exception as e: | |
| 213 self.disp(f"error while retrieving the contacts: {e}", error=True) | |
| 214 self.host.quit(C.EXIT_BRIDGE_ERRBACK) | |
|
1799
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
215 |
| 3040 | 216 no_sub, no_from, no_to = [], [], [] |
| 217 for contact, attrs, groups in contacts: | |
| 218 from_, to = C.bool(attrs["from"]), C.bool(attrs["to"]) | |
| 219 if not from_: | |
| 220 if not to: | |
| 221 no_sub.append(contact) | |
| 222 elif self.args.no_from: | |
| 223 no_from.append(contact) | |
| 224 elif not to and self.args.no_to: | |
| 225 no_to.append(contact) | |
| 226 if not no_sub and not no_from and not no_to: | |
| 227 self.disp( | |
| 228 f"Nothing to do - there's a from and/or to subscription(s) between " | |
| 229 f"profile {self.host.profile!r} and each of its contacts" | |
| 230 ) | |
| 231 elif await self.ask_confirmation(no_sub, no_from, no_to): | |
| 232 for contact in no_sub + no_from + no_to: | |
| 233 try: | |
| 234 await self.host.bridge.delContact( | |
| 235 contact, profile_key=self.host.profile) | |
| 236 except Exception as e: | |
| 237 self.disp(f"can't delete contact {contact!r}: {e}", error=True) | |
| 238 else: | |
| 239 self.disp(f"contact {contact!r} has been removed") | |
|
1799
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
240 |
| 3040 | 241 self.host.quit() |
| 242 | |
| 243 async def ask_confirmation(self, no_sub, no_from, no_to): | |
| 244 """Ask the confirmation before removing contacts. | |
|
1799
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
245 |
| 3040 | 246 @param no_sub (list[unicode]): list of contacts with no subscription |
| 247 @param no_from (list[unicode]): list of contacts with no 'from' subscription | |
| 248 @param no_to (list[unicode]): list of contacts with no 'to' subscription | |
| 249 @return bool | |
|
1799
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
250 """ |
| 3040 | 251 if no_sub: |
| 252 self.disp( | |
| 253 f"There's no subscription between profile {self.host.profile!r} and the " | |
| 254 f"following contacts:") | |
| 255 self.disp(" " + "\n ".join(no_sub)) | |
| 256 if no_from: | |
| 257 self.disp( | |
| 258 f"There's no 'from' subscription between profile {self.host.profile!r} " | |
| 259 f"and the following contacts:") | |
| 260 self.disp(" " + "\n ".join(no_from)) | |
| 261 if no_to: | |
| 262 self.disp( | |
| 263 f"There's no 'to' subscription between profile {self.host.profile!r} and " | |
| 264 f"the following contacts:") | |
| 265 self.disp(" " + "\n ".join(no_to)) | |
| 266 message = f"REMOVE them from profile {self.host.profile}'s roster" | |
| 267 while True: | |
| 268 res = await self.host.ainput(f"{message} (y/N)? ") | |
| 269 if not res or res.lower() == 'n': | |
| 270 return False | |
| 271 if res.lower() == 'y': | |
| 272 return True | |
|
1799
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
273 |
|
f1f4775f8cc0
jp (roster): add command "roster get" with parameters "--subscriptions", "--name" and "--groups"
souliane <souliane@mailoo.org>
parents:
1798
diff
changeset
|
274 |
|
2893
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
275 class Resync(base.CommandBase): |
|
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
276 |
|
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
277 def __init__(self, host): |
|
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
278 super(Resync, self).__init__( |
| 3028 | 279 host, 'resync', help=_('do a full resynchronisation of roster with server')) |
|
2893
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
280 |
|
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
281 def add_parser_options(self): |
|
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
282 pass |
|
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
283 |
| 3040 | 284 async def start(self): |
| 285 try: | |
| 286 await self.host.bridge.rosterResync(profile_key=self.host.profile) | |
| 287 except Exception as e: | |
| 288 self.disp(f"can't resynchronise roster: {e}", error=True) | |
| 289 self.host.quit(C.EXIT_BRIDGE_ERRBACK) | |
| 290 else: | |
| 291 self.disp(_("Roster resynchronized")) | |
| 292 self.host.quit(C.EXIT_OK) | |
|
2893
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
293 |
|
7c8773723200
jp (roster): added resync command to do a full roster resynchronisation
Goffi <goffi@goffi.org>
parents:
2771
diff
changeset
|
294 |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
295 class Roster(base.CommandBase): |
|
3255
012e89fb2dd1
jp (roster): new roster/set command
Goffi <goffi@goffi.org>
parents:
3137
diff
changeset
|
296 subcommands = (Get, Set, Stats, Purge, Resync) |
|
1793
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
297 |
|
f39ca2832774
jp: add command "roster purge" to remove the contacts with no from/to subscription
souliane <souliane@mailoo.org>
parents:
diff
changeset
|
298 def __init__(self, host): |
| 3040 | 299 super(Roster, self).__init__( |
| 300 host, 'roster', use_profile=True, help=_("Manage an entity's roster")) |