Changeset 122

Timestamp:

10/30/07 20:17:18 (2 years 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:
-    def __init__(self, jid, pwd, server, port=5347):
+    def __init__(self, users, jid, pwd, server, port=5347):
+        self.users = users
         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))
-        
+
+        #register the handlers 
         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):
-        print '***********************************************************'
-        print prs.getFrom()
-        print '***********************************************************'
+        pass
 
-rc = RandomChat(config.JID, config.PASS, config.SERVER, config.PORT)
+    def message_handler(self, con, msg):
+        self.init_chat(msg.getFrom())
+
+    def init_chat(self, jid):
+        other_jid = self.users.get_chat_partner(jid)
+
+
+db = db_connect(config)
+users = Users(db)
+rc = RandomChat(users, config.JID, config.PASS, config.SERVER, config.PORT)