Python制作简易聊天器，搭建UDP网络通信模型

2024-04-02 19:55

短信预约 -IT技能 免费直播动态提醒

1.导入模块

import socket

socket是套接字，链接电脑并且接收/发送消息的一个对象

socket有两种类型：udp和tcp

udp比较简单不安全丢包的情况速度快
tcp比较复杂的网络模型安全速度慢

2.创建一个套接字对象

def main():
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

udp需要借助网络：IP协议
IP协议分为两种：IPv4和IPv6
PC电脑采取的是IPv4，IPv6一般是移动端使用：手机

3.发送数据到ubuntu系统中

udp_socket.sendto(b'hello python', ('192.168.3.40', 8080))
udp_socket.close()

4.发送任意数据给网络条数助手

def main():
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    send_data = input('请输入任意字符:')
    udp_socket.sendto(send_data.encode('utf-8'), ('192.168.3.40', 8080))
    udp_socket.close()

5.循环发送数据

def main():
    # 创建套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 循环发送
    while True:
        send_data = input('请输入你想要发送的数据:')
        udp_socket.sendto(send_data.encode('utf-8'), ('192.168.3.40', 8080))
        if send_data == "exit":
            break

    udp_socket.close()

6.循环接受数据

绑定本机信息告诉ubuntu我的当前地址以及端口

def main():
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    localhost_addr = ('', 6666)
    udp_socket.bind(localhost_addr)

循环接收

while True:
    recv_data = udp_socket.recvfrom(1024)
    # 元组解包
    recv_msg, recv_addr = recv_data
    print(recv_msg.decode('utf-8'), recv_addr)

将相关的功能抽离出来做成一个函数

发送消息：

def send_msg(udp_socket):
    msg = input('请输入你想要发送的消息:')
    udp_socket.sendto(msg.encode('utf-8'), ('192.168.3.40', 8080))

接收消息：

def recv_msg(udp_socket):
    # 接收
    msg = udp_socket.recvfrom(1024)
    # 解码
    recv_msg = msg[0].decode('utf-8')
    print(recv_msg)

发送 / 接收功能：

def main():
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    udp_socket.bind(('', 6666))

    while True:
        op_num = input('请输入要操作的功能序号 [1.发送消息 2.接收消息]:' )
        # 根据用户输入的数字去调用相关的功能
        if op_num == "1":
            send_msg(udp_socket)
        elif op_num == "2":
            recv_msg(udp_socket)
        else:
            print('输入有误, 请重新输入...')