diff options
| -rw-r--r-- | src/room.py | 85 | ||||
| -rw-r--r-- | src/user.py | 39 |
2 files changed, 124 insertions, 0 deletions
diff --git a/src/room.py b/src/room.py new file mode 100644 index 00000000..87a78107 --- /dev/null +++ b/src/room.py @@ -0,0 +1,85 @@ +#!/usr/bin/python +# -*- coding:utf-8 -*- +# +# 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 datetime import datetime +from random import randrange +from config import config +from logging import logger + +class Room(object): + """ + """ + def __init__(self, name, nick, number): + self.name = name + self.own_nick = nick + self.color_state = 11 # color used in RoomInfo + self.nb = number # number used in RoomInfo + self.joined = False # false until self presence is received + self.users = [] + self.lines = [] # (time, nick, msg) or (time, info) + self.topic = '' + + def disconnect(self): + self.joined = False + self.users = [] + + def add_message(self, nick, msg, date=None): + if not date: + date = datetime.now() + color = None + self.set_color_state(12) + if nick != self.own_nick and self.joined: # do the highlight thing + if self.own_nick in msg: + self.set_color_state(13) + color = 3 + else: + highlight_words = config.get('highlight_on', '').split(':') + for word in highlight_words: + if word.lower() in msg.lower() and word != '': + self.set_color_state(13) + color = 3 + if not msg: + logger.info('msg is None..., %s' % (nick)) + return + self.lines.append((date, nick.encode('utf-8'), + msg.encode('utf-8'), color)) + if self.joined: # log only NEW messages, not the history received on join + logger.message(self.name, nick.encode('utf-8'), msg.encode('utf-8')) + return color + + def add_info(self, info, date=None): + """ info, like join/quit/status messages""" + if not date: + date = datetime.now() + try: + self.lines.append((date, info.encode('utf-8'))) + return info.encode('utf-8') + except: + self.lines.append((date, info)) + return info + + def get_user_by_name(self, nick): + for user in self.users: + if user.nick == nick.encode('utf-8'): + return user + return None + + def set_color_state(self, color): + if self.color_state < color or color == 11: + self.color_state = color diff --git a/src/user.py b/src/user.py new file mode 100644 index 00000000..0dee9e42 --- /dev/null +++ b/src/user.py @@ -0,0 +1,39 @@ +#!/usr/bin/python +# -*- coding:utf-8 -*- +# +# 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 random import randrange +from config import config + +class User(object): + """ + keep trace of an user in a Room + """ + def __init__(self, nick, affiliation, show, status, role): + self.update(affiliation, show, status, role) + self.change_nick(nick) + self.color = randrange(2, 10) + + def update(self, affiliation, show, status, role): + self.affiliation = affiliation + self.show = show + self.status = status + self.role = role + + def change_nick(self, nick): + self.nick = nick.encode('utf-8') |