python学习记录（一）横竖屏图片分类器，可以筛选指定路径下JPG与PNG格式图片，分别存放到横竖屏子文件夹中

wylc123 1年前 ⋅ 3903 阅读

运行环境
Windows10
python3.8

用途：可以遍历输入的路径的指定后缀的文件,主要是用来筛选图片,将图片分成
横屏,竖屏分别存放在两个文件夹中

"""
可以遍历输入的路径的指定后缀的文件,主要是用来筛选图片,将图片分成
横屏,竖屏分别存放在两个文件夹中
作者：小邑走天下
CSDN博客主页：https://blog.csdn.net/weixin_42133116
"""
# 安装模块
'''
pip install pillow  PIL包含在里边
pip install image
pip install tqdm
pip install shutil
'''

from PIL import Image
import os
import os.path
from tqdm import tqdm
import shutil
import sys, os
if hasattr(sys, 'frozen'):
    os.environ['PATH'] = sys._MEIPASS + ";" + os.environ['PATH']



# 函数查找指定路径中所有文件的路径
def get_file(path):
    list1 = []  # 用于存储递归查找到的所有文件,传递给函数
    fileList = os.listdir(path)  # 获取path目录下所有文件
    for filename in fileList:
        pathTmp = os.path.join(path, filename)  # 获取path与filename组合后的路径
        if os.path.isdir(pathTmp):  # 如果是目录
            a = get_file(pathTmp)  # 则递归查找(注意一定要有接受变量,不然就出错了)
            for i in a:
                list1.append(i)
        else:
            list1.append(pathTmp)
    return list1


# ---------------------修改处1,修改原始文件位置------------------------------------
print("横竖屏图片分类器，可以筛选指定路径下JPG与PNG格式图片，分别存放到横竖屏子文件夹中")
print("分类结束后按任意键退出程序")
path = input('请输入路径：').strip()  #.strip()去除首尾的空格，# 待分开文件的位置
file_path_list = get_file(path)

# 筛选后缀函数,传入包含所有后缀名的列表,以及需要筛选的后缀(默认筛选txt文件)
def shai_xuan_hou_zhui(file_path_list, hou_zhui='.txt'):
    list2 = []  # 用于储存筛选好的文件的路径
    for filepath in file_path_list:
        # os.path.splitext():分离文件名与扩展名
        if os.path.splitext(filepath)[1] == hou_zhui:
            list2.append(filepath)
    return list2


# ----------------------修改2，修改文件后缀，(可完善）------------------------------------
pig_list = shai_xuan_hou_zhui(file_path_list, '.jpg')  # 筛选jpg格式文件
pig_list = shai_xuan_hou_zhui(file_path_list, '.png')  # 筛选jpg格式文件


# -----------------------修改处3，修改图片存储路径-------------------------------
# 如果不存在创建套图文件夹
if not os.path.exists(r'{}/横屏图片'.format(path)):
    os.mkdir(r'{}/横屏图片'.format(path))
if not os.path.exists(r'{}/竖屏图片'.format(path)):
    os.mkdir(r'{}/竖屏图片'.format(path))

hengping = r'{}/横屏图片'.format(path)  # 存放横屏图片的地址
shuping = r'{}/竖屏图片'.format(path)   # 存放竖屏图片的地址


for i in tqdm(range(len(pig_list))):
    lujing = pig_list[i]
    picture = Image.open(lujing)
    width = picture.width
    height = picture.height
    picture.close()
    if width > height:
        #move移动，copy复制
        #shutil.move(lujing, folder1)
        shutil.copy(lujing, hengping)
    else:
        #shutil.move(lujing, folder2)
        shutil.copy(lujing, shuping)
        print('over,over,over!')
input('please input any key to exit!')