udp_socket聊天器demo

# -*- coding:utf-8 -*-
# Author:Sure Feng
import socket


def send_msg(udp_socket):
    """获取键盘数据，并将其发送给对方"""
    # 获取对方ip、端口
    dest_ip = input("请输入对方的ip")
    dest_port = input("请输入对方的端口")
    # 获取要发送的信息
    msg = input("请输入要发送的信息")
    # 按照utf8编码后，发送数据
    udp_socket.sendto(msg.encode("utf-8"), (dest_ip, int(dest_port)))


def rece_msg(udp_socket):
    # 接收数据
    msg = udp_socket.recvfrom(1024)
    # 解码
    recv_ip = msg[1]
    recv_msg = msg[0].decode("utf-8")
    # 显示解码后数据
    print("%s:%s" % (str(recv_ip), recv_msg))


def main():
    # 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 绑定本地信息
    udp_socket.bind(("", 9890))
    while True:
        print("=" * 30)
        print("1：发送消息")
        print("2：接收消息")
        print("=" * 30)
        op_num = input("请输入要操作的功能序号：")

        # 根据需求调用相应的函数
        if op_num == "1":
            send_msg(udp_socket)
        elif op_num == "2":
            rece_msg(udp_socket)
        else:
            print(">>> 输入有误，请重新选择")


        # 关闭套接字
        udp_socket.close()


if __name__ == "__main__":
    print("main")
    main()