"""
【1】 有人进入聊天室需要输入姓名,姓名不能重复
【2】 有人进入聊天室时,其他人会收到通知:xxx 进入了聊天室
【3】 一个人发消息,其他人会收到:xxx : xxxxxxxxxxx
【4】 有人退出聊天室,则其他人也会收到通知:xxx退出了聊天室
创建udp服务端
"""
from socket import *
from multiprocessing import Process
service_address = ('0.0.0.0', 10000)
user = {}
# 进入聊天室
def in_room(s, name, address):
if name in user or '管理' in name:
s.sendto("该用户名已存在".encode(), address)
return
else:
s.sendto(b"ok", address)
# 否则遍历字典,将进入消息发给在聊天室的user
for i in user:
msg = '\n欢迎%s进入聊天室' % name
s.sendto(msg.encode(), user[i])
user[name] = address
# print(user)
def do_chat(s, name, content):
msg = '\n%s : %s' % (name, content)
for i in user:
if name != i:
# 将一个用户说的话发送给其他用户
s.sendto(msg.encode(), user[i])
def do_quit(s, name):
del user[name]
msg = '\n%s退出聊天室' % name
for i in user:
# 通过遍历到的键获取其他用户的ip地址
s.sendto(msg.encode(), user[i])
def operate(udp_sock):
"""
将聊天操作作为子进程进行
:return:
"""
# 接收消息
while True:
data, address = udp_sock.recvfrom(1024 * 10)
# 预览一下收到的数据信息
print(data.decode())
# 判断消息有什么需求
info = data.decode().split(' ', 2)
if info[0] == 'L':
# ['L','name']
in_room(udp_sock, info[1], address)
elif info[0] == 'C':
# ['C','name','content']
do_chat(udp_sock, info[1], info[2])
elif info[0] == 'Q':
do_quit(udp_sock, info[1])
def main():
# 创建udp套接字
udp_sock = socket(AF_INET, SOCK_DGRAM)
# 声明ip地址和端口
udp_sock.bind(service_address)
# 创建子进程对象 执行聊天的所有操作
p = Process(target=operate, args=(udp_sock,))
p.start()
while True:
info = input("管理员消息:")
msg = 'C 管理员信息 ' + info
udp_sock.sendto(msg.encode(), service_address)
p.join()
if __name__ == '__main__':
main()
"""
【1】 有人进入聊天室需要输入姓名,姓名不能重复
【2】 有人进入聊天室时,其他人会收到通知:xxx 进入了聊天室
【3】 一个人发消息,其他人会收到:xxx : xxxxxxxxxxx
【4】 有人退出聊天室,则其他人也会收到通知:xxx退出了聊天室
创建udp客户端
"""
import sys
from socket import *
from multiprocessing import Process
# 服务器ip地址
service_address = ('127.0.0.1', 10000)
def in_room(s):
while True:
name = input("请输入用户名:")
if not name:
sys.exit()
# 自定义网络协议 'L'表示请求进入聊天室
msg = 'L ' + name
s.sendto(msg.encode(), service_address)
# 收到服务端的反馈
data, address = s.recvfrom(1024)
if data.decode() == 'ok':
print("您已进入聊天室")
return name
else:
print("该用户名存在,请重新输入")
# 接收消息
def recv_data(s):
# 无限循环 接收来自服务端发过来的消息
while True:
data, address = s.recvfrom(1024 * 10)
print(data.decode(), "\n发言", end=' ')
# 发送消息
def send_data(s, name):
while True:
try:
data = input("\n发言:")
# 突发情况强制退出
except KeyboardInterrupt:
data = 'quit'
if data == 'quit':
# 'Q'表示发出退出请求
msg = 'Q ' + name
s.sendto(msg.encode(), service_address)
sys.exit("退出聊天室")
# 'C'表示聊天请求
msg = 'C %s %s' % (name, data) # C 和各个% 之间用空格隔开,方便服务端切割分离
s.sendto(msg.encode(), service_address)
# 启动函数
def main():
# 创建udp套接字
udp_sock = socket(AF_INET, SOCK_DGRAM)
name = in_room(udp_sock)
# 子进程不可以输入,所以将收消息设置为子进程
p = Process(target=recv_data, args=(udp_sock,))
# 随父进程退出子进程退出
p.daemon = True
p.start()
send_data(udp_sock, name)
if __name__ == '__main__':
main()
这是源码,但是在windows下的pychram里执行起来客户端会阻塞在姓名输入的地方,相反 ,放在linux环境的pychram里就不会出现这种状况,问一下大神这是什么原因