Day 27 subprocess模块/粘包问题以及解决/UDP协议/socketserver

目录

• subprocess模块
• 粘包问题
  • 粘包的两种情况
  • 解决粘包问题
    • struct模块
• 上传大文件
• UDP协议
• QQ聊天室
• socketserver模块

subprocess模块

subprocess模块的作用:

1. 可以通过代码执行操作系统的终端命令
2. 并返回终端执行命令后的结果

import subprocess

cmd = input('>>> ').strip()

obj = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

result = obj.stdout.read() + obj.stderr.read()

print(result.encode('gbk'))

粘包问题

粘包的两种情况

1. 接收方不能一次接收完缓存区的包,造成下次接收继续从缓存区中拿取遗留的数据
2. 多次连续发送数据量小、并且时间间隔短的数据会一次性打包发送。

解决粘包问题

struct模块

# 服务端
import socket
import json
import struct

server = socket.socket()

server.bind(
    ('127.0.0.1', 8080)
)

server.listen(3)

conn, addr = server.accept()
print(addr)

file_dic = {
    'file_name': 'test.txt',
    'file_size': 10000000
}
# 序列化
json_data = json.dumps(file_dic)
# 转成utf8格式
bytes_data = json_data.encode('utf8')

# 生成报头
header = struct.pack('i', len(bytes_data))
# 发送报头
conn.send(header)
conn.send(bytes_data)

# 客户端
import socket
import json
import struct

client = socket.socket()

client.connect(
    ('127.0.0.1', 8080)
)

# 接收报头
header = client.recv(4)
# 打开报头,获取长度
data_len = struct.unpack('i', header)[0]

bytes_data = client.recv(data_len)
# 反序列化
back_dic = json.loads(bytes_data.decode('utf8'))

print(back_dic['file_name'])

上传大文件

# 服务端
import socket
import json
import struct

server = socket.socket()

server.bind(
    ('127.0.0.1', 8080)
)

server.listen(3)

conn, addr = server.accept()
print(addr)

with open(r'E:电子书Python电子书Head_First_Python（中文版）.pdf', 'rb') as f:
    e_book_bytes = f.read()

file_dic = {
    'file_name': 'Head Of Python.pdf',
    'file_size': len(e_book_bytes)
}

json_data = json.dumps(file_dic)
bytes_data = json_data.encode('utf8')

header = struct.pack('i', len(bytes_data))
conn.send(header)
conn.send(bytes_data)

init_data = 0
with open(r'E:电子书Python电子书Head_First_Python（中文版）.pdf', 'rb') as f:
    while init_data < len(e_book_bytes):
        data = f.read(1024)
        conn.send(data)
        init_data += 1024

# 客户端
import socket
import json
import struct

client = socket.socket()

client.connect(
    ('127.0.0.1', 8080)
)

header = client.recv(4)
data_len = struct.unpack('i', header)[0]

bytes_data = client.recv(data_len)
back_dic = json.loads(bytes_data.decode('utf8'))
file_name = back_dic['file_name']
file_size = back_dic['file_size']


init_data = 0
with open(file_name, 'wb') as f:
    while init_data < file_size:
        data = client.recv(1024)
        f.write(data)
        init_data += len(data)

print(f'{file_name}下载成功!')

UDP协议

UDP协议是一种传输协议

1. 不需要建立双向管道
2. 不会粘包
3. 客户端给服务端发送数据,不需要等服务端返回接收成功
4. 数据容易丢失,数据不安全

# 服务端
import socket

server = socket.socket(type=socket.SOCK_DGRAM)

server.bind(
	('127.0.0.1', 8080)
)

msg, addr = server.recvfrom(1024)
msg1, addr1 = server.recvfrom(1024)
msg2, addr2 = server.recvfrom(1024)

# 客户端
import socket

client = socket.socket(type=socket.SOCK_DGRAM)

ip_port = ('127.0.0.1', 8080)

client.sendto(b'Hello', ip_port)
client.sendto(b'Hello', ip_port)
client.sendto(b'Hello', ip_port)
client.sendto(b'Hello', ip_port)

QQ聊天室

# 服务端
import socket

server = socket.socket(type=socket.SOCK_DGRAM)

server.bind(
    ('127.0.0.1', 9527)
)

while True:

    # 服务端接收客户端传过来的消息
    msg, addr = server.recvfrom(1024)  # (消息，客户端地址)
    msg1, addr1 = server.recvfrom(1024)  # (消息，客户端地址)
    msg2, addr2 = server.recvfrom(1024)  # (消息，客户端地址)

    print(addr)
    print(addr1)
    print(addr2)
    print(msg.decode('utf-8'))
    print(msg1.decode('utf-8'))
    print(msg2.decode('utf-8'))

    # 服务端往客户端发送消息
    send_msg = input('服务端发送消息:').encode('utf-8')
    server.sendto(send_msg, addr)
    server.sendto(send_msg, addr1)
    server.sendto(send_msg, addr2)

# 客户端
import socket

client = socket.socket(type=socket.SOCK_DGRAM)

server_ip_port = ('127.0.0.1', 9527)

while True:
    send_msg = input('客户端1: ').encode('utf-8')

    # 发送消息必须要加上对方地址
    client.sendto(send_msg, server_ip_port)

    # 能接收任何人的消息
    msg = client.recv(1024)

    print(msg.decode('utf-8'))

socketserver模块

socketserver是python内置的模块,可以简化socket套接字服务端的代码

# 服务端
import socketserver

# 定义类
# TCP: 必须继承BaseRequestHandler类
class MyTcpServer(socketserver.BaseRequestHandler):
    # 必须重写父类的handle
    def handle(self):

        # 1.接收消息
        data = self.request.recv(1024)  # conn.recv(1024)
        print(data)
        # 2.给客户端发送消息
        send_msg = input('服务端: ').encode('utf-8')
        self.request.send(send_msg)


if __name__ == '__main__':
    socketserver.TCPServer.allow_reuse_address = True
    server = socketserver.TCPServer(
        ('127.0.0.1', 8888), MyTcpServer
    )

    # 永久执行服务
    server.serve_forever()

# 客户端
import socket

client = socket.socket()

client.connect(

    ('127.0.0.1', 8888)
)

while True:
    send_msg = input('客户端: ')

    client.send(send_msg.encode('utf-8'))

    back_msg = client.recv(1024)

    print(back_msg.decode('utf-8'))