Python UDP聊天器

注释:下面的逻辑分析思路来自李伟老师,感谢在python学习道路上提供的帮助。

程序需求说明:
在一个客户端中编写个程序,有3个基本的功能:

  • 1.首先获取键盘数据,并将其发送给对方
  • 2.对方接收数据并显示
  • 3.退出聊天系统

接下来我们分析下思路

功能: 1、发送消息 2、接收消息 3、退出系统

设计:

  • 发送消息 send_msg()
  • 接收消息 recv_msg()
  • 程序入口 main()

主入口:
1、导入模块 2、创建套接字 3、绑定端口 发送消息:

  • 1、保存接收方信息(用户输入)
  • 2、保存接收方端口号(用户输入)
  • 3、保存要发送的内容(用户输入)
  • 4、把用户输入的内容编码并发送

接收消息:
1、使用socket recvfrom 接收数据
2、解码数据并且输出

代码实现:

定义一个send_msg()发送消息和recv_msg()接受消息的函数,是最基本的消息功能,下面通讯所用的是底层socket套接字。

import socket

def send_msg(udp_socket):
    """发送消息的方法"""
    # 1、输入接收方的ip地址
    ipaddr = input("请输入接收方地址:\n")
    if len(ipaddr) == 0:    #如果不输入对应的ip地址,默认设置为本机的ip地址
        ipaddr = "192.168.1.1."   #使用 ifconfig 命令 查看本机的ip地址
        print("默认设置为:%s" % ipaddr)
    # 2、输入接收方的端口
    port = input("请输入接收方端口号:\n")
    if len(port) == 0:      #  端口不写,提供一个默认值
        port = "2438"
        print("默认设置为:%s" % port)
    # 3、要求输入要发送的内容
    content = input("请输入要发送的内容:\n")
    # 4、发送数据
    udp_socket.sendto(content.encode(), (ipaddr, int(port)))

def recv_msg(udp_socket):
    # 1、接收数据  1024表示本次接受的最大的字节数
    recv_data = udp_socket.recvfrom(1024)
    # 2、把接收到的数据(二进制)解码并且显示出来
    re_text = recv_data[0].decode()
    print("接收到消息为:%s" % re_text)
    ip_port = recv_data[1]
    # 3、打印显示发送方的IP和端口信息
    print(ip_port)

main()函数是主程序的入口,上面是定义,下面我们来调用

if __name__ == '__main__':
    
# 创建套接字 
udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

# 绑定端口为2438
udp_socket.bind(("", 2438))

while True:
    # 给出提示,显示聊天器的主要功能
    print("*************************")
    print("****   1.发送消息  ****")
    print("****   2.接收消息  ****")
    print("****   3.退出系统  ****")
    print("*************************")

    # 1、提示用户选择功能
    num = int(input("请选择功能:\n"))
    # 2、判断用户选择
    # 如果选择1,调用发送消息的函数
    if num == 1:
        # print("您选择了发送消息")
        send_msg(udp_socket)
    # 如果选择2,调用接收消息的函数
    elif num == 2:
        # print("您
        recv_msg(udp_socket)

    # 如果选择3,退出程序执行
    else:
        print("程序正在退出...")
        break
        print("程序已退出!~")

# 关闭套接字
udp_socket.close()

注意:
这个程序是简单的了解下udp的通信过程,有许多的bug。比如说当我们选择了接受消息的时候,可是这个时候并没有消息,会造成堵塞的问题,解决的方法是接受发送消息一起进行,这个问题我们先放在这里以后再解决。

;