python自动聊天神器_python-udp聊天器

import socket

def send_udp(udp_socket, dest_ip, dest_port):

# 发送

send_data = input('请输入要发送的内容：')

udp_socket.sendto(send_data.encode('utf-8'), (dest_ip, dest_port))

def recv_udp(udp_socket):

# 接收返回的数据

recv_data = udp_socket.recvfrom(1024)

print(recv_data[0].decode('gbk'), recv_data[1])

def main():

# 创建一个udp套接字

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

#绑定信息

udp_socket.bind(('', 7788))

# 获取对方的ip/port

dest_ip = input('请输入对方的IP:')

dest_port = int(input('请输入对方的端口:'))

while True:

print('-----制作聊天器------')

print('1: 发送消息')

print('2：接收消息')

print('0: 退出系统')

udp_input = input('请输入功能：')

if udp_input == '1':

# 发送

send_udp(udp_socket, dest_ip, dest_port)

elif udp_input == '2':

# 接收返回的数据

recv_udp(udp_socket)

elif udp_input == '0':

break

else:

print('输入错误，请重新输入')

#关闭套接字

udp_socket.close()

if __name__ == '__main__':

main()