思路分析
1.功能:(1)发送消息 (2)接收消息 (3)退出系统
2.架构:
(1)发送消息:send_msg()
(2)接收消息:recv_msg()
(3)主入口:main()
发送消息:(1)用户手动输入信息 (2)将消息编码并发送
接收消息:(1)循环接收消息 (2)解码并显示
主入口:(1)创建套接字,绑定端口 (2)创建一个线程,执行接收消息 (3)选择功能
代码实现
# 导入模块
import socket
import threading
# 发送信息
def send(udp_socket):
ip = input("请输入目标IP地址:\n")
if len(ip) == 0:
ip = '192.168.43.17'
print("默认为:192.168.43.17")
port = input("请输入目标端口:\n")
if len(port) == 0:
port = '8080'
print("默认为:8080")
content = input("请输入要发送的内容:\n")
udp_socket.sendto(content.encode(), (ip, int(port)))
# 接收信息
def recv(udp_socket):
# 循环接收消息
while True:
content_data, ip_port = udp_socket.recvfrom(1024)
print("【新消息】", ip_port, ":", content_data.decode())
if __name__ == '__main__':
# 创建UDP套接字
udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
# 绑定端口
udp_socket.bind(("", 8080))
# 创建一个子线程,执行接收消息的方法
t1 = threading.Thread(target=recv, args=(udp_socket,))
# 设置线程守护,主线程停止,子线程也停止
t1.setDaemon(True)
# 开启线程
t1.start()
while True:
print("**********************")
print("***** 1、发送信息 ****")
print("***** 2、退出系统 ****")
print("**********************")
# 4)接收用户输入的菜单编号
sel_num = input("请输入菜单编号:\n")
# 5)根据编号判断要执行的功能
if sel_num == '1':
send(udp_socket)
elif sel_num == '2':
print("系统正在退出...")
print("退出成功!")
break
else:
print("输入错误,请重新输入")
# 6)关闭套接字
udp_socket.close()
执行结果
用多线程改写UDP聊天器后,不用手动去接收消息,子线程会自动接收消息,并实时显示出来。