Tạo và xâu chuỗi các hàm trang trí trong Python

Tăng cường các hàm Python với Trình trang trí

Trong Python, trình trang trí là một công cụ mạnh mẽ để sửa đổi hành vi của các hàm hoặc phương thức. Chúng cho phép các nhà phát triển bao bọc chức năng bổ sung xung quanh chức năng hiện có một cách rõ ràng và dễ đọc. Bằng cách hiểu cách tạo và xâu chuỗi các trình trang trí, bạn có thể nâng cao đáng kể tính mô-đun và khả năng đọc mã của mình.

Bài viết này sẽ hướng dẫn bạn quy trình tạo hai công cụ trang trí cụ thể: một để in đậm văn bản và một để làm văn bản in nghiêng. Chúng tôi cũng sẽ trình bày cách xâu chuỗi các trang trí này để đạt được kết quả mong muốn. Đến cuối hướng dẫn này, bạn sẽ có thể gọi một hàm đơn giản và nhận một chuỗi được định dạng bằng cả thẻ HTML in đậm và in nghiêng.

Tìm hiểu về trang trí hàm Python

Các tập lệnh được cung cấp trình bày cách tạo và xâu chuỗi các trình trang trí hàm trong Python để sửa đổi hành vi của các hàm. Trình trang trí trong Python được xác định bằng cách sử dụng def từ khóa để tạo một hàm lấy hàm khác làm đối số và trả về một hàm mới. Các make_bold trình trang trí bao bọc kết quả của hàm mà nó trang trí bằng các thẻ in đậm HTML bằng cách sử dụng định dạng chuỗi f: f"<b>{func()}</b>". Tương tự, các make_italic trang trí gói kết quả trong các thẻ in nghiêng: f"<i>{func()}</i>". Khi các trang trí này được áp dụng cho một hàm bằng cách sử dụng @decorator_name cú pháp, chúng sửa đổi kết quả đầu ra của hàm bằng cách thêm các thẻ HTML tương ứng.

Tập lệnh thứ hai giới thiệu một cách tiếp cận linh hoạt hơn bằng cách tạo hàm bậc cao hơn, add_html_tag, tạo ra các trình trang trí cho bất kỳ thẻ HTML được chỉ định nào. Hàm này lấy thẻ HTML làm đối số và trả về một trình trang trí bao bọc đầu ra của hàm trong thẻ đã chỉ định: f"<{tag}>{func()}</{tag}>". Bằng cách sử dụng số 8 Và @add_html_tag("i"), chúng ta có thể xâu chuỗi các trang trí này để bao bọc đầu ra của say_hello hoạt động ở cả thẻ in đậm và in nghiêng, mang lại kết quả "Xin chào". Những ví dụ này minh họa sức mạnh và tính linh hoạt của trình trang trí Python trong việc nâng cao và tùy chỉnh hành vi của hàm một cách rõ ràng và có thể tái sử dụng.

def make_bold(func):
    def wrapper():
        return f"<b>{func()}</b>"
    return wrapper

def make_italic(func):
    def wrapper():
        return f"<i>{func()}</i>"
    return wrapper

@make_bold
@make_italic
def say():
    return "Hello"

print(say())

def add_html_tag(tag):
    def decorator(func):
        def wrapper():
            return f"<{tag}>{func()}</{tag}>"
        return wrapper
    return decorator

@add_html_tag("b")
@add_html_tag("i")
def say_hello():
    return "Hello"

print(say_hello())

Kỹ thuật trang trí Python nâng cao

Ngoài việc sửa đổi chức năng đơn giản, trình trang trí Python còn cung cấp một cách mạnh mẽ để nâng cao khả năng sử dụng lại và bảo trì mã. Một trường hợp sử dụng nâng cao là các trình trang trí được tham số hóa, cho phép các trình trang trí chấp nhận các đối số. Kỹ thuật này được minh họa bằng add_html_tag trang trí trong các ví dụ trước. Bằng cách xác định một trình trang trí tạo ra các trình trang trí khác, chúng ta có thể tạo các cấu trúc mã rất linh hoạt và có thể tái sử dụng. Trình trang trí được tham số hóa cho phép chúng ta truyền tham số cho chính trình trang trí đó, cho phép sửa đổi động và theo ngữ cảnh cụ thể đối với hành vi của hàm.

Một khía cạnh quan trọng khác của trình trang trí là khả năng duy trì siêu dữ liệu chức năng. Khi một hàm được bao bọc bởi một trình trang trí, siêu dữ liệu của nó, chẳng hạn như tên và chuỗi tài liệu, có thể bị mất. Để bảo toàn siêu dữ liệu này, Python functools.wraps được sử dụng trong trang trí. Bằng cách áp dụng @functools.wraps đối với hàm bao bọc, siêu dữ liệu của hàm ban đầu sẽ được sao chép lại, đảm bảo rằng các công cụ dựa vào siêu dữ liệu này, chẳng hạn như trình tạo tài liệu, tiếp tục hoạt động chính xác. Ngoài ra, các vật trang trí có thể được xếp chồng lên nhau, như minh họa với @make_bold Và @make_italic ví dụ, để áp dụng nhiều lớp sửa đổi hành vi một cách rõ ràng và dễ đọc.