python练习四—简单的聊天软件

时间:2023-03-09 00:39:00
python练习四—简单的聊天软件

python最强大的是什么?库支持!!有了强大的库支持,一个简单的聊天软件实现就更简单了,本项目思路如下

# 项目思路
1. 服务器的工作
* 初始化服务器
* 新建一个聊天房间
* 维护一个已链接用户的会话列表
* 维护一个已登录用户的字典,用户名和会话
* 监听端口,接受会话,并启动一个ChatSession处理 2. 会话线程
* 初始化一个接收数据缓冲区
* 处理用户输入的命令,并提醒用户先登录(也就是说目前指处理登陆命令)
* 将用户输入的数据加入缓冲区
* 用户一次输入结束后,将缓冲区数据发送,并清空缓冲区 3. 命令分发
* 对用的输入进行分割,并将对应的命令分发到相应的处理函数,如果没有则采用默认处理函数 4. 房间,继承自命令分发类,即:具有处理命令能力的房间 5. LoginRoom,继承自Room,处理登陆相关的命令
* 加入房间,有用户进入房间,欢迎该用户
* 登录命令,登陆成功之后进入房间,将该session加入队列 6. ChatRoom,继承自Room,处理发言、查看所有用户等命令 7. LogoutRoom,继承自Room,处理用户登出,相当于用户输入登出命令之后进入一个房间,
然后抛出异常通知ChatSession关闭线程

代码如下:

#! /usr/bin/env python
# -*- coding=utf-8 -*- from asyncore import dispatcher
import socket
import asyncore
from asynchat import async_chat # 服务器监听端口号
PORT = 5005
# 服务器名称
NAME = 'QQ' class EndException(Exception):
pass class CommandHandler:
'''
用来处理用户输入简单的命令
''' def unknown(self, session, cmd):
'''处理未知命令'''
session.push('Unknown command %s\r\n' %cmd) def handler(self, session, line):
'''
处理命令(命令路由),根据用户输入分析储命令,调用对应的处理程序
'''
if not line.strip():
# 如果输入为空,返回,不做处理
return
# 切分命令
print line + '----------'
parts = line.split(' ', 1)
cmd = parts[0]
try:
line = parts[1]
except IndexError:
line = ''
method = getattr(self, 'do_' + cmd, None)
try:
method(session, line)
except TypeError:
# 如果该方法不存在,说明不能处理该命令,未知命令
self.unknown(session, cmd) class Room(CommandHandler):
'''
房间类,用来处理具体情况的命令
''' def __init__(self, server):
'房间初始化的时候,需要一个服务器对象用来获得服务器信息,初始化房间内的人员' self.server = server
self.sessions = [] def add(self, session):
'''
用户加入该房间
'''
self.sessions.append(session) def remove(self, session):
'用户退出房间'
self.sessions.remove(session) def broadcast(self, line):
'用户在房间中发言'
for session in self.sessions:
session.push(line) def do_logout(self, sessioni, line):
'用户退出房间'
raise EndException class LoginRoom(Room):
'''
用户登录房间,处理用户登录房间的一些操作
''' def add(self, session):
'扩展父类方法,在加入房间之后,给用户发送欢迎语'
Room.add(self, session)
session.push('welcome to %s \r\n' % self.server.name) def unknown(self, session, cmd):
'处理未知命令,提示用户输入登录命令'
session.push('please log in \r "Use login <nick>" \r\n') def do_login(self, session, line):
'处理用户登录命令'
name = line.strip()
if not name:
# 如果没有输入名称,提示用户输入
session.push('Please input a name')
elif name in self.server.users:
# 如果名称已被占用
session.push('the % is taken \r\n' %name)
session.push('Please try again')
else:
# 如果正常则让用户进入房间
session.name = name
session.enter(self.server.main_room) class ChatRoom(Room):
'''
聊天房间,处理用户在房间中聊天的命令
''' def add(self, session):
'用户进入聊天房间的时候,通知其他用户'
Room.add(self, session)
self.broadcast('%s has enter in the room \r\n' %session.name)
self.server.users[session.name] = session def remove(self, session):
'提醒其他用户,xx离开房间'
Room.remove(self, session)
self.broadcast('%s has left the room \r\n' %session.name) def do_say(self, session, line):
'用户发言'
self.broadcast(session.name + ':' + line + '\r\n') def do_look(self, session, line):
'查看房间中所有用户'
self.broadcast('the following are in the room: \r\n')
for user in self.sessions:
self.broadcast(user.name + '\r\n') class LogoutRoom(Room):
'''
用户退出房间,所需要的命令
''' def do_logout(self, session, line):
try:
del self.server.users[session.name]
except KeyError:
pass class ChatSession(async_chat):
'''
具体处理客户端链接, 接受客户端发送来的信息,
广播给所有在线的客户端,关闭连接,
服务器保留所有连接到服务器的会话
''' def __init__(self, server, sock):
''
async_chat.__init__(self, sock)
self.server = server
self.set_terminator('\r\n')
self.data = []
self.name = None # 如果有新用户连接,则先要求登陆,登陆成功以后进入房间
self.enter(LoginRoom(self.server)) def enter(self, room):
try:
cur = self.room
except AttributeError:
pass
else:
cur.remove(self)
self.room = room
room.add(self) def collect_incoming_data(self, data):
self.data.append(data) def found_terminator(self):
line = "".join(self.data)
self.data = []
try:
self.room.handler(self, line)
except EndException:
self.handle_close() def handle_close(self):
async_chat.handle_close(self)
self.enter(LogoutRoom(self)) class ChatServer(dispatcher):
'''
服务器,负责接受客户端链接并分配给具体的程序(线程处理)
''' def __init__(self, port, name):
'初始化服务器,初始会话列表,初始化用户字典(用户名对应用户session),初始化服务器主房间'
dispatcher.__init__(self)
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
self.set_reuse_addr()
self.bind(('', port))
self.listen(5)
self.sessions = []
self.users = {}
self.name = name
self.main_room = ChatRoom(self) def handle_accept(self):
conn, addr = self.accept()
print 'connection attemp from ', addr[0]
ChatSession(self, conn) s = ChatServer(PORT, NAME)
try:
asyncore.loop()
except KeyboardInterrupt:
pass

其实从第一个练习开始我就开始意识到,书中的引导是逐步的——使用的是迭代式的开发,一开始知识一个简单的例子视线了基本的功能,然后基于简单的列子将代码模块化,逐渐加功能,我也是试着这样做的,但是鉴于篇幅,其他代码就没有贴上来,在完整的代码可以下载

再进行本练习中,有一下基础知识得到巩固:

# 判断正在运行的模块是直接运行还是被import执行的,即:
# 直接运行的话__name='__main__'
# 如果是import的话就不是__main__
if __name__ == '__main__': # 变量作用域
data = 123
* 在函数内部申明,作用域就是函数内部,
* 在类内部申明就是类变量,访问的时候直接data
* 在类内部的函数内部申明如:self.data=124,调用的时候只能是self.data,如果使用data,默认查找的是全局变量
* 如果不适用self指明,python默认调用的全局变量
* python变量可以重名,在同一作用于内如果已经定义过,只是进赋值 # 在子类方法内部调用父类方法(如:Room父类,ChatRoom子类)
Room.__init__()

完整代码

http://pan.baidu.com/s/1nu5KE4T