import os

def list_files_in_directory(directory):
    # 使用os.listdir列出指定目錄中的所有文件和子目錄
    files_and_directories = os.listdir(directory)
    # 篩選出文件（不是子目錄）
    only_files = [file for file in files_anddirectories if os.path.isfile(os.path.join(directory, file))]
    return only_files

# 示例使用
my_directory = 'C:/Users/username/Documents'
files = list_files_in_directory(my_directory)
print("以下是文件列表：")
for f in files:
    print(f)

# 輸出：將顯示指定目錄的所有文件名稱。

def sort_files_by_extension(directory):
    from collections import defaultdict
    files_dict = defaultdict(list)  # 創(chuàng)建默認(rèn)字典用于存儲(chǔ)不同擴(kuò)展名的文件
    
    for file_name in os.listdir(directory):
        if os.path.isfile(os.path.join(directory, file_name)):  # 檢查是否是文件
            _, ext = os.path.splitext(file_name)  # 分割文件名和擴(kuò)展名
            if ext:  # 如果有擴(kuò)展名，則加入到相應(yīng)的鍵中
                files_dict[ext].append(file_name)

    return dict(files_dict)

# 示例使用
my_directory = "C:/temp"
files_sorted = sort_files_by_extension(my_directory)

# 打印分類后的結(jié)果
for extension, files_list in files_sorted.items():
    print(f"文件類型 {extension}:")
    for fname in files_list:
        print(fname)

import shutil

def move_files_based_on_extension(source_dir, destination_base_dir):
    sorted_files = sort_files_by_extension(source_dir)
    
    for ext, filenames in sorted_files.items():
        destination_folder = os.path.join(destination_base_dir, ext[1:])  # 忽略點(diǎn)號(hào)
        
        if not os.path.exists(destination_folder):  # 如果不存在則創(chuàng)建
            os.makedirs(destination_folder)
        
        for filename in filenames:
            full_path = os.path.join(source_dir, filename)
            new_path = os.path.join(destination_folder, filename)
            shutil.move(full_path, new_path)  # 移動(dòng)文件
            print(f"已將文件{filename}移動(dòng)到{new_path}")

# 示例使用
move_files_based_on_extension('C:/Source', 'C:/Destination')

def get_total_size_of_files(directory, filenames=None):
    total_size = 0
    
    if filenames is None:  # 如果沒有指定文件列表，則讀取全部文件
        filenames = os.listdir(directory)

    for filename in filenames:
        filepath = os.path.join(directory, filename)
        if os.path.isfile(filepath):  # 只考慮實(shí)際存在的文件
            total_size += os.path.getsize(filepath)

    return total_size

# 示例
size = get_total_size_of_files('C:/SourceFiles')
print(f"該目錄下文件總大小為： {size/(1024**2)} MB")  # 轉(zhuǎn)換為MB

from datetime import datetime, timedelta

def remove_old_files(path_to_scan, days_limit=30):
    critical_time = datetime.now() - timedelta(days=days_limit)

    for root, dirs, files in os.walk(path_to_scan):
        for file in files:
            file_full_path = os.path.join(root, file)
            mod_time = datetime.fromtimestamp(os.path.getmtime(file_full_path))
            if mod_time < critical_time:  # 超過限制日期，刪除
                print(f"刪除文件：{file_full_path}")
                os.remove(file_full_path)

# 使用示例
remove_old_files('C:/OldData', 365)  # 刪除一年以上的文件

import zipfile

def compress_directory(dir_to_zip, output_filename='archive'):
    # 創(chuàng)建壓縮包實(shí)例
    zipf = zipfile.ZipFile(output_filename + '.zip', 'w', zipfile.ZIP_DEFLATED)
    
    # 遍歷并添加所有文件到zip包內(nèi)
    for root, dirs, files in os.walk(dir_to_zip):
        for file in files:
            zipf.write(os.path.join(root, file), 
                       os.path.relpath(os.path.join(root, file), 
                                        os.path.join(dir_to_zip, '..')))

    zipf.close()

# 使用
compress_directory('C:/BackupThis', 'backup_archive')