python分离不同后缀名的文件

功能描述

• 根据文件后缀名处理文件，分别拷贝到对应的文件夹下
• example >>> .jpg 后缀 和.mp4 后缀文件处理

代码实现

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import os
import shutil

def deal_file(src,mp41,jpg2):
    # 区分jpg和mp4
    mp4 = []
    jpg = []
    for f in os.listdir(src):
        if f.endswith('.mp4'):
            mp4.append(f)
        elif f.endswith('.jpg'):
            jpg.append(f)
    # 创建目标文件夹
    if not os.path.isdir(mp41):
        os.mkdir(mp41)
    if not os.path.isdir(jpg2):
        os.mkdir(jpg2)
    # 拷贝文件到目标文件夹
    for m in mp4:
        mp4_1 = os.path.join(src,m)
        shutil.copy(mp4_1,mp41)
    for j in jpg:
        jpg_1 = os.path.join(src,j)
        shutil.copy(jpg_1,jpg2)

if __name__ == "__main__":
     file = "./"
     src = os.path.join(file,'datafile')
     mp41 = os.path.join(file, 'mp4_file')
     jpg2 = os.path.join(file, 'jpg_file')
     deal_file(src, mp41, jpg2)