Adding logging to close so I can be sure my channels close
[theonering] / src / channel / contact_list.py
1 import logging
2
3 import telepathy
4
5 import tp
6 import util.coroutines as coroutines
7 import gtk_toolbox
8 import handle
9
10
11 _moduleLogger = logging.getLogger("channel.contact_list")
12
13
14 class AllContactsListChannel(
15                 tp.ChannelTypeContactList,
16                 tp.ChannelInterfaceGroup,
17         ):
18         """
19         The group of contacts for whom you receive presence
20
21         @bug On Maemo 5 this isn't even being created, I think
22         """
23
24         def __init__(self, connection, manager, props, listHandle):
25                 tp.ChannelTypeContactList.__init__(self, connection, manager, props)
26                 tp.ChannelInterfaceGroup.__init__(self)
27
28                 self.__manager = manager
29                 self.__props = props
30                 self.__session = connection.session
31                 self.__listHandle = listHandle
32
33                 self._callback = coroutines.func_sink(
34                         coroutines.expand_positional(
35                                 self._on_contacts_refreshed
36                         )
37                 )
38                 self.__session.addressbook.updateSignalHandler.register_sink(
39                         self._callback
40                 )
41
42                 self.GroupFlagsChanged(0, 0)
43
44                 addressbook = connection.session.addressbook
45                 contacts = addressbook.get_numbers()
46                 self._process_refresh(addressbook, set(contacts), set())
47
48
49         @gtk_toolbox.log_exception(_moduleLogger)
50         def Close(self):
51                 self.close()
52
53         def close(self):
54                 _moduleLogger.info("Closing contact list")
55                 self.__session.addressbook.updateSignalHandler.unregister_sink(
56                         self._callback
57                 )
58                 self._callback = None
59
60                 tp.ChannelTypeContactList.Close(self)
61                 self.remove_from_connection()
62
63         @gtk_toolbox.log_exception(_moduleLogger)
64         def _on_contacts_refreshed(self, addressbook, added, removed, changed):
65                 self._process_refresh(addressbook, added, removed)
66
67         def _process_refresh(self, addressbook, added, removed):
68                 _moduleLogger.info(
69                         "%s Added: %r, Removed: %r" % (self.__listHandle.get_name(), len(added), len(removed))
70                 )
71                 connection = self._conn
72                 handlesAdded = [
73                         handle.create_handle(connection, "contact", contactNumber)
74                         for contactNumber in added
75                 ]
76                 handlesRemoved = [
77                         handle.create_handle(connection, "contact", contactNumber)
78                         for contactNumber in removed
79                 ]
80                 message = ""
81                 actor = 0
82                 reason = telepathy.CHANNEL_GROUP_CHANGE_REASON_NONE
83                 self.MembersChanged(
84                         message,
85                         handlesAdded, handlesRemoved,
86                         (), (),
87                         actor,
88                         reason,
89                 )
90
91
92 _LIST_TO_FACTORY = {
93         # The group of contacts for whom you receive presence
94         'subscribe': AllContactsListChannel,
95         # The group of contacts who may receive your presence
96         'publish': AllContactsListChannel,
97         # A group of contacts who are on the publish list but are temporarily
98         # disallowed from receiving your presence
99         # This doesn't make sense to support
100         'hide': None,
101         # A group of contacts who may send you messages
102         # @todo Allow-List would be cool to support
103         'allow': None,
104         # A group of contacts who may not send you messages
105         # @todo Deny-List would be cool to support
106         'deny': None,
107         # On protocols where the user's contacts are stored, this contact list
108         # contains all stored contacts regardless of subscription status.
109         'stored': AllContactsListChannel,
110 }
111
112
113 _SUPPORTED_LISTS = frozenset(
114         name
115         for name in _LIST_TO_FACTORY.iterkeys()
116         if name is not None
117 )
118
119
120 def create_contact_list_channel(connection, manager, props, h):
121         factory = _LIST_TO_FACTORY.get(h.get_name(), None)
122         if factory is None:
123                 raise telepathy.errors.NotCapable("Unsuported type %s" % h.get_name())
124         return factory(connection, manager, props, h)
125
126
127 def get_spported_lists():
128         return _SUPPORTED_LISTS