Thông báo: Download 4 khóa học Python từ cơ bản đến nâng cao tại đây.
Decorator với tham số trong Python
Trong Python, decorator là một công cụ mạnh mẽ cho phép bạn mở rộng hoặc thay đổi hành vi của các hàm mà không cần chỉnh sửa mã nguồn của chúng. Để tăng tính linh hoạt, bạn có thể sử dụng decorator với các tham số thông qua việc sử dụng decorator factory. Trong bài viết này, bạn sẽ học cách định nghĩa các decorator với tham số trong Python, giúp bạn tạo ra những đoạn mã dễ bảo trì và tái sử dụng hơn.

Giới thiệu về decorator với tham số trong Python
Giả sử bạn có một hàm say để in ra một thông điệp:
def say(message):
    ''' in ra thông điệp
    Arguments:
        message: thông điệp cần hiển thị
    '''
    print(message)
Bạn muốn hàm say() in ra thông điệp 5 lần mỗi khi bạn gọi nó. Ví dụ:
say('Hi')
Kết quả sẽ hiển thị thông điệp "Hi" năm lần như sau:
Bài viết này được đăng tại [free tuts .net]
Hi Hi Hi Hi Hi
Để làm điều đó, bạn có thể sử dụng một decorator thông thường:
@repeat
def say(message):
    ''' in ra thông điệp
    Arguments:
        message: thông điệp cần hiển thị
    '''
    print(message)
Và bạn có thể định nghĩa decorator repeat như sau:
def repeat(fn):
    @wraps(fn)
    def wrapper(*args, **kwargs):
        for _ in range(5):
            result = fn(*args, **kwargs)
        return result
    return wrapper
Đoạn mã hoàn chỉnh như sau:
from functools import wraps
def repeat(fn):
    @wraps(fn)
    def wrapper(*args, **kwargs):
        for _ in range(5):
            result = fn(*args, **kwargs)
        return result
    return wrapper
@repeat
def say(message):
    ''' in ra thông điệp
    Arguments:
        message: thông điệp cần hiển thị
    '''
    print(message)
say('Hello')
Nhưng nếu bạn muốn hàm say() được thực thi 10 lần thay vì 5 lần? Trong trường hợp này, bạn cần thay đổi giá trị 5 được hard-coded trong decorator repeat. Tuy nhiên, giải pháp này không linh hoạt. Ví dụ, bạn muốn sử dụng decorator repeat để thực thi một hàm 5 lần và một hàm khác 10 lần. Decorator repeat sẽ không đáp ứng yêu cầu này.
Để khắc phục, bạn cần thay đổi decorator repeat để nó chấp nhận một tham số chỉ định số lần mà một hàm nên được thực thi như sau:
@repeat(5)
def say(message):
    ...
Để định nghĩa decorator repeat, repeat(5) nên trả về decorator gốc.
def repeat(times):
    # trả về decorator gốc "repeat"
Hàm repeat mới này sẽ trả về một decorator. Và nó thường được gọi là decorator factory.
Hàm repeat sau đây trả về một decorator:
def repeat(times):
    ''' gọi một hàm một số lần '''
    def decorate(fn):
        @wraps(fn)
        def wrapper(*args, **kwargs):
            for _ in range(times):
                result = fn(*args, **kwargs)
            return result
        return wrapper
    return decorate
Trong đoạn mã này, hàm decorate là một decorator. Nó tương đương với decorator repeat gốc.
Lưu ý rằng hàm repeat mới không phải là một decorator. Nó là một decorator factory trả về một decorator.
Tích hợp tất cả lại với nhau trong Python
from functools import wraps
def repeat(times):
    ''' gọi một hàm một số lần '''
    def decorate(fn):
        @wraps(fn)
        def wrapper(*args, **kwargs):
            for _ in range(times):
                result = fn(*args, **kwargs)
            return result
        return wrapper
    return decorate
@repeat(10)
def say(message):
    ''' in ra thông điệp
    Arguments:
        message: thông điệp cần hiển thị
    '''
    print(message)
say('Hello')
Kết bài
Sử dụng một decorator factory để trả về một decorator có thể chấp nhận các tham số là một kỹ thuật mạnh mẽ và linh hoạt trong Python. Nó giúp mã của bạn trở nên dễ bảo trì và tái sử dụng hơn, cho phép bạn điều chỉnh hành vi của các hàm một cách linh hoạt mà không cần thay đổi mã nguồn của chúng. Bằng cách hiểu và áp dụng kỹ thuật này, bạn có thể viết các đoạn mã Python tinh gọn, hiệu quả và dễ dàng mở rộng.

            Các kiểu dữ liệu trong C ( int - float - double - char ...)        
            Thuật toán tìm ước chung lớn nhất trong C/C++        
            Cấu trúc lệnh switch case trong C++ (có bài tập thực hành)        
            ComboBox - ListBox trong lập trình C# winforms        
            Random trong Python: Tạo số random ngẫu nhiên        
            Lệnh cin và cout trong C++        
                Cách khai báo biến trong PHP, các loại biến thường gặp            
                Download và cài đặt Vertrigo Server            
                Thẻ li trong HTML            
                Thẻ article trong HTML5            
                Cấu trúc HTML5: Cách tạo template HTML5 đầu tiên            
                Cách dùng thẻ img trong HTML và các thuộc tính của img            
                Thẻ a trong HTML và các thuộc tính của thẻ a thường dùng