`
实验楼
  • 浏览: 11513 次
  • 来自: 成都
社区版块
存档分类
最新评论

如何用Python编写一个聊天室

阅读更多

本课程来自实验楼Python聊天室,若需转载请注明原文出处。

一、课程介绍

1.简介

本次项目课是实现简单聊天室程序的服务器端和客户端。

2.知识点

服务器端涉及到asyncore、asynchat和socket这几个模块,客户端用到了telnetlib、wx、time和thread这几个模块。

3.所需环境

本次课中编写客户端需要用到wxPython,它是一个GUI工具包,请先使用下面的命令安装:

$ sudo apt-get install python-wxtools

密码为shiyanlou

4.项目效果截图

登录窗口

聊天窗口

5.源代码下载

git clone https://github.com/shiyanlou/pythonchat.git

说明:如果你不理解上述代码的下载方式或者下载后在环境中找不到代码,可以点击查看这里

二、项目实战(服务器端)

1.服务器类

首先需要一个聊天服务器,这里继承asyncore的dispatcher类来实现,代码如下

class ChatServer(dispatcher):
    """
    聊天服务器
    """

    def __init__(self, port):
        dispatcher.__init__(self)
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.set_reuse_addr()
        self.bind(('', port))
        self.listen(5)
        self.users = {}
        self.main_room = ChatRoom(self)

    def handle_accept(self):
        conn, addr = self.accept()
        ChatSession(self, conn)

2.会话类

有了服务器类还需要能维护每个用户的连接会话,这里继承asynchat的async_chat类来实现,代码如下:

class ChatSession(async_chat):
    """
    负责和单用户通信
    """

    def __init__(self, server, sock):
        async_chat.__init__(self, sock)
        self.server = server
        self.set_terminator('\n')
        self.data = []
        self.name = None
        self.enter(LoginRoom(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.handle(self, line)
        except EndSession:
            self.handle_close()

    def handle_close(self):
        async_chat.handle_close(self)
        self.enter(LogoutRoom(self.server))

3.命令解释器

现在就需要一个命令解释器能够解释用户的命令,例如登录、查询在线用户和发消息等,代码如下:

class CommandHandler:
    """
    命令处理类
    """

    def unknown(self, session, cmd):
        '响应未知命令'
        session.push('Unknown command: %s\n' % cmd)

    def handle(self, session, line):
        '命令处理'
        if not line.strip():
            return
        parts = line.split(' ', 1)
        cmd = parts[0]
        try:
            line = parts[1].strip()
        except IndexError:
            line = ''
        meth = getattr(self, 'do_' + cmd, None)
        try:
            meth(session, line)
        except TypeError:
            self.unknown(session, cmd)

4.房间

接下来就需要实现聊天室的房间了,这里我们定义了三种房间,分别是用户刚登录时的房间、聊天的房间和退出登录的房间,这三种房间都有一个公共的父类,代码如下:

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, session, line):
        '退出房间'
        raise EndSession

class LoginRoom(Room):
    """
    刚登录的用户的房间
    """

    def add(self, session):
        '用户连接成功的回应'
        Room.add(self, session)
        session.push('Connect Success')

    def do_login(self, session, line):
        '登录命令处理'
        name = line.strip()
        if not name:
            session.push('UserName Empty')
        elif name in self.server.users:
            session.push('UserName Exist')
        else:
            session.name = name
            session.enter(self.server.main_room)

class ChatRoom(Room):
    """
    聊天用的房间
    """

    def add(self, session):
        '广播新用户进入'
        session.push('Login Success')
        self.broadcast(session.name + ' has entered the room.\n')
        self.server.users[session.name] = session
        Room.add(self, session)

    def remove(self, session):
        '广播用户离开'
        Room.remove(self, session)
        self.broadcast(session.name + ' has left the room.\n')

    def do_say(self, session, line):
        '客户端发送消息'
        self.broadcast(session.name + ': ' + line + '\n')

    def do_look(self, session, line):
        '查看在线用户'
        session.push('Online Users:\n')
        for other in self.sessions:
            session.push(other.name + '\n')

class LogoutRoom(Room):
    """
    用户退出时的房间
    """

    def add(self, session):
        '从服务器中移除'
        try:
            del self.server.users[session.name]
        except KeyError:
            pass

5.服务器端完整代码

#!/usr/bin/python
# encoding: utf-8

from asyncore import dispatcher
from asynchat import async_chat
import socket, asyncore

PORT = 6666 #端口

class EndSession(Exception):
    """
    自定义会话结束时的异常
    """
    pass

class CommandHandler:
    """
    命令处理类
    """

    def unknown(self, session, cmd):
        '响应未知命令'
        session.push('Unknown command: %s\n' % cmd)

    def handle(self, session, line):
        '命令处理'
        if not line.strip():
            return
        parts = line.split(' ', 1)
        cmd = parts[0]
        try:
            line = parts[1].strip()
        except IndexError:
            line = ''
        meth = getattr(self, 'do_' + cmd, None)
        try:
            meth(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, session, line):
        '退出房间'
        raise EndSession

class LoginRoom(Room):
    """
    刚登录的用户的房间
    """

    def add(self, session):
        '用户连接成功的回应'
        Room.add(self, session)
        session.push('Connect Success')

    def do_login(self, session, line):
        '登录命令处理'
        name = line.strip()
        if not name:
            session.push('UserName Empty')
        elif name in self.server.users:
            session.push('UserName Exist')
        else:
            session.name = name
            session.enter(self.server.main_room)

class ChatRoom(Room):
    """
    聊天用的房间
    """

    def add(self, session):
        '广播新用户进入'
        session.push('Login Success')
        self.broadcast(session.name + ' has entered the room.\n')
        self.server.users[session.name] = session
        Room.add(self, session)

    def remove(self, session):
        '广播用户离开'
        Room.remove(self, session)
        self.broadcast(session.name + ' has left the room.\n')

    def do_say(self, session, line):
        '客户端发送消息'
        self.broadcast(session.name + ': ' + line + '\n')

    def do_look(self, session, line):
        '查看在线用户'
        session.push('Online Users:\n')
        for other in self.sessions:
            session.push(other.name + '\n')

class LogoutRoom(Room):
    """
    用户退出时的房间
    """

    def add(self, session):
        '从服务器中移除'
        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('\n')
        self.data = []
        self.name = None
        self.enter(LoginRoom(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.handle(self, line)
        except EndSession:
            self.handle_close()

    def handle_close(self):
        async_chat.handle_close(self)
        self.enter(LogoutRoom(self.server))

class ChatServer(dispatcher):
    """
    聊天服务器
    """

    def __init__(self, port):
        dispatcher.__init__(self)
        self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
        self.set_reuse_addr()
        self.bind(('', port))
        self.listen(5)
        self.users = {}
        self.main_room = ChatRoom(self)

    def handle_accept(self):
        conn, addr = self.accept()
        ChatSession(self, conn)

if __name__ == '__main__':
    s = ChatServer(PORT)
    try:
        asyncore.loop()
    except KeyboardInterrupt:
        print

三、项目实战(客户端)

完成了服务器端后,就需要实现客户端了,这里客户端连接服务器使用了telnetlib模块。

1.登录窗口

这里的图形界面包选择了wxPython,前面有安装说明,登录窗口通过继承wx.Frame类来实现,代码如下:

class LoginFrame(wx.Frame):
    """
    登录窗口
    """

    def __init__(self, parent, id, title, size):
        '初始化,添加控件并绑定事件'
        wx.Frame.__init__(self, parent, id, title)
        self.SetSize(size)
        self.Center()
        self.serverAddressLabel = wx.StaticText(self, label = "Server Address", pos = (10, 50), size = (120, 25))
        self.userNameLabel = wx.StaticText(self, label = "UserName", pos = (40, 100), size = (120, 25))
        self.serverAddress = wx.TextCtrl(self, pos = (120, 47), size = (150, 25))
        self.userName = wx.TextCtrl(self, pos = (120, 97), size = (150, 25))
        self.loginButton = wx.Button(self, label = 'Login', pos = (80, 145), size = (130, 30))
        self.loginButton.Bind(wx.EVT_BUTTON, self.login)
        self.Show()

    def login(self, event):
        '登录处理'
        try:
            serverAddress = self.serverAddress.GetLineText(0).split(':')
            con.open(serverAddress[0], port = int(serverAddress[1]), timeout = 10)
            response = con.read_some()
            if response != 'Connect Success':
                self.showDialog('Error', 'Connect Fail!', (95, 20))
                return
            con.write('login ' + str(self.userName.GetLineText(0)) + '\n')
            response = con.read_some()
            if response == 'UserName Empty':
                self.showDialog('Error', 'UserName Empty!', (135, 20))
            elif response == 'UserName Exist':
                self.showDialog('Error', 'UserName Exist!', (135, 20))
            else:
                self.Close()
                ChatFrame(None, -2, title = 'ShiYanLou Chat Client', size = (500, 350))
        except Exception:
            self.showDialog('Error', 'Connect Fail!', (95, 20))

    def showDialog(self, title, content, size):
        '显示错误信息对话框'
        dialog = wx.Dialog(self, title = title, size = size)
        dialog.Center()
        wx.StaticText(dialog, label = content)
        dialog.ShowModal()

2.聊天窗口

聊天窗口中最主要的就是向服务器发消息并接受服务器的消息,这里通过子线程来接受,代码如下:

class ChatFrame(wx.Frame):
    """
    聊天窗口
    """

    def __init__(self, parent, id, title, size):
        '初始化,添加控件并绑定事件'
        wx.Frame.__init__(self, parent, id, title)
        self.SetSize(size)
        self.Center()
        self.chatFrame = wx.TextCtrl(self, pos = (5, 5), size = (490, 310), style = wx.TE_MULTILINE | wx.TE_READONLY)
        self.message = wx.TextCtrl(self, pos = (5, 320), size = (300, 25))
        self.sendButton = wx.Button(self, label = "Send", pos = (310, 320), size = (58, 25))
        self.usersButton = wx.Button(self, label = "Users", pos = (373, 320), size = (58, 25))
        self.closeButton = wx.Button(self, label = "Close", pos = (436, 320), size = (58, 25))
        self.sendButton.Bind(wx.EVT_BUTTON, self.send)
        self.usersButton.Bind(wx.EVT_BUTTON, self.lookUsers)
        self.closeButton.Bind(wx.EVT_BUTTON, self.close)
        thread.start_new_thread(self.receive, ())
        self.Show()

    def send(self, event):
        '发送消息'
        message = str(self.message.GetLineText(0)).strip()
        if message != '':
            con.write('say ' + message + '\n')
            self.message.Clear()

    def lookUsers(self, event):
        '查看当前在线用户'
        con.write('look\n')

    def close(self, event):
        '关闭窗口'
        con.write('logout\n')
        con.close()
        self.Close()

    def receive(self):
        '接受服务器的消息'
        while True:
            sleep(0.6)
            result = con.read_very_eager()
            if result != '':
                self.chatFrame.AppendText(result)

3.客户端完整代码

#!/usr/bin/python
# encoding: utf-8

import wx
import telnetlib
from time import sleep
import thread

class LoginFrame(wx.Frame):
    """
    登录窗口
    """

    def __init__(self, parent, id, title, size):
        '初始化,添加控件并绑定事件'
        wx.Frame.__init__(self, parent, id, title)
        self.SetSize(size)
        self.Center()
        self.serverAddressLabel = wx.StaticText(self, label = "Server Address", pos = (10, 50), size = (120, 25))
        self.userNameLabel = wx.StaticText(self, label = "UserName", pos = (40, 100), size = (120, 25))
        self.serverAddress = wx.TextCtrl(self, pos = (120, 47), size = (150, 25))
        self.userName = wx.TextCtrl(self, pos = (120, 97), size = (150, 25))
        self.loginButton = wx.Button(self, label = 'Login', pos = (80, 145), size = (130, 30))
        self.loginButton.Bind(wx.EVT_BUTTON, self.login)
        self.Show()

    def login(self, event):
        '登录处理'
        try:
            serverAddress = self.serverAddress.GetLineText(0).split(':')
            con.open(serverAddress[0], port = int(serverAddress[1]), timeout = 10)
            response = con.read_some()
            if response != 'Connect Success':
                self.showDialog('Error', 'Connect Fail!', (95, 20))
                return
            con.write('login ' + str(self.userName.GetLineText(0)) + '\n')
            response = con.read_some()
            if response == 'UserName Empty':
                self.showDialog('Error', 'UserName Empty!', (135, 20))
            elif response == 'UserName Exist':
                self.showDialog('Error', 'UserName Exist!', (135, 20))
            else:
                self.Close()
                ChatFrame(None, -2, title = 'ShiYanLou Chat Client', size = (500, 350))
        except Exception:
            self.showDialog('Error', 'Connect Fail!', (95, 20))

    def showDialog(self, title, content, size):
        '显示错误信息对话框'
        dialog = wx.Dialog(self, title = title, size = size)
        dialog.Center()
        wx.StaticText(dialog, label = content)
        dialog.ShowModal()

class ChatFrame(wx.Frame):
    """
    聊天窗口
    """

    def __init__(self, parent, id, title, size):
        '初始化,添加控件并绑定事件'
        wx.Frame.__init__(self, parent, id, title)
        self.SetSize(size)
        self.Center()
        self.chatFrame = wx.TextCtrl(self, pos = (5, 5), size = (490, 310), style = wx.TE_MULTILINE | wx.TE_READONLY)
        self.message = wx.TextCtrl(self, pos = (5, 320), size = (300, 25))
        self.sendButton = wx.Button(self, label = "Send", pos = (310, 320), size = (58, 25))
        self.usersButton = wx.Button(self, label = "Users", pos = (373, 320), size = (58, 25))
        self.closeButton = wx.Button(self, label = "Close", pos = (436, 320), size = (58, 25))
        self.sendButton.Bind(wx.EVT_BUTTON, self.send)
        self.usersButton.Bind(wx.EVT_BUTTON, self.lookUsers)
        self.closeButton.Bind(wx.EVT_BUTTON, self.close)
        thread.start_new_thread(self.receive, ())
        self.Show()

    def send(self, event):
        '发送消息'
        message = str(self.message.GetLineText(0)).strip()
        if message != '':
            con.write('say ' + message + '\n')
            self.message.Clear()

    def lookUsers(self, event):
        '查看当前在线用户'
        con.write('look\n')

    def close(self, event):
        '关闭窗口'
        con.write('logout\n')
        con.close()
        self.Close()

    def receive(self):
        '接受服务器的消息'
        while True:
            sleep(0.6)
            result = con.read_very_eager()
            if result != '':
                self.chatFrame.AppendText(result)

'程序运行'
if __name__ == '__main__':
    app = wx.App()
    con = telnetlib.Telnet()
    LoginFrame(None, -1, title = "Login", size = (280, 200))
    app.MainLoop()

四、小结

最后就可以运行程序进行聊天了,注意需要先启动服务器再启动客户端。这个项目中使用了asyncore的dispatcher来实现服务器,asynchat的asyn_chat来维护用户的连接会话,用wxPython来实现图形界面,用telnetlib来连接服务器,在子线程中接受服务器发来的消息,由此一个简单的聊天室程序就完成了。

分享到:
评论

相关推荐

    python编写简易聊天室实现局域网内聊天功能

    根据提供的文件内容,本文将详细介绍如何使用Python编写一个简易的局域网内聊天室程序。局域网聊天室是指在同一局域网(Local Area Network, LAN)内的用户可以相互发送和接收消息的平台。在本实例中,主要使用了UDP...

    Python实现的多人聊天室源码,基于socket tcp通信,使用tkinter做客户端界面,含可执行文件

    在本项目中,我们探讨的是一个使用Python编程语言实现的多人聊天室,它基于socket TCP通信协议,并且利用Tkinter库构建了用户友好的客户端界面。这个聊天室系统允许多个用户同时在线交流,增强了实时互动性。以下是...

    基于python进行开发的聊天室(单聊、群聊、gui界面、自动回复)

    利用Socket进行编程,应用多线程进行多方的持续通信;可以进行私人、多人聊天,在登录和聊天室使用图形界面;与智能机器人聊天,通过读取csv文件...聊天室一侧可显示聊天室在线人数。 备注:使用pycharm直接打开即可

    python写的socket聊天室程序,包括客户端和服务端

    在这个场景中,我们有一个基于Python编写的socket聊天室程序,它包含了客户端(client.py)和服务端(server.py)。这个程序允许多个客户端同时连接到服务端,进行双向的信息交流。这里我们将深入探讨涉及的Python...

    python编写的在线聊天室,用于期末大作业.zip

    以上就是使用Python编写在线聊天室所需掌握的关键技术点。在实践中,还可以根据需求进行扩展,比如增加实时消息推送、文件传输、语音聊天等功能,进一步提升聊天室的用户体验。通过这个项目,不仅可以深入理解Python...

    python聊天室---pyqt5+socket+Thread聊天室

    本项目“python聊天室---pyqt5+socket+Thread聊天室”旨在教授如何利用Python的PyQt5库创建一个聊天室应用,并通过Socket进行网络通信,同时采用多线程(Thread)来实现后台任务的异步执行。以下将详细解析这个项目的...

    计算机网络的最终项目,一个用Python编写的聊天室,名为network pj chatroom.zip

    网鼎杯计算机网络的最终项目,一个用Python编写的聊天室,名为network pj chatroom.zip计算机网络的最终项目,一个用Python编写的聊天室,名为network pj chatroom.zip计算机网络的最终项目,一个用Python编写的聊天...

    python多人在线聊天室

    【Python 多人在线聊天室】是一个基于 IRC(Internet Relay Chat)协议的项目,旨在创建一个允许用户通过Python实现的在线交流平台。IRC 是一种早期的实时通讯协议,专为多用户聊天室设计,至今仍被许多技术社区广泛...

    python聊天室,python聊天室项目代码,Python

    在这个基于Python的聊天室项目中,我们将探讨如何使用Python的socket模块来创建一个多用户聊天环境。 一、Python socket基础知识 在Python中,socket模块是进行网络通信的基础,它提供了低级别的、原始的TCP/IP套接...

    Python 聊天室 程序 客户端 服务端

    总的来说,Python聊天室程序是一个综合性的学习项目,涵盖了网络编程、多线程(或多进程)、用户交互、文件操作等多个方面。通过实践这个项目,开发者不仅可以掌握Python的基本网络通信,还能提升对并发处理和用户...

    完整版 基于socket和tkinter的python网络聊天室程序.zip

    总的来说,这个基于socket和tkinter的Python网络聊天室程序展示了如何结合使用网络编程和GUI编程来创建一个功能齐全的应用。通过学习和理解这个程序,开发者可以进一步提升在网络编程和GUI设计方面的技能。

    多人聊天室python实现

    2. **wxPython包**:wxPython是一个用于创建图形用户界面(GUI)的库,它使开发者能够用Python编写出与平台无关的应用程序。在这个聊天室项目中,wxPython可能被用来设计和实现聊天窗口、输入框、发送按钮等元素。 3...

    基于Socket的聊天室Python实现(GUI版)

    资源内容:基于Socket的聊天室Python实现(GUI版) 使用/学习目标:了解 Socket 核心原理 特点:简单、带GUI方便使用 适用人群:想阅读Python源码的初学者,想自己搭建远程聊天室的站长们 可以参考本资源来搭建属于...

    Dos.rar_nine44k_python_python socket_python socket聊天_python聊天室

    总之,"Dos.rar_nine44k_python_python socket_python socket聊天_python聊天室"是一个关于使用Python socket模块创建多人聊天室的项目,它涵盖了网络编程、并发处理和用户交互等多个关键知识点,是学习Python网络...

    python聊天室

    Python聊天室是一个基于Python编程语言实现的网络通信应用,它结合了socket库来处理网络连接,同时使用图像界面(GUI)来提供用户友好的交互体验。在这个项目中,用户能够发送文字消息、表情等,模仿了类似QQ的聊天...

    基于python和qt实现的聊天室

    这个项目使用Python编程语言和Qt图形库来构建一个基本的聊天室,利用socket进行网络通信。这里我们将深入探讨相关知识点。 【Python】: Python是一种高级编程语言,以其简洁明了的语法和丰富的标准库而闻名。在本...

    基于Python与HTML5实现的websocket聊天室

    一个机遇WebSocket的聊天室,后端用Python编写。只要浏览器支持WebSocket即可。 功能: 查看在线人数 清屏 拖拽图片 修改背景颜色 修改前景颜色 发送远程图片 表情 特点: 是一个类似于linux操作界面的聊天室,...

Global site tag (gtag.js) - Google Analytics