python udp的应用 ,多线程实现聊天功能

之前我们一起实现过单线程的udp聊天功能,但是并不实用,这里我们用多线程重新写下

代码如下:

import socket

import threading

def print_menu():

"""显示菜单"""

print("请输入你要进行的操作: 1 发送数据　２接收数据　３退出")

def recv_data(udp_socket):

"""接收数据"""

# 数据 发送人地址

data, remote_address = udp_socket.recvfrom(4096)

print("收到来自%s的数据-----%s" % (remote_address, data.decode()))

def send_data(udp_socket):

"""发送数据"""

"""

请输入你要发送的数据:你吃了大蒜吗?

IP地址:192.168.113.113

端口:8080

"""

# 接收用户输入的数据 IP port

data = input("请输入你要发送的数据:")

ip = input("IP地址:")

port = int(input("端口:"))

udp_socket.sendto(data.encode(), (ip, port))

def main():

# 显示一个界面(菜单)　请输入你要进行的操作: 1 发送数据 2退出

# 创建一个子线程 专门收数据

# 主线程 用以接收用户键盘输入 发送

# 创建套接字

udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

# 创建线程 收数据

recv_thd = threading.Thread(target=recv_data, args=(udp_socket,))

recv_thd.start()

# 在使用之前进行绑定

udp_socket.bind(('', 8888))

while True:

print_menu()

# 接收用户输入

op = input("请选择:")

if op == '1':

send_data(udp_socket)

elif op == '2':

break

else:

print("出错了")

# 关闭

udp_socket.close()

if __name__ == '__main__':

main()