企业微信json文件大于20m而报错的解决方法python

原创

mob649e815f494b 2024-11-28 05:11:21 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e815f494b的原创作品，请联系作者获取转载授权，否则将追究法律责任

解决企业微信 JSON 文件大于 20m 报错的 Python 方法

在企业微信中，上传的 JSON 文件不能超过 20MB，否则会报错。以下是解决这个问题的步骤和实现方法。我们首先会概述整个流程，然后逐步介绍每一步需要实现的具体代码。

流程概述

步骤	描述
1	检查 JSON 文件大小
2	如果文件大于 20MB，进行分割处理
3	将分割后的文件逐一上传
4	处理上传后的结果
5	合并最终结果（如果需要）

具体实现

1. 检查 JSON 文件大小

首先，我们需要检查文件的大小。以下是 Python 代码示例:

import os

def check_file_size(file_path):
    """检查文件大小，返回文件大小（字节）"""
    return os.path.getsize(file_path)

# 示例使用
file_path = 'your_file.json'
file_size = check_file_size(file_path)
if file_size > 20 * 1024 * 1024:
    print("文件大小大于20MB")
else:
    print("文件大小合适")

2. 分割处理

如果文件大于 20MB，我们需要将其分割成多个小文件。可以使用以下代码实现分割：

import json

def split_json_file(file_path, max_size):
    """分割 JSON 文件，返回分割后的文件路径列表"""
    with open(file_path, 'r', encoding='utf-8') as f:
        data = json.load(f)
    
    file_paths = []
    current_size = 0
    current_chunk = []
    
    for entry in data:
        entry_size = len(json.dumps(entry).encode('utf-8'))  # 计算当前条目的字节数
        if current_size + entry_size > max_size:
            # 写入当前分割文件
            chunk_file_path = f'chunk_{len(file_paths)}.json'
            with open(chunk_file_path, 'w', encoding='utf-8') as chunk_file:
                json.dump(current_chunk, chunk_file)
            file_paths.append(chunk_file_path)
            current_chunk = []
            current_size = 0
        
        current_chunk.append(entry)
        current_size += entry_size
    
    # 写入最后一部分
    if current_chunk:
        chunk_file_path = f'chunk_{len(file_paths)}.json'
        with open(chunk_file_path, 'w', encoding='utf-8') as chunk_file:
            json.dump(current_chunk, chunk_file)
        file_paths.append(chunk_file_path)
    
    return file_paths

# 示例使用
chunk_files = split_json_file(file_path, 20 * 1024 * 1024)  # 20MB
print("分割文件列表:", chunk_files)

3. 上传文件

接下来，我们需要将每个分割后的文件上传。下面是一个函数示例:

import requests

def upload_file(file_path):
    """上传文件到企业微信，返回上传结果"""
    url = '
    
    with open(file_path, 'rb') as f:
        response = requests.post(url, files={'file': f})
    
    return response.json()

# 示例使用上传每个分割文件
for chunk_file in chunk_files:
    result = upload_file(chunk_file)
    print("上传结果:", result)

4. 处理上传结果

这里我们可以根据需要处理上传后的结果，例如保存上传成功的 ID 或做日志记录。

5. 合并结果

最后，如果业务需求需要合并处理后的结果，你可以根据上传的结果进行合并。

def merge_results(results):
    """合并上传的结果"""
    merged_data = []
    for result in results:
        merged_data.append(result)  # 根据具体需求修改
    return merged_data

以下是类图示例，描述主要类和功能之间的关系：

classDiagram
    class FileHandler {
        +check_file_size(file_path)
        +split_json_file(file_path, max_size)
        +upload_file(file_path)
        +merge_results(results)
    }