Cách lấy tên tệp hiệu quả trong Python: mô-đun os và pathlib

1. Giới thiệu

2. Cơ bản về cách lấy tên tệp trong Python

2.1 Cách sử dụng os.path.basename()

import os

file_path = "/user/home/document/file.txt"
file_name = os.path.basename(file_path)
print(file_name)  # Kết quả: "file.txt"

2.2 Lấy tên thư mục bằng os.path.dirname()

dir_name = os.path.dirname(file_path)
print(dir_name)  # Kết quả: "/user/home/document"

3. Lấy tất cả tên tệp trong thư mục

3.1 Lấy tên tệp bằng os.listdir()

import os

dir_path = "/user/home/document"
files = os.listdir(dir_path)
print(files)  # Kết quả: ["file1.txt", "file2.txt", "subfolder"]

3.2 Lấy tên tệp một cách đệ quy với os.walk()

import os

dir_path = "/user/home/document"
for root, dirs, files in os.walk(dir_path):
    for file in files:
        print(os.path.join(root, file))

3.3 Thêm xử lý lỗi

dir_path = "/invalid/path"
try:
    files = os.listdir(dir_path)
    print(files)
except FileNotFoundError:
    print(f"Không tìm thấy {dir_path}.")

4. Lấy và thay đổi phần mở rộng tệp

4.1 Lấy phần mở rộng bằng os.path.splitext()

import os

file_path = "/user/home/document/file.txt"
file_name, file_ext = os.path.splitext(file_path)
print(file_name)  # Kết quả: "/user/home/document/file"
print(file_ext)   # Kết quả: ".txt"

4.2 Thay đổi phần mở rộng

new_file_path = file_name + ".jpg"
print(new_file_path)  # Kết quả: "/user/home/document/file.jpg"

5. Ứng dụng lấy tên tệp bằng mô-đun pathlib

5.1 Lấy tên tệp bằng Path.iterdir()

from pathlib import Path

dir_path = Path("/user/home/document")
for item in dir_path.iterdir():
    print(item.name)

5.2 Lấy tệp đệ quy: rglob()

for file in dir_path.rglob("*"):
    print(file)

6. Lọc tên tệp

6.1 Lọc theo phần mở rộng

from pathlib import Path

dir_path = Path("/user/home/document")
for file in dir_path.glob("*.txt"):
    print(file)

7. Tổng kết