Python 基于 Socket 实现群聊

共 4141字,需浏览 9分钟

 ·

2020-12-29 11:52

↑ 关注 + 星标 ,每天学Python新技能

后台回复【大礼包】送你Python自学大礼包


来源:博客园-huny

https://www.cnblogs.com/huny/p/14051152.html

互联网时代,怎么能不懂点网络编程?套接字(Sockets)是双向通信信道的端点,本文详细介绍了使用Socket实现类似微信群聊的功能,快进来一起学习下吧!


1.前言


套接字(Sockets)是双向通信信道的端点。套接字可以在一个进程内,在同一机器上的进程之间,或者在不同主机的进程之间进行通信,主机可以是任何一台有连接互联网的机器。


套接字可以通过多种不同的通道类型实现:Unix域套接字,TCP,UDP等。套接字库提供了处理公共传输的特定类,以及一个用于处理其余部分的通用接口。


1.1 socket模块:


要创建套接字,必须使用套接字模块中的socket.socket()函数,该函数具有一般语法

s = socket.socket (socket_family, socket_type, protocol = 0)


1.2 常用方法



2. 示例1


2.1 服务器端

#sever.pyimport socket
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)host = socket.gethostname()port = 8088s.bind((host,port))try: while True: receive_data,addr = s.recvfrom(1024) print("来自服务器" + str(addr) + "的消息:") print(receive_data.decode('utf-8')) msg = input('please input send to msg:') s.sendto(msg.encode('utf-8'),addr)except: s.close()

2.2 客户端

#client.pyimport sockets = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)try:    while True:        host = socket.gethostname()        port = 8088        send_data = input('please input msg:')        s.sendto(send_data.encode('utf-8'),(host,port))        msg,addr = s.recvfrom(1024)        print("来自服务器" + str(addr) + "的消息:")        print(msg.decode('utf-8'))except:    s.close()

服务器示例



客户端示例



简易的UDP聊天实现了,下面我们来优化一下示例。


3.示例优化


服务端

#server.pyimport socketimport logging
def main(): s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) # 创建socket对象
addr = ('127.0.0.1', 9999) s.bind(addr) # 绑定地址和端口
logging.info('UDP Server on %s:%s...', addr[0], addr[1])
user = {} # 存放字典{addr:name} while True: try: data, addr = s.recvfrom(1024) # 等待接收客户端消息存放在2个变量data和addr里 if not addr in user: # 如果addr不在user字典里则执行以下代码 for address in user: # 从user遍历数据出来address s.sendto(data + ' 进入聊天室...'.encode('utf-8'), address) # 发送user字典的data和address到客户端 user[addr] = data.decode('utf-8') # 接收的消息解码成utf-8并存在字典user里,键名定义为addr continue # 如果addr在user字典里,跳过本次循环
if 'EXIT'.lower() in data.decode('utf-8'):#如果EXIT在发送的data里 name = user[addr] #user字典addr键对应的值赋值给变量name user.pop(addr) #删除user里的addr for address in user: #从user取出address s.sendto((name + ' 离开了聊天室...').encode(), address) #发送name和address到客户端 else: print('"%s" from %s:%s' %(data.decode('utf-8'), addr[0], addr[1])) for address in user: #从user遍历出address if address != addr: #address不等于addr时间执行下面的代码 s.sendto(data, address) #发送data和address到客户端
except ConnectionResetError: logging.warning('Someone left unexcept.')
if __name__ == '__main__': main()

客户端

#clinet.pyimport socketimport threading
def recv(sock, addr): ''' 一个UDP连接在接收消息前必须要让系统知道所占端口 也就是需要send一次,否则win下会报错 ''' sock.sendto(name.encode('utf-8'), addr) while True: data = sock.recv(1024) print(data.decode('utf-8'))

def send(sock, addr): ''' 发送数据的方法 参数: sock:定义一个实例化socket对象 server:传递的服务器IP和端口 ''' while True: string = input('') message = name + ' : ' + string data = message.encode('utf-8') sock.sendto(data, addr) if string.lower() == 'EXIT'.lower(): break
def main(): ''' 主函数执行方法,通过多线程来实现多个客户端之间的通信 ''' s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) server = ('127.0.0.1', 9999) tr = threading.Thread(target=recv, args=(s, server), daemon=True) ts = threading.Thread(target=send, args=(s, server)) tr.start() ts.start() ts.join() s.close()
if __name__ == '__main__': print("-----欢迎来到聊天室,退出聊天室请输入'EXIT(不分大小写)'-----") name = input('请输入你的名称:') print('-----------------%s------------------' % name) main()

支持多人的简易聊天室示例,多个客户端通过一个服务器进行之间通信。







(完)

推荐阅读

  1. 推荐一个新手学习Python数据分析实战项目

  2. 华为员工违规领夜宵被发现后,被罚冻薪降考评,终身不得领夜宵,HR:哪个员工上班健身,定性考勤造假

  3. 小姐姐盗用他人设计作品找工作,入职京东...


扫码回复「大礼包」后获取大礼


码加我微信备注「三剑客」送你上图三本电子书

浏览 42
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报