从S3下载文件并按行数拆分

import boto3

s3 = boto3.resource('s3')

def download_file_from_s3(bucket_name, key):
    try:
        # 下载文件
        s3.Bucket(bucket_name).download_file(key, 'local_file.txt')
        print("文件下载成功！")
    except Exception as e:
        print("文件下载失败：", e)

def split_file_by_line(file_path, line_count):
    try:
        with open(file_path, 'r') as file:
            lines = file.readlines()
            
            # 按行数拆分文件
            split_files = []
            for i in range(0, len(lines), line_count):
                split_file_path = f'split_{i}.txt'
                split_files.append(split_file_path)
                
                with open(split_file_path, 'w') as split_file:
                    split_file.writelines(lines[i:i+line_count])
            
            print(f"文件成功拆分为 {len(split_files)} 个文件。")
            return split_files
    except Exception as e:
        print("文件拆分失败：", e)

# 设置S3存储桶和文件的键
bucket_name = 'your_bucket_name'
file_key = 'your_file_key'

# 下载文件
download_file_from_s3(bucket_name, file_key)

# 拆分文件
split_files = split_file_by_line('local_file.txt', 100)