python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > python搜索指定类型文件

python搜索指定类型文件以及批量移动文件程序详解

作者:侯小啾

这篇文章主要给大家介绍了关于python搜索指定类型文件以及批量移动文件程序的相关资料,文中通过实例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下

搜索文件并移动的python程序

使用python写一个程序,其功能满足:可以搜索指定目录下的某类型的文件,并可以移动到指定的目录。

                   请添加图片描述

1.示例文件及路径准备

作为示例,在D盘中放一个名为pic1的文件夹,在该文件夹中放入两个png图片,并在pic1内再创建一个文件夹,里边也随便放两张png图片。如图所示。

在D盘的ABC文件夹中,创建一个名为pic2的文件夹。

本程序要求:搜索目录"D://pic1"下的所有文件扩展名为png的图片。并将其全部移动至"D://ABC/pic2"目录下。

2.代码示例

分别定义一个搜索函数和一个批量移动函数。

搜索函数需要传入的参数为root:要搜索的目录;target为目标文件类型(即扩展名)。

搜索函数输出 由目标文件path+文件名构成的列表。

批量移动函数传入的file_list参数是一个列表,形如搜索函数的输出结果,元素为带有path的文件名。dest为需要移动到的目录。

import os, shutil

file_list = []

# 搜索函数
def search_file(root, target):
    for file in os.listdir(root):
        path = root
        try:
            path = path + os.sep + file
            if os.path.isdir(path):
                search_file(path, target)
            else:
                if file.split('.')[-1] == target:
                    file_list.append(path)
        except PermissionError as e:
            print(e)
    return file_list


# 批量移动函数
def move_file(file_list, dest):
    for file in file_list:
        try:
            shutil.move(file, dest)
        except shutil.Error as e:
            print(e)


# 写入目标参数root,
def main():
    root = "D:\\pic1"
    target = "png"
    dest_dir = "D:\\ABC\\pic2"
    result = search_file(root, target)
    print(result)
    move_file(result, dest_dir)


if __name__ == '__main__':
    main()

如图,目标文件被成功搜索出并转移:

3.命令行写法

将上述代码改写文能在命令行传参、调用的形式。

首先将目标文件放回原路径,放好。

import sys
import os, shutil
file_list = []

# 定义一个搜索功能的函数
def search_file(root, file_name):
    for file in os.listdir(root):
        path = root
        try:
            path = path + os.sep + file
            if os.path.isdir(path):
                search_file(path, file_name)
            else:
                if file.split('.')[-1] == file_name:
                    file_list.append(path)
        except PermissionError:
            pass
    return file_list

def move_file(file_list, dest):
    for file in file_list:
        shutil.move(file, dest)

def main(argv):
    root = argv[1]
    target = argv[2]
    dest_dir = argv[3]
    result = search_file(root, target)
    print(result)
    move_file(result, dest_dir)

if __name__ == '__main__':
    main(sys.argv)

打开cmd,目录调整好后,输入py文件名(这里为demo.py)及依次传入三个参数。

即执行命令:

代码成功运行,目标文件位置被更改。

总结

到此这篇关于python搜索指定类型文件以及批量移动文件程序详解的文章就介绍到这了,更多相关python搜索指定类型文件内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文