From f638427afc92dda86452e563524d5528fee2b344 Mon Sep 17 00:00:00 2001 From: "louiz@4325f9fc-e183-4c21-96ce-0ab188b42d13" Date: Thu, 16 Sep 2010 17:57:27 +0000 Subject: LALALALALALALALALA --- src/roster.py | 45 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 src/roster.py (limited to 'src/roster.py') diff --git a/src/roster.py b/src/roster.py new file mode 100644 index 00000000..421e05e4 --- /dev/null +++ b/src/roster.py @@ -0,0 +1,45 @@ +# Copyright 2010 Le Coz Florent +# +# This file is part of Poezio. +# +# Poezio 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 of the License. +# +# Poezio 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 Poezio. If not, see . + +from contact import Contact + +class Roster(object): + """ + Defines the roster + """ + def __init__(self): + self._contacts = {} + + def addContactToList(self, contact): + assert isinstance(contact, Contact) + assert contact not in self._contacts + self._contacts[contact.getJid().bare] = contact + + def getContacts(self): + """ + returns all the contacts in a list + TODO: sorted + TODO: only some contacts (online only for example) + """ + return [contact for contact in self._contacts.keys()] + + def __len__(self): + return len(self._contacts) + + def getContact(self, bare_jid): + if bare_jid not in self._contacts: + return None + return self._contacts[bare_jid] -- cgit v1.2.3