UDP聊天器

'''
一、功能
1、发送信息
2、接收信息
3、退出系统

二、框架设计
1、发送信息 send_msg()
2、接收信息 recv_msg()
3、程序的主入口 main()
4、当程序独立运行的时候，才启动聊天器


三、实现步骤
1、发送信息
  1）定义变量接收 对方IP地址
  2）定义变量接收 对方端口号
  3)定义变量接收 发送给对方的内容
  4）使用socket的sendto()发送信息

2、接收信息
  1）使用socket 接收数据
  2）解码数据
  3）输出显示

3、主入口
  1）创建套接字
  2）绑定端口
  3）打印菜单
  4）接收用户输入的选项
  5）判断用户的选择并调用
  6）关闭套接字
'''
import socket
def menu():
    print("*"*25)
    print("-------1、发送信息--------")
    print("-------2、接收信息--------")
    print("-------3、退出------------")
    print("*" * 25)

def send_msg(udp_socket):
    #1、定义变量接收用户与输入的接收方的IP地址
    ipaddr=input("请输入接收方的IP地址:")
    if len(ipaddr)==0:
        ipaddr="192.168.145.1"
        print("IP默认为%s:"%ipaddr)
    #2、定义变量接受用户与输入的接收方的端口号
    port=input("请输入接收方的端口号:")
    if len(port)==0:
        port='8080'
        print("端口号默认为:%s"%port)
    #3、定义变量接收用户与输入的接收方的内容
    content=input("请输入发送的内容:")
    #4、使用socket中的sendto发送信息
    udp_socket.sendto(content.encode(),(ipaddr,int(port)))

def recv_msg(udp_scoket):
    #1、使用socket接收数据
    recv_data,ip_port=udp_scoket.recvfrom(1024)
    #2、解码数据
    recv_text=recv_data.decode()
    #3、输出显示
    print("接收到【%s】的消息:%s"%(str(ip_port),recv_text))
def main():
    # 1）创建套接字
    upd_socket=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
    # 2）绑定端口
    upd_socket.bind(("",8080))
    # 3）打印菜单
    while True:
        menu()
        # 4）接收用户输入的选项
        select_num=input("请输入选项:")
        # 5）判断用户的选择并调用
        if select_num=='1':
            send_msg(upd_socket)
        elif select_num=='2':
            recv_msg(upd_socket)
        elif select_num=='3':
            print("正在退出...")
            break
    # 6）关闭套接字
    upd_socket.close()
if __name__ == '__main__':
  #程序独立运行的时候，才去启动聊天器
   main()