Garmelon/yaboli
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Generalize querying sessions in listing

Browse source
This commit is contained in:
Joscha 2017-09-05 22:33:52 +00:00
parent 3945c6ae45
commit d783abf014
1 changed files with 20 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

32
yaboli/utils.py
View file

@ -150,21 +150,29 @@ class Listing:
def by_uid(self, user_id): def by_uid(self, user_id):
return [ses for ses in self._sessions if ses.user_id == user_id] return [ses for ses in self._sessions if ses.user_id == user_id]
def get_people(self): def get(self, types=["agent", "account", "bot"], lurker=None):
return {uid: ses for uid, ses in self._sessions.items() sessions = []
if ses.client_type in ["agent", "account"]} for uid, ses in self._sessions.items():
if ses.client_type not in types:
continue
is_lurker = not ses.nick # "" or None
if lurker is None or lurker == is_lurker:
sessions.append(ses)
return sessions
def get_accounts(self): #def get_people(self):
return {uid: ses for uid, ses in self._sessions.items() #return self.get(types=["agent", "account"])
if ses.client_type is "account"}
def get_agents(self): #def get_accounts(self):
return {uid: ses for uid, ses in self._sessions.items() #return self.get(types=["account"])
if ses.client_type is "agent"}
def get_bots(self): #def get_agents(self):
return {uid: ses for uid, ses in self._sessions.items() #return self.get(types=["agent"])
if ses.client_type is "bot"}
#def get_bots(self):
#return self.get(types=["bot"])
class Message(): class Message():
def __init__(self, message_id, time, sender, content, parent=None, previous_edit_id=None, def __init__(self, message_id, time, sender, content, parent=None, previous_edit_id=None,