局域网内传文件-python

在局域网内部一台电脑给另一台电脑传文件

加了一点随机采样

服务端/接收图片

import random
import string

from flask import request
from flask import Flask
import os
from loguru import logger

app = Flask(__name__)


# 定义路由
@app.route("/upload_img", methods=['POST'])
def get_frame():
    # 接收图片
    upload_file = request.files['file']
    # 获取图片名
    file_name = ''.join(random.sample(string.ascii_letters + string.digits, 8)) + upload_file.filename
    # 文件保存目录（桌面）
    file_path = r'保存位置'
    if upload_file:
        # 地址拼接
        file_paths = os.path.join(file_path, file_name)
        # 保存接收的图片到桌面
        upload_file.save(file_paths)
        logger.info(f'{file_paths} save finished!')
        return 'ok'

if __name__ == "__main__":
    app.run(host='0.0.0.0')

发送端/发送图片到服务端

import os
import random

import requests
from loguru import logger

url = 'http://ip:5000/upload_img'

def send_img(img_path):
    file = open(img_path, 'rb')
    files = {'file': (os.path.basename(img_path), file, 'image/jpg')}
    resp = requests.post(url, files=files)
    if resp.status_code == 200:
        logger.info('upload finish!')
    else:
        logger.error(f'upload has error: {resp.status_code}')

#遍历文件夹
def iter_files(rootDir):
    #遍历根目录
    for root,dirs,files in os.walk(rootDir):
        logger.info(len(files))
        logger.info(root)
        if len(files) >= 1:
        	# 随机采样10张
            imgs = random.sample(files, 10)
            logger.info(f'{imgs}')
            for img in imgs:
                logger.info(f'{os.path.join(root, img)}')
                send_img(os.path.join(root, img))


if __name__ == '__main__':
    iter_files('文件夹位置')