python小工具

2018-11-19

日常使用的Python小工具脚本文件合集：暂时包括按照文件类型分类和文件名日期命名统一脚本。

文件类型分类

import os
import shutil
from enum import Enum

class dirStruct(Enum):
    DirNone = 1     #直接拷贝到指定的目录
    DirExt = 2      #按后缀名新建文件夹，将相同的文件拷贝到指定的目录
    DirOrigin = 3   #按照原来目录来新建目录并且拷贝文件


def classify_file(srcpath, dstpath, ext, dirstrut):
    for root, _, files in os.walk(srcpath):
        if dirstrut is dirStruct.DirOrigin:
            newpath = root.replace(srcpath, dstpath)
            if not os.path.exists(newpath):
                os.mkdir(newpath)
            for filename in files:
                if os.path.splitext(filename)[1] in ext:
                    filepath = os.path.join(root, filename)
                    shutil.copy(filepath, newpath)

        if dirstrut is dirStruct.DirExt:
            for dirext in ext:
                dirpath = os.path.join(dstpath, dirext.lstrip('.'))
                if not os.path.exists(dirpath):
                    os.mkdir(dirpath)
            for filename in files:
                extname = os.path.splitext(filename)[1]
                if extname in ext:
                    filepath = os.path.join(root, filename)
                    newfilepath = os.path.join(dstpath, extname.lstrip('.'))
                    shutil.copy(filepath, newfilepath)

        if dirstrut is dirStruct.DirNone:
            for filename in files:
                if os.path.splitext(filename)[1] in ext:
                    filepath = os.path.join(root, filename)
                    shutil.copy(filepath, dstpath)


if __name__ == "__main__":
    srcpath = r'E:\download\speedpan\xxxx'
    dstpath = r'E:\download\speedpan\classify'
    classify_file(srcpath, dstpath, ['.jpg'], dirStruct.DirExt)

统一文件名日期

import os
import shutil
import re


def normalize_date_1(filepath, year):
    pattern = re.compile(r'^([a-z]*)(\d{4,})(.*)')
    for root, _, files in os.walk(filepath):
        for filename in files:
            old_filename = os.path.join(root, filename)
            match = pattern.match(filename)
            if match:
                if len(match.group(2)) == 4:
                    new_filename = match.group(1) + year + match.group(2) + match.group(3)
                    new_filename = os.path.join(root, new_filename)
                    os.rename(old_filename, new_filename)


"""
将形如wj0908文件名开始的文件整理成wj20180908开始的文件名
"""
def normalize_1():
    filepath = r'E:\网盘空间\speedpan\整理\硅谷来信'
    normalize_date_1(filepath, "2017")


def normalize_date_2(filepath):
    pattern = re.compile(r'^([a-z]*)(\d{1,2})(\.)(\d{1,2})(.*)')
    for root, _, files in os.walk(filepath):
        for filename in files:
            old_filename = os.path.join(root, filename)
            match = pattern.match(filename)
            if match:
                prefix = match.group(1)
                month = match.group(2)
                day = match.group(4)
                suffix = match.group(5)
                if(len(month) == 1):
                    month = '0' + month
                if(len(day) == 1):
                    day = '0' + day
                new_filename = prefix + month + day + suffix
                new_filename = os.path.join(root, new_filename)
                os.rename(old_filename, new_filename)


"""
将形如wj4.1文件名开始的文件整理成wj0401开始的文件名
"""
def normalize_2():
    filepath = r'E:\网盘空间\speedpan\整理\硅谷来信'
    normalize_date_2(filepath)


def normalize_date_3(filepath):
    pattern = re.compile(r'^([a-z]*)(\d{1,})(.*)')
    for root, _, files in os.walk(filepath):
        for filename in files:
            old_filename = os.path.join(root, filename)
            match = pattern.match(filename)
            if match:
                prefix = match.group(1)
                month_day = match.group(2)
                suffix = match.group(3)
                if(len(month_day) == 2):
                    month_day = '0' + month_day[0] + '0' + month_day[1]
                if(len(month_day) == 3):
                    month_day = '0' + month_day
                new_filename = prefix + month_day + suffix
                new_filename = os.path.join(root, new_filename)
                os.rename(old_filename, new_filename)


"""
将形如wj41文件名开始的文件整理成wj0401开始的文件名
"""
def normalize_3():
    filepath = r'E:\网盘空间\speedpan\整理\硅谷来信'
    #filepath = r'E:\网盘空间\speedpan\整理\test'
    normalize_date_3(filepath)


if __name__ == "__main__":
    normalize_1()