Python：如何將具有特定名稱的文件移動到具有相同名稱的文件夾中

Question

我有一種情況，我正在處理的目錄有很多文件和文件夾（類似這樣）：

AAA_SUBFOLDER
AAA_FILE_1
AAA_FILE_2
BBB_SUBFOLDER
BBB_FILE_1
BBB_FILE_2

所以文件和子文件夾都以AAA或BBB開頭（等等以 CCC、DDD.. 開頭）。 我想做的是一個python 腳本，它將所有AAA文件移動到AAA子文件夾，並對所有具有相同名稱的文件和子文件夾進行迭代，以獲得如下內容：

AAA_SUBFOLDER
  - AAA_FILE_1
  - AAA_FILE_2
BBB_SUBFOLDER
  - BBB_FILE_1
  - BBB_FILE_2
...

你能幫助我嗎？ 提前致謝！

Answer 1

該解決方案應該可以滿足您的要求。 腳步：

1. 確保您已安裝 python
2. 將腳本保存到文件（比如說 main.py）
3. 使用 2 arguments 運行保存的文件 - 1 用於您要組織的路徑，2 用於子文件夾的分隔符。 例如： python main.py -p "D:\path\to\files" -d "_"

,. 這不會重新排列內部文件夾，只會重新排列頂層。

如果您有任何問題，我很樂意提供幫助。

import os
import argparse
from pathlib import Path

def parse_args() -> tuple:
    ap = argparse.ArgumentParser()
    ap.add_argument("-p", "--path", default="./files", help="path to the folder that needs organizing")
    ap.add_argument("-d", "--delimiter", default="_", help="delimiter of file names")
    return ap.parse_args()

args = parse_args()

for filename in os.listdir(args.path):
    file_path = os.path.join(args.path, filename)
    if not os.path.isfile(file_path):
        continue
    subfolder_name = Path(filename).stem.split(args.delimiter)[0]
    subfolder_path = os.path.join(args.path,subfolder_name)
    os.makedirs(subfolder_path, exist_ok=True)
    os.rename(file_path, os.path.join(subfolder_path, filename))

Answer 2

這是我使用pathlib重命名的解決方案；）假設當前腳本是您的文件和文件夾的路徑。

# using pathlip
from collections import defaultdict
from pathlib import Path

TARGET_DIR = Path('.') # script dir

FILES_AND_FOLDERS = TARGET_DIR.rglob('*')

# storage with files and folders 
storage = defaultdict(list)
for blob in FILES_AND_FOLDERS:
    if blob.is_file():
        storage['files'].append(blob)
    elif blob.is_dir():
        storage['dir'].append(blob)

# for each dir check if file first 3 characters 
# match if so move by rename
    
for directory in storage['dir']:
    for file in storage['files']:
        if file.name[:3] == directory.name[:3]:
            file.rename(directory / file)
    
# you can look at shutil too