• 基于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()
    

      

  • 相关阅读:
    网络-基础知识
    手机开发-安卓手机的更新换代
    C-基础
    手机开发-IOS
    前端- html 和css
    jmeter接口测试-文件下载
    JDBC接口
    jmeter接口测试实例7-关联
    jmeter接口测试实例6-注册(参数化)
    jmeter接口测试实例5-文件上传
  • 原文地址:https://www.cnblogs.com/huangxiaohen/p/4236191.html
Copyright © 2020-2023  润新知