基于Twisted的简单聊天室
实现了私聊，公聊。用户名密码存在redis。为了拓展，做了简单的模块。
#coding=utf8
from twisted.internet import reactor, defer
from twisted.protocols.basic import LineReceiver
from twisted.internet.protocol import Factory, Protocol, ServerFactory
import redis
class Server(ServerFactory):
def __init__(self):
self.users = {}
self.last_messages = {}
self.__db = dbServer
#def start(self):
#def stop(self):
def do_login(self, pro, args):
username, passwd = args
if self.__db.get(username) is not None:
if passwd != self.__db.get(username):
return "do_login", False, "login failed"
else:
user = User(username, passwd, pro, 1)
self.users[username] = user
pro.user = user
return "do_login", True, "login success"
else:
self.__db.set(username, passwd)
self.__db.save()
user = User(username, passwd, pro, 1)
self.users[username] = user
pro.user = user
return "do_login", True, "login success"
def checklogin(self, user):
return user.logined
def do_privatemsg(self, pro, args):
target, msg = args
user = pro.user
if self.checklogin(user):
self.last_messages[user] = msg
msg = '['+user.name+']' + msg
return "do_privatemsg", self.users[target], msg
def do_publicmsg(self, pro, args):
msg = args
user = pro.user
print user.name
if self.checklogin(user):
self.last_messages[user] = msg
msg = '['+user.name+']' + msg[0]
return "do_publicmsg", self.users.values(), msg
class User(object):
def __init__(self, name, passwd, pro, logined):
self.name = name
self.password = passwd
self.nick_name = '&&' + name + '&&'
self.proto = pro
self.logined = logined
class ChatProtocol(LineReceiver):
#line format: "func target_name *contents"
#example: "privatemsg hs how are you....."
#example: "publicmsg how are you....."
#example: "login hs password"
def __init__(self):
self.user = 0
self.pro = self
def connectionMade(self):
self.sendLine("please input: 'login,username,password'")
def lineReceived(self, line):
defer.maybeDeferred(self.parse, line).addCallback(self.processor).addCallback(self.ack)
def parse(self, line):
result = line.split(',')
func = result[0]
args = result[1:]
return func, args
def processor(self, args):
func, arg = args
_func = getattr(server, 'do_' + func)
if _func:
return _func(self.pro, arg)
else:
raise Exception("xxxxx")
def ack(self, args):
func, target, msg = args
if func == 'do_login':
if not target:
self.kickOut()
self.sendLine(msg)
elif func == 'do_privatemsg':
target.proto.sendLine(msg)
elif func == 'do_publicmsg':
for user in target:
if user.proto != self:
user.proto.sendLine(msg)
else:
pass
def kickOut(self):
self.sendLine('passwd is wrong')
self.transport.loseConnection()
class redisServer(object):
def __init__(self):
r = redis.Redis(host='localhost', port=6379, db=0)
self.db = r
def set(self, key, value):
self.db.set(key, value)
def save(self):
self.db.save()
def get(self, key):
return self.db.get(key)
dbServer = redisServer()
server = Server()
server.protocol = ChatProtocol
reactor.listenTCP(8008,server)
print 'server start!'
reactor.run()