10行Python代码助你整理杂乱无章的文件

下面是“10行Python代码助你整理杂乱无章的文件”的完整攻略：

介绍

有时候我们的电脑中会存在着大量杂乱的文件，这些文件名字不规范，文件格式混乱，给我们的使用带来非常大的不便。本文旨在介绍如何使用Python代码，将大量杂乱无章的文件整理成为有规律、有序的文件夹。

步骤

1. 导入必要的库

使用Python进行文件操作的时候，需要导入os和shutil库。

import os
import shutil

2. 设置工作目录

设置工作目录为待整理文件夹所在路径。

path = '/path/to/folder'
os.chdir(path)

3. 合并文件夹

如有多个文件夹需要整理，需先将它们合并成为一个文件夹。

folder_list = os.listdir()
for folder in folder_list:
    # 合并文件夹
    if os.path.isdir(folder):
        folder_path = path + '/' + folder
        for dirpath, dirnames, filenames in os.walk(folder_path):
            for filename in filenames:
                shutil.move(os.path.join(dirpath,filename), path)
        # 删除空文件夹
        os.rmdir(folder_path)

4. 筛选文件

筛选符合规则的文件，进行转移。

img_list = []
video_list = []
doc_list = []
other_list = []
for filename in os.listdir():
    if os.path.isfile(filename):
        # 筛选图片格式文件
        if filename.endswith(('.png', '.jpg', '.jpeg', '.gif')):
            img_list.append(filename)
        # 筛选视频格式文件
        elif filename.endswith(('.mp4', '.avi', '.mkv')):
            video_list.append(filename)
        # 筛选文档格式文件
        elif filename.endswith(('.doc', '.docx', '.pdf', '.xls', '.xlsx', '.ppt', '.pptx')):
            doc_list.append(filename)
        # 将其它文件移动到一个单独的文件夹
        else:
            other_list.append(filename)
            if not os.path.exists('other_files'):
                os.mkdir('other_files')
            shutil.move(filename, 'other_files')

5. 创建文件夹并移动文件

根据筛选出来的文件格式创建对应文件夹，将符合格式的文件移动到对应文件夹中。

# 创建文件夹
if img_list:
    if not os.path.exists('images'):
        os.mkdir('images')
if video_list:
    if not os.path.exists('videos'):
        os.mkdir('videos')
if doc_list:
    if not os.path.exists('documents'):
        os.mkdir('documents')

# 移动文件
for img in img_list:
    shutil.move(img, 'images')
for video in video_list:
    shutil.move(video, 'videos')
for doc in doc_list:
    shutil.move(doc, 'documents')

至此，整理文件的逻辑就处理完了。下面将给出两个示例说明。

示例

假设现有一个文件夹my_folder, 里面包含三个文件：

my_folder/
    - IMG_001.jpg
    - Video0002.mkv
    - Test.docx

执行以上的代码后，我们期望达到的结果是my_folder这个文件夹变成以下结构：

my_folder/
    - images/
        - IMG_001.jpg
    - videos/
        - Video0002.mkv
    - documents/
        - Test.docx

还有一个示例文件夹。假设现有一个文件夹another_folder, 里面包含两个子文件夹和一些文件：

another_folder/
    - folder1/
        - image1.png
        - document1.docx
    - folder2/
        - video1.mp4
        - document2.pdf
    - other_file.txt
    - random.jpg

执行以上的代码后，我们期望达到的结果是another_folder这个文件夹变成以下结构：

another_folder/
    - images/
        - image1.png
    - videos/
        - video1.mp4
    - documents/
        - document1.docx
        - document2.pdf
    - other_files/
        - other_file.txt
    - random.jpg

以上就是“10行Python代码助你整理杂乱无章的文件”的完整攻略和示例说明。