Changeset 122

Timestamp:

10/30/07 20:17:18 (1 year ago)

Author:

omega

Message:

Y'a encore rien qui marche, mais y'a une bonne base pour le randomchat

Files:

• randomchat/randomchat.py (modified) (2 diffs)

randomchat/randomchat.py

@@ -5 +5 @@
-import config
-
+PRS_OFFLINE = 1
+PRS_CHAT = 2
+PRS_AVAILABLE = 3
+PRS_AWAY = 4
+PRS_XA = 5
+PRS_DND = 6
+
+def db_connect(config):
+    if config.DB_MODULE == 'sqlite':
+      from pysqlite2 import dbapi2 as sqlite
+      return sqlite.connect(config.SQLITE_DB)
+    else:
+        raise Exception("Incorrect DB module")
+
+class Users:
+    def __init__(self, db):
+        self.db = db
+        self.users_list = {}
+        self.read_users()
+
+    def read_users(self):
+        cursor = self.db.cursor()
+        cursor.execute("SELECT jid FROM randomchat WHERE 1")
+        user = cursor.fetchone()
+        while user:
+            self.users_list[user[0]] = PRS_OFFLINE
+            user = cursor.fetchone()
+        cursor.close()
+
+    def has_user(self, jid):
+        return jid in self.users_list
+
+    def add_user(self, jid):
+        self.db.execute("INSERT INTO randomchat (jid) VALUES (?)", (jid,))
+        self.db.commit()
+
+    def remove_user(self, jid):
+        self.db.execute("DELETE FROM randomchat WHERE jid=?", (jid,))
+        self.db.commit()
+
+    def get_chat_partner(jid):
+        pass
+
+
-class RandomChat:
-
+
+
-        self.con = xmpp.Component(server, port)
+        self.jid = jid
-        if not self.con.connect():
-            raise Exception('Unable to connect to %s:%s' % (server, port))
-        if not self.con.auth(jid, pwd):
-            raise Exception('Unable to authenticate as %s' % (jid))
-
+
+
-        self.con.RegisterHandler('presence', self.presence_handler)
+        self.con.RegisterHandler('presence', self.subscribe_handler, 'subscribe')
+        self.con.RegisterHandler('presence', self.subscribed_handler, 'subscribed')
+        self.con.RegisterHandler('presence', self.unsubscribe_handler, 'unsubscribe')
+        self.con.RegisterHandler('presence', self.unsubscribed_handler, 'unsubscribed')
+        self.con.RegisterHandler('message', self.message_handler)
-
-
@@ -19 +71 @@
-            self.con.Process(10)
-
+
+    def subscribe_handler(self, con, prs):
+        user = prs.getFrom().getStripped()
+        if not self.users.has_user(user):
+            self.con.send(xmpp.Presence(to = prs.getFrom(), frm = self.jid, typ = 'subscribe'))
+        self.con.send(xmpp.Presence(to = prs.getFrom(), frm = self.jid, typ = 'subscribed'))
+
+    def subscribed_handler(self, con, prs):
+        user = prs.getFrom().getStripped()
+        if not self.users.has_user(user):
+            self.users.add_user(user)
+
+    def unsubscribe_handler(self, con, prs):
+        self.con.send(xmpp.Presence(to = prs.getFrom(), frm = self.jid, typ = 'unsubscribe'))
+
+    def unsubscribed_handler(self, con, prs):
+        user = prs.getFrom().getStripped()
+        self.users.remove_user(user)
+        self.con.send(xmpp.Presence(to = prs.getFrom(), frm = self.jid, typ = 'unsubscribed'))
+
-    def presence_handler(self, con, prs):
-
-
-
+
-
-
+
+
+
+
+
+
+
+
+
+