Python之復制文件目錄和快捷方式

需求

假設現在需要將某一個目錄A下的文件生成快捷方式到另一個目錄B下。具體要求如下(可以參考下圖):

• 在目錄A的第一級子目錄下的內容如果為文件夾，則目錄B子目錄創建同名文件夾；如果是文件，則創建該文件的快捷方式。
• 在目錄A的子目錄的子目錄下，創建文件或文件夾的快捷方式。

實現思路

1. 循環可遞歸判斷當前文件夾下的子目錄。

這里需要引用os—— import os ；獲取當前目錄的文件/文件夾列表方法為 os.listdir(path) # path表示目錄名稱 ；判斷是否為文件的方法為：  os.path.isfile(file) #file表示文件的完整路徑  ；如果不存在則創建文件夾的方法為：  os.makedirs(path)#path表示文件夾路徑 。

2. 創建快捷方法。

創建快捷方式的代碼如下：

import pythoncom
import  os
from win32com.shell import shell, shellcon
'''
sourceName 文件名稱完整路徑
targetName 快捷鍵名稱完整路徑
icon 圖標完整路徑——如無需特別設置，可去掉iconname 參數
'''
def setShortcut(sourceName ="" , targetName="" , iconName = ""):
    try:
        shortcut = pythoncom.CoCreateInstance(
            shell.CLSID_ShellLink, None,
            pythoncom.CLSCTX_INPROC_SERVER, shell.IID_IShellLink)
        # 獲取接口
        persist = shortcut.QueryInterface(pythoncom.IID_IPersistFile)
        # 設置數據
        shortcut.SetPath(sourceName)
        shortcut.SetIconLocation(iconName, 0)  # 可有可無，沒有就默認使用文件本身的圖標
        # 保存快捷文件位置
        persist.Save(targetName, 0)
        return  True
    except Exception as e:
        print(e.args)
        return False

源碼

import pythoncom
import  os
from win32com.shell import shell, shellcon

sourcePath = 'D:\\臨時\\劇本殺大全\\1..經典本'
targetPath = 'D:\\臨時\\劇本殺大全\\已閱讀'

'''
sourceName 文件名稱完整路徑
targetName 快捷鍵名稱完整路徑
icon 圖標完整路徑——如無需特別設置，可去掉iconname 參數
'''
def setShortcut(sourceName ="" , targetName="" , iconName = ""):
    try:
        shortcut = pythoncom.CoCreateInstance(
            shell.CLSID_ShellLink, None,
            pythoncom.CLSCTX_INPROC_SERVER, shell.IID_IShellLink)
        # 獲取接口
        persist = shortcut.QueryInterface(pythoncom.IID_IPersistFile)
        # 設置數據
        shortcut.SetPath(sourceName)
        shortcut.SetIconLocation(iconName, 0)  # 可有可無，沒有就默認使用文件本身的圖標
        # 保存快捷文件位置
        persist.Save(targetName, 0)
        return  True
    except Exception as e:
        print(e.args)
        return False
# 保存文件的快捷目錄到另一個文件夾（1.小於層級時，如果是目錄，則復制目錄；如果是文件，生成快捷方式）
# sourcePath 源目錄
# targetPath 目標目錄
# level
def saveShortCuts(sourcePath, targetPath , level = 2) :
    try:
        # 判斷源目錄和目標目錄是否存在(目標目錄不存在則創建)
        if os.path.exists(sourcePath) == False :
            return False
        if os.path.exists(targetPath) == False:
            os.makedirs(targetPath)

        # 循環遞歸獲取當前文件夾
        tempPaths = os.listdir(sourcePath)
        for tempPath in tempPaths:
            if os.path.isfile(sourcePath +'\\' +tempPath) or level <= 1:
                # 類型為文件 或 等級小於等於1 ，創建快捷方式
                setShortcut(sourceName=sourcePath + '\\' + tempPath , targetName= targetPath + '\\' + tempPath + '.lnk')
            else:
                # 文件類型為文件夾 且等級大於1 ，創建當前目錄，並遞歸調用
                os.makedirs(targetPath +  '\\' + tempPath )
                saveShortCuts(sourcePath = sourcePath + '\\' + tempPath , targetPath = targetPath +  '\\' + tempPath , level = level -1)

        return True
    except Exception as e:
        print(e.args)
        return False

result = saveShortCuts(sourcePath = sourcePath , targetPath = targetPath)
print("返回結果" , result)

參考網址

• https://jingyan.baidu.com/article/b907e62701060b06e6891c6f.html
• https://blog.csdn.net/weixin_43903378/article/details/94392277
• Python pythoncom.CoCreateInstance方法代碼示例:https://vimsky.com/examples/detail/python-method-pythoncom.CoCreateInstance.html
• Python判斷文件是否存在的3中辦法 :https://www.cnblogs.com/jhao/p/7243043.html
• Python 創建文件和文件夾：https://jingyan.baidu.com/article/1709ad80411ca64635c4f046.html
• Python 創建文件和文件夾 ：https://blog.csdn.net/weixin_41290863/article/details/110733222
• 源碼位置：https://github.com/mhammond/pywin32