summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/contact.py29
-rw-r--r--src/roster.py45
2 files changed, 74 insertions, 0 deletions
diff --git a/src/contact.py b/src/contact.py
new file mode 100644
index 00000000..c3b07fc2
--- /dev/null
+++ b/src/contact.py
@@ -0,0 +1,29 @@
+# Copyright 2010 Le Coz Florent <louizatakk@fedoraproject.org>
+#
+# 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 <http://www.gnu.org/licenses/>.
+
+from sleekxmpp.xmlstream.jid import JID
+
+class Contact(object):
+ """
+ Defines a roster item
+ """
+ def __init__(self, jid):
+ self._jid = JID(jid) # a SleekXMPP jid object
+ self._display_name = None
+ self.groups = [] # a list of groups the contact is in
+
+ def getJid(self):
+ return self._jid
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 <louizatakk@fedoraproject.org>
+#
+# 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 <http://www.gnu.org/licenses/>.
+
+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]