socket发送、接收信息----UDP

# 导入套接字包
import socket


def welcome():
    print("------欢迎进入UDP聊天器--------")
    print("1.发送信息")
    print("2.接收信息")
    print("0.退出聊天器")

def send_msg(udp_socket):
    dest_ip = input("请输入目标ip:")
    dest_port = int(input("请输入目标端口："))
    send_data = input("请输入要发送的内容")
    # sendto(内容，（目标IP，端口）)
    udp_socket.sendto(send_data.encode("gbk"), (dest_ip, dest_port))


def recv_msg(udp_socket):
    # recvfrom(接收最大字节)
    rece_data = udp_socket.recvfrom(1024)
    # 要解码，不然会乱码
    print("%s:%s" % (rece_data[1], rece_data[0].decode("gbk")))


def main():
    # 创建套接字对象
    udp_socket = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    # 绑定端口
    # bing((ip,port)) bind里面是一个元组，第一个放ip 可以为空，第二个是端口
    udp_socket.bind(('',7788))
    while True:
        # 欢迎菜单
        welcome()
        # 选择功能
        op = input("请输入想要执行的命令：")
        if op == "1":
            # 发送信息
            send_msg(udp_socket)
        elif op == "2":
            # 接收信息
            recv_msg(udp_socket)
        elif op == "0":
            break
        else:
            print("请选择正确的指令：")
    # 关闭套接字
    udp_socket.close()

if __name__ == '__main__':
    main()