Python Socket 编程——聊天室示例程序
上一篇 我们学习了简单的 Python TCP Socket 编程,通过分别写服务端和客户端的代码了解基本的 Python Socket 编程模型。本文再通过一个例子来加强一下对 Socket 编程的理解。
聊天室程序需求
我们要实现的是简单的聊天室的例子,就是允许多个人同时一起聊天,每个人发送的消息所有人都能接收到,类似于 QQ 群的功能,而不是点对点的 QQ 好友之间的聊天。如下图:
图来自:http://www.ibm.com/developerworks/linux/tutorials/l-pysocks/
我们要实现的有两部分:
- Chat Server:聊天服务器,负责与用户建立 Socket 连接,并将某个用户发送的消息广播到所有在线的用户。
- Telnet Client:用户聊天客户端,可以输入聊天的内容并发送,同时可以显示其他用户的消息记录。
同样,我们的消息通信采用 TCP 连接保证可靠性。在分别对服务端和客户端进行程序设计之前,首先要学习一下 Python 中实现异步 I/O 的一个函数 —— select
。
Python 异步 I/O
Python 在 select
模块中提供了异步 I/O(Asynchronous I/O),这与 Linux 下的 select 机制相似,但进行一些简化。我首先介绍一下 select
,然后告诉你在 Python 中如何使用它。
前面文章使用多线程来并行处理多路 socket I/O,这里介绍的select
方法允许你响应不同 socket 的多个事件以及其它不同事件。例如你可以让 select
在某个 socket 有数据到达时,或者当某个 socket 可以写数据时,又或者是当某个 socket 发生错误时通知你,好处是你可以同时响应很多 socket 的多个事件。
Linux 下 C 语言的 select
使用到位图来表示我们要关注哪些文件描述符的事件,Python 中使用 list 来表示我们监控的文件描述符,当有事件到达时,返回的也是文件描述符的 list,表示这些文件有事件到达。下面的简单程序是表示等待从标准输入中获得输入:
rlist, wlist, elist = select.select( [sys.stdin], [], [] )print sys.stdin.read()
select
方法的三个参数都是 list 类型,分别代表读事件、写事件、错误事件,同样方法返回值也是三个 list,包含的是哪些事件(读、写、异常)满足了。上面的例子,由于参数只有一个事件 sys.stdin
,表示只关心标准输入事件,因此当 select
返回时 rlist 只会是 [sys.stdin]
,表示可以从 stdin 中读入数据了,我们使用 read
方法来读入数据。
当然 select
对于 socket 描述符也是有效的,下面的一个例子是创建了两个 socket 客户端连接到远程服务器,select
用来监控哪个 socket 有数据到达:
import socket
import selectsock1 = socket.socket( socket.AF_INET, socket.SOCK_STREAM )
sock2 = socket.socket( socket.AF_INET, socket.SOCK_STREAM )sock1.connect( ('192.168.1.1', 25) )
sock2.connect( ('192.168.1.1', 25) )while 1:# Await a read eventrlist, wlist, elist = select.select( [sock1, sock2], [], [], 5 )# Test for timeoutif [rlist, wlist, elist] == [ [], [], [] ]:print "Five seconds elapsed.\n"else:# Loop through each socket in rlist, read and print the available datafor sock in rlist:print sock.recv( 100 )
好了,有了上面的基础,我们就可以来设计聊天室的服务器和客户端了。
聊天室服务器
聊天室服务器主要完成下面两件事:
- 接收多个客户端的连接
- 从每个客户端读入消息病广播到其它连接的客户端
我们定义一个 list 型变量 CONNECTION_LIST
表示监听多个 socket 事件的可读事件,那么利用上面介绍的我们的服务器使用 select
来处理多路复用 I/O 的代码如下:
# Get the list sockets which are ready to be read through select
read_sockets,write_sockets,error_sockets = select.select(CONNECTION_LIST,[],[])
当 select
返回时,说明在 read_sockets
上有可读的数据,这里又分为两种情况:
- 如果是主 socket(即服务器开始创建的 socket,一直处于监听状态)有数据可读,表示有新的连接请求可以接收,此时需要调用
accept
函数来接收新的客户端连接,并将其连接信息广播到其它客户端。 - 如果是其它 sockets(即与客户端已经建立连接的 sockets)有数据可读,那么表示客户端发送消息到服务器端,使用
recv
函数读消息,并将消息转发到其它所有连接的客户端。
上面两种情况到涉及到广播消息的过程,广播也就是将从某个 socket 获得的消息通过 CONNECTION_LIST
的每个 socket (除了自身和主 socket)一个个发送出去:
def broadcast_data (sock, message):#Do not send the message to master socket and the client who has send us the messagefor socket in CONNECTION_LIST:if socket != server_socket and socket != sock :try :socket.send(message)except :# broken socket connection may be, chat client pressed ctrl+c for examplesocket.close()CONNECTION_LIST.remove(socket)
如果发送失败,我们假设某个客户端已经断开了连接,关闭该 socket 病将其从连接列表中删除。
完整的聊天室服务器源代码如下:
# Tcp Chat serverimport socket, select#Function to broadcast chat messages to all connected clients
def broadcast_data (sock, message):#Do not send the message to master socket and the client who has send us the messagefor socket in CONNECTION_LIST:if socket != server_socket and socket != sock :try :socket.send(message)except :# broken socket connection may be, chat client pressed ctrl+c for examplesocket.close()CONNECTION_LIST.remove(socket)if __name__ == "__main__":# List to keep track of socket descriptorsCONNECTION_LIST = []RECV_BUFFER = 4096 # Advisable to keep it as an exponent of 2PORT = 5000server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)# this has no effect, why ?server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)server_socket.bind(("0.0.0.0", PORT))server_socket.listen(10)# Add server socket to the list of readable connectionsCONNECTION_LIST.append(server_socket)print "Chat server started on port " + str(PORT)while 1:# Get the list sockets which are ready to be read through selectread_sockets,write_sockets,error_sockets = select.select(CONNECTION_LIST,[],[])for sock in read_sockets:#New connectionif sock == server_socket:# Handle the case in which there is a new connection recieved through server_socketsockfd, addr = server_socket.accept()CONNECTION_LIST.append(sockfd)print "Client (%s, %s) connected" % addrbroadcast_data(sockfd, "[%s:%s] entered room\n" % addr)#Some incoming message from a clientelse:# Data recieved from client, process ittry:#In Windows, sometimes when a TCP program closes abruptly,# a "Connection reset by peer" exception will be throwndata = sock.recv(RECV_BUFFER)if data:broadcast_data(sock, "\r" + '<' + str(sock.getpeername()) + '> ' + data) except:broadcast_data(sock, "Client (%s, %s) is offline" % addr)print "Client (%s, %s) is offline" % addrsock.close()CONNECTION_LIST.remove(sock)continueserver_socket.close()
在控制台下运行该程序:
$ python chat_server.py
Chat server started on port 5000
聊天室客户端
我们写一个客户端程序可以连接到上面的服务器,完成发送消息和接收消息的过程。主要做下面两件事:
- 监听服务器是否有消息发送过来
- 检查用户的输入,如果用户输入某条消息,需要发送到服务器
这里有两个 I/O 事件需要监听:连接到服务器的 socket 和标准输入,同样我们可以使用 select
来完成:
rlist = [sys.stdin, s]# Get the list sockets which are readable
read_list, write_list, error_list = select.select(rlist , [], [])
那逻辑就很简单了,如果是 sys.stdin
有数据可读,表示用户从控制台输入数据并按下回车,那么就从标准输入读数据,并发送到服务器;如果是与服务器连接的 socket 有数据可读,表示服务器发送消息给该客户端,那么就从 socket 接收数据。加上一些提示信息及异常处理的完整客户端代码如下:
# telnet program example
import socket, select, string, sysdef prompt() :sys.stdout.write('<You> ')sys.stdout.flush()#main function
if __name__ == "__main__":if(len(sys.argv) < 3) :print 'Usage : python telnet.py hostname port'sys.exit()host = sys.argv[1]port = int(sys.argv[2])s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)s.settimeout(2)# connect to remote hosttry :s.connect((host, port))except :print 'Unable to connect'sys.exit()print 'Connected to remote host. Start sending messages'prompt()while 1:rlist = [sys.stdin, s]# Get the list sockets which are readableread_list, write_list, error_list = select.select(rlist , [], [])for sock in read_list:#incoming message from remote serverif sock == s:data = sock.recv(4096)if not data :print '\nDisconnected from chat server'sys.exit()else :#print datasys.stdout.write(data)prompt()#user entered a messageelse :msg = sys.stdin.readline()s.send(msg)prompt()
可以在多个终端下运行该代码:
$ python telnet.py localhost 5000
Connected to remote host. Start sending messages
<You> hello
<You> I am fine
<('127.0.0.1', 38378)> ok good
<You>
在另一个终端显示的信息:
<You> [127.0.0.1:39339] entered room
<('127.0.0.1', 39339)> hello
<('127.0.0.1', 39339)> I am fine
<You> ok good
总结
上面的代码注意两点:
- 聊天室客户端代码不能在 windows 下运行,因为代码使用
select
同时监听 socket 和输入流,在 Windows 下select
函数是由 WinSock 库提供,不能处理不是由 WinSock 定义的文件描述符。 - 客户端代码还有个缺陷是,当某个客户端在输入消息但还未发送出去时,服务器也发送消息过来,这样会冲刷掉客户端正在输入的消息。这目前来看没办法解决的,唯一的解决方法是使用像 ncurses 终端库使用户输入和输出独立开,或者写一个 GUI 的程序。
那么本文通过一个聊天室的范例进一步学习了 Python 下 Socket 编程。
参考文献
- Python socket – chat server and client with code example
- Sockets programming in Python
转载于:https://www.cnblogs.com/hazir/p/python_chat_room.html
Python Socket 编程——聊天室示例程序相关推荐
- Linux网络编程--聊天室客户端程序
聊天室客户端程序 #define _GNU_SOURCE 1 #include <sys/types.h> #include <sys/socket.h> #include & ...
- python socket 网络聊天室_Python基于Socket实现简单聊天室
本文实例为大家分享了Python基于Socket实现简单聊天室,供大家参考,具体内容如下 服务端 #!/usr/bin/env python # -*- coding: utf-8 -*- # @Ti ...
- java socket编程聊天室_Java Socket通信之聊天室功能
Java Socket通信之聊天室功能 发布时间:2020-10-17 14:36:00 来源:脚本之家 阅读:73 作者:LY_624 本文实例为大家分享了Java Socket聊天室功能的具体代码 ...
- python socket编程 实现简单p2p聊天程序
转载自:https://www.cnblogs.com/wuxie1989/p/7204887.html python socket编程 实现简单p2p聊天程序 目标是写一个python的p2p聊天的 ...
- python多人聊天室_Python基于Socket实现简易多人聊天室
前言 套接字(Sockets)是双向通信信道的端点. 套接字可以在一个进程内,在同一机器上的进程之间,或者在不同主机的进程之间进行通信,主机可以是任何一台有连接互联网的机器. 套接字可以通过多种不同的 ...
- 【java毕业设计】基于java+原生Sevlet+socket的聊天室系统设计与实现(毕业论文+程序源码)——聊天室系统
基于java+原生Sevlet+socket的聊天室系统设计与实现(毕业论文+程序源码) 大家好,今天给大家介绍基于java+原生Sevlet+socket的聊天室系统设计与实现,文章末尾附有本毕业设 ...
- 【python】基于Socket的聊天室Python开发
基于Socket的聊天室Python开发 一.Socket简述 二.创建服务端Server 2.1 创建服务端初始化 2.2 监听客户端连接 2.3 处理客户端消息 三.创建客户端Client 3.1 ...
- 基于python实现的聊天室(客户端:一)窗口设置
前言 就是自己想用python做一个聊天室,然后看看socket库,websocket库,有点底层,然后也会用到协程的东西,不是很明白,一时间也不知道怎么写,因为前面使用过了flask-socketi ...
- Android 基于Socket的聊天室
原文地址为: Android 基于Socket的聊天室 Socket是TCP/IP协议上的一种通信,在通信的两端各建立一个Socket,从而在通信的两端之间形成网络虚拟链路.一旦建立了虚拟的网络链路, ...
最新文章
- 1.8 centos7 的PATH、cp/mv/文档查看命令介绍
- 关闭计算机后自动开机的解决方法
- 如何替换 Ubuntu 11.10 登录屏幕背景和logo
- 【转】Asp.NetMve移除HTTP Header中服務器信息Server、X-AspNet-Version、X-AspNetMvc-Version、X-Powered-By:ASP.NET...
- Jlink commander、MCU GD32启动方式、Bootloader
- asp.mvc 4项目发布文件目录结构_Spring Boot项目搭建与启动
- android中momery检测,Android性能优化第(二)篇---Memory Monitor检测内存泄露
- 某单位配置GRE_×××命令
- 校赛热身 Problem C. Sometimes Naive (状压dp)
- mysql只update不做修改_105岁老人抽烟喝酒不运动,长寿秘诀:只做4件事,不做1件事...
- 支持向量机的前世与今生
- Delphi- repeate
- 谁有好的反P2p终结者的软件啊!
- 黑客防线 2007~2012 NPM、PYPI、DockerHub 备份
- Windows 11 22H2 中文版、英文版 (x64、ARM64) 下载 (updated Jan 2023)
- Tsunami: A Learned Multi-dimensional Index for Correlated Data and SkewedWorkloads(VLDB21)
- 教你十步实现内网穿透
- 关于720vr 全景平台 全景图片制作系统 仿720云需要注意的坑 诚意贴
- 【软考-中级】系统集成项目管理工程师-【1信息化知识】
- 中小企业筹资新法:应收账款也可质押融资
热门文章
- Linux 查看命令路径 以及相关信息
- 左耳朵耗子:程序员如何把控自己的职业?
- 关于 equals 和 hashCode,看这一篇真的够了!
- 最近学习了 HTTP 连接池
- 两个线程如何交替执行,一个输出偶数一个输出奇数?
- spring security CSRF 问题 Invalid CSRF Token 'null' was found on ......
- mysql高性能学习笔记03_MySQL高性能学习笔记
- 心理正常与异常的区分_心理正常与心理异常的区别是什么呢
- python第二大奇数_python-2.7 – matplotlib,包含奇数个子图
- 成功解决The NVIDIA driver on your system is too old (found version 9010). Please update your GPU driver