python-udp聊天器(同网络同协议)

import socket


def fasong(udp_socket):
    data_ip = input("输入对方ip：")
    data_post = int(input("输入对方端口："))
    udp_data = input("输入数据：")
    # 发送数据 并将数据编码
    # udp_socket.sendto(udp_data.encode("utf-8"), ("192.168.11.54", 8080))  
   udp_socket.sendto(udp_data.encode("utf-8"), (data_ip, data_post))


def jieshou(udp_socket):
    # 接收数据
    recv_data = udp_socket.recvfrom(1024)
    sor = recv_data[0]
    you = recv_data[1]
    # 解码
    print("%s:%s" % (str(you), sor.decode("gbk")))


def main():
    # 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 绑定端口
    udp_socket.bind(("", 6688))
    while True:
        print("---聊天器---")
        print("1:发送消息")
        print("2:接收消息")
        print("0:退出系统")
        op = input("输入相应功能:")
        if op == "1":
            fasong(udp_socket)
        elif op == "2":
            jieshou(udp_socket)
        elif op == "0":
            break
        else:
            print("输入有误重新输入....")

    udp_socket.close()


if __name__ == '__main__':
    main()