029python自动拷贝优盘的指定后缀文件

这里指定复制的文件只有png后缀的文件,可以自行修改添加需要复制的文件

# -*- coding: utf-8 -*-
"""
Created on Sun Apr 28 13:12:44 2019

@author: zangz
"""

# -*- coding: utf-8 -*-
"""
Created on Mon Nov  5 16:18:42 2018

@author: zangz
"""
import time
import os
import os.path
import shutil
from time import sleep
from shutil import copytree
from psutil import disk_partitions  #获取分区的信息

def copy_u_to_computer():
    a=True  #定义while循环结束的标志
    while (a):  #写一个一直循环的while
        sleep(1)  #暂停一秒
        #检查所有驱动器
        for item in disk_partitions():
            #发现可移动驱动器
            #(if语句的意思是):如果在opts中包含'removable'信息就继续往下执行if语句中的内容
            #不使用==的原因是opts信息中不仅包含'removable'信息和'fixed'信息
            if 'removable' in item.opts:   #removable是可拆卸的意思,fixed是固定的意思(移动硬盘同样是固定的,只有u盘会检测是removable)
                driver,opts=item.device,item.opts  #device是驱动器的根目录比如F:\
                #输出可移动驱动器符号
                print('发现优盘,嘿嘿',driver)
                a=False  #发现优盘就停止while循环
                                #复制根目录
                #
                #生成拷贝到的路径:使用时间作为路径
                # 获取当前时间
                now_time = time.strftime('%Y-%m-%d-%H:%M:%S', time.localtime(time.time()))
                #由于创建文件夹时不能有:等非法的字符,所以使用replace函数进行过滤
                now_time1=now_time.replace(':','-')
                
                print(now_time1)
                #创建一个使用时间为文件名的路径
                path='E:\\拷贝\\'+str(now_time1)
                os.mkdir(path)
                
                file_path_list=get_file(driver)
                
#                png_list=shai_xuan_hou_zhui(file_path_list,'.png')
#                #拷贝优盘中指定的后缀文件
#                for file_path in png_list:
#                    shutil.copy(file_path,path)
                #
                #想要复制指定的后缀文件只需要在这个位置进行添加修改就可以了
                copy_houzhui(file_path_list,path,'.png')

                copy_houzhui(file_path_list,path,'.docx')
                
                
                break
            else:
                print('没发现可移动优盘')
        a=False  #循环查看完一遍所有的驱动器还是没有发现可移动的优盘就跳出循环(有可能是移动硬盘)
    
#函数查找指定路径中所有文件的路径
#函数查找指定路径中所有文件的路径
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



#筛选后缀函数,传入包含所有后缀名的列表,以及需要筛选的后缀(默认筛选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)
    #        print(filepath +'\n')
    
    return list2

#拷贝指定后缀的文件到指定路径的函数
def copy_houzhui(file_path_list,path,hou_zhui):
    fil_list=shai_xuan_hou_zhui(file_path_list,hou_zhui)
    #拷贝优盘中指定的后缀文件
    for file_path in fil_list:
        shutil.copy(file_path,path)


qu_num=len(disk_partitions())

print(qu_num)

while(True):  #这里的while循环保证了,只要驱动器的个数与设定的不一样就会执行拷贝函数

    if len(disk_partitions()) == qu_num:
        print("驱动器个数未改变")
        sleep(3)
        
    else:
        print("驱动器个数发生改变,准备拷贝优盘的内容")
        qu_num=len(disk_partitions())  #改变驱动器的个数防止执行拷贝函数后死循环
        print(qu_num)
        copy_u_to_computer()  #执行拷贝函数

执行的结果:自己测试就好了,如有问题请在底下评论探讨