溫馨提示×

Python裝飾器如何降低開發(fā)成本

小樊
81
2024-11-09 12:47:45
欄目: 編程語言

Python裝飾器是一種強(qiáng)大的工具，可以幫助我們減少重復(fù)代碼，提高代碼的可讀性和可維護(hù)性。以下是一些使用裝飾器降低開發(fā)成本的方法：

  1. 代碼復(fù)用：裝飾器允許我們在不修改原始函數(shù)的情況下，為函數(shù)添加新的功能。這可以減少重復(fù)代碼，提高代碼的可讀性和可維護(hù)性。
def my_decorator(func):
    def wrapper():
        print("Before the function is called.")
        func()
        print("After the function is called.")
    return wrapper

@my_decorator
def say_hello():
    print("Hello!")

say_hello()
  1. 日志記錄：裝飾器可以用于記錄函數(shù)的調(diào)用日志，幫助我們更好地了解程序的運(yùn)行情況。
import logging

logging.basicConfig(level=logging.INFO)

def log_decorator(func):
    def wrapper(*args, **kwargs):
        logging.info(f"Calling {func.__name__} with args: {args} and kwargs: {kwargs}")
        result = func(*args, **kwargs)
        logging.info(f"{func.__name__} returned: {result}")
        return result
    return wrapper

@log_decorator
def add(a, b):
    return a + b

add(1, 2)
  1. 權(quán)限驗(yàn)證：裝飾器可以用于驗(yàn)證用戶權(quán)限，確保只有具有特定權(quán)限的用戶才能訪問受保護(hù)的資源。
def permission_required(permission):
    def decorator(func):
        def wrapper(user, *args, **kwargs):
            if user.has_permission(permission):
                return func(user, *args, **kwargs)
            else:
                raise PermissionError("Permission denied")
        return wrapper
    return decorator

class User:
    def __init__(self, name, permissions):
        self.name = name
        self.permissions = permissions

    def has_permission(self, permission):
        return permission in self.permissions

@permission_required("admin")
def delete_user(user, user_id):
    print(f"Deleting user with ID: {user_id}")

user = User("Alice", ["admin"])
delete_user(user, 1)
  1. 緩存：裝飾器可以用于緩存函數(shù)的結(jié)果，避免重復(fù)計(jì)算，提高程序的性能。
def memoize(func):
    cache = {}
    def wrapper(*args):
        if args in cache:
            return cache[args]
        result = func(*args)
        cache[args] = result
        return result
    return wrapper

@memoize
def fibonacci(n):
    if n <= 1:
        return n
    return fibonacci(n - 1) + fibonacci(n - 2)

print(fibonacci(10))

通過使用裝飾器，我們可以更簡潔地實(shí)現(xiàn)這些功能，從而降低開發(fā)成本。

0 踩