Cách tạo và xóa thư mục bằng Python: os vs pathlib

1. Cách cơ bản để tạo thư mục trong Python

Cách tạo thư mục trong Python: mô-đun os

import os

# Chỉ định đường dẫn của thư mục cần tạo
path = 'example_folder'

# Tạo thư mục
os.mkdir(path)

Tạo thư mục nhiều tầng: os.makedirs()

import os

# Chỉ định thư mục nhiều tầng
path = 'example_folder/subfolder'

# Tạo thư mục ngay cả khi thư mục cha không tồn tại
os.makedirs(path)

Xử lý lỗi: dùng exist_ok=True

import os

path = 'example_folder/subfolder'

# Bỏ qua lỗi ngay cả khi thư mục đã tồn tại
os.makedirs(path, exist_ok=True)

2. Tạo thư mục bằng mô-đun Pathlib

Cách sử dụng mô-đun Pathlib

from pathlib import Path

# Chỉ định đường dẫn thư mục cần tạo
path = Path('example_folder/subfolder')

# Tạo thư mục
path.mkdir(parents=True, exist_ok=True)

Ưu điểm của mô-đun Pathlib

3. Xử lý lỗi khi tạo thư mục

Đối phó với lỗi quyền hạn và đường dẫn không hợp lệ

import os

path = 'example_folder/subfolder'

try:
    os.makedirs(path, exist_ok=True)
    print(f'Thư mục "{path}" đã được tạo.')
except PermissionError:
    print('Bạn không có quyền tạo thư mục.')
except FileNotFoundError:
    print('Đường dẫn được chỉ định không hợp lệ.')
except Exception as e:
    print(f'Đã xảy ra lỗi không mong đợi: {e}')

Xử lý lỗi nâng cao

4. Cách xóa thư mục

Xóa thư mục bằng os.rmdir()

import os

# Chỉ định đường dẫn của thư mục cần xóa
path = 'example_folder/subfolder'

# Xóa thư mục
os.rmdir(path)

Xóa đệ quy thư mục bằng shutil.rmtree()

import shutil

# Chỉ định đường dẫn của thư mục cần xóa
path = 'example_folder/subfolder'

# Xóa thư mục và toàn bộ nội dung của nó
shutil.rmtree(path)

Xử lý lỗi

import shutil

path = 'example_folder/subfolder'

try:
    shutil.rmtree(path)
    print(f'Thư mục "{path}" đã được xóa.')
except PermissionError:
    print('Bạn không có quyền xóa thư mục.')
except FileNotFoundError:
    print('Không tìm thấy thư mục đã chỉ định.')
except Exception as e:
    print(f'Đã xảy ra lỗi ngoài dự kiến: {e}')

5. Các trường hợp ứng dụng thực tế

Tạo thư mục theo năm/tháng

import os
from datetime import datetime

# Lấy năm và tháng hiện tại
current_year = datetime.now().year
current_month = datetime.now().month

# Thiết lập đường dẫn thư mục
folder_path = f'data/{current_year}/{current_month}'

# Tạo thư mục
os.makedirs(folder_path, exist_ok=True)
print(f'Thư mục "{folder_path}" đã được tạo.')

Xóa thư mục hàng loạt

import shutil
import os
from datetime import datetime, timedelta

# Lấy ngày của 30 ngày trước
threshold_date = datetime.now() - timedelta(days=30)

# Đường dẫn cơ sở để xóa thư mục
base_path = 'data/'

# Kiểm tra thư mục và xóa các thư mục cũ
for folder_name in os.listdir(base_path):
    folder_path = os.path.join(base_path, folder_name)
    if os.path.isdir(folder_path):
        folder_date = datetime.strptime(folder_name, '%Y-%m-%d')
        if folder_date < threshold_date:
            shutil.rmtree(folder_path)
            print(f'Thư mục cũ "{folder_path}" đã được xóa.')

6. Tổng kết