Changeset 126

Timestamp:

10/31/07 17:36:28 (3 years ago)

Author:

omega

Message:

invite seulement ceux qui sont en free for chat

Files:

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

randomchat/randomchat.py

@@ -10 +10 @@
 
 NS_MUC_USER = xmpp.NS_MUC+'#user'
-
-PRS_OFFLINE = 1
-PRS_CHAT = 2
-PRS_AVAILABLE = 3
-PRS_AWAY = 4
-PRS_XA = 5
-PRS_DND = 6
 
 def db_connect(config):
@@ -36 +29 @@
         user = cursor.fetchone()
         while user:
-            self.users_list[user[0]] = PRS_OFFLINE
+            self.users_list[user[0]] = False
             user = cursor.fetchone()
         cursor.close()
@@ -46 +39 @@
         self.db.execute("INSERT INTO randomchat (jid) VALUES (?)", (jid,))
         self.db.commit()
-        self.users_list[jid] = PRS_OFFLINE
+        self.users_list[jid] = False
 
     def remove_user(self, jid):
@@ -55 +48 @@
     def get_chat_partner(self, jid):
         user = jid.getStripped()
-        if len(self.users_list) < 2:
+        users = [jid for jid in self.users_list.keys() if self.users_list[jid]]
+        #if we have only one participant don't start a chat, except when the user is not in this list (so in fact we have two participants)
+        if (len(users) < 2 and user in users) or len(users) < 1:
             return None
         else:
-            users = self.users_list.keys()
             other = random.choice(users)
             while other == user:
                 other = random.choice(users)
             return other
+
+    def update_status(self, jid, show):
+        if not self.has_user(jid):
+            return
+        if show == 'chat':
+            self.users_list[jid] = True
+        else:
+            self.users_list[jid] = False
+        print self.users_list
+
+    def jid_list(self):
+        return self.users_list.keys()
+
+
 
 class MUC:
@@ -189 +197 @@
         self.con.RegisterHandler('message', self.message_handler)
 
+        self.send_probes()
+
 
         while 1:
@@ -217 +227 @@
             self.muc.presence_handler(con, prs)
             return
-        pass
+        self.users.update_status(prs.getFrom().getStripped(), prs.getShow())
+
 
     def message_handler(self, con, msg):
@@ -233 +244 @@
         return True
 
+    def send_probes(self):
+        for jid in self.users.jid_list():
+            self.con.send(xmpp.Presence(to = jid, frm = self.jid, typ='probe'))
+
 
 db = db_connect(config)