python socket模块实现上传文件到服务器

socket模块文件上传案例

cat server.py

# -*- coding:UTF-8 -*-
import socket

sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('0.0.0.0', 8001))  # 127.0.0.1 或 查看自己局域网本地IP地址
sock.listen(1)

conn, addr = sock.accept()

# 接收文件大小
data = conn.recv(1024)
total_file_size = int(data.decode('utf-8'))

# 接收文件内容
file_object = open('krb5-libs-1.15.1-50.el7.x86_64.rpm', mode='wb')
recv_size = 0
while True:
    # 每次最多接收1024字节
    data = conn.recv(1024)
    file_object.write(data)
    file_object.flush()

    recv_size += len(data)
    # 上传完成
    if recv_size == total_file_size:
        break

# 接收完毕，关闭连接
conn.close()
sock.close()

cat client.py

# -*- coding:UTF-8 -*-
import time
import os
import socket

client = socket.socket()
client.connect(('10.10.10.201', 8001))

file_path = 'krb5-libs-1.15.1-50.el7.x86_64.rpm'

# 先发送文件大小
file_size = os.stat(file_path).st_size
client.sendall(str(file_size).encode('utf-8'))

print("准备...")
time.sleep(2)
print("开始上传..")
file_object = open(file_path, mode='rb')
read_size = 0
while True:
    chunk = file_object.read(1024) # 每次读取1024字节
    client.sendall(chunk)
    read_size += len(chunk)
    if read_size == file_size:
        break

client.close()

将server.py上传到服务器上并运行

python3 server.py

本地运行client.py，修改要上传的文件路径和服务器的IP即可上传成功

使用场景：

服务器误操作将krb5-libs的rpm包强制删除，导致服务器无法ssh连接，并且不能上传rpm包进行修复
在强制删除依赖后，当前终端没有退出的时候，可以使用上述方法补救