以下是一个通用的包装函数作为日志记录器的示例代码:
import logging
def logger(func):
def wrapper(*args, **kwargs):
logging.basicConfig(filename='log.txt', level=logging.INFO)
logging.info(f"Calling function: {func.__name__}")
try:
result = func(*args, **kwargs)
logging.info(f"Function {func.__name__} executed successfully.")
return result
except Exception as e:
logging.error(f"Error in function {func.__name__}: {str(e)}")
return wrapper
@logger
def example_function(input):
# Example function to be wrapped
return input + 1
result = example_function(5)
print(result)
在上面的示例中,我们首先导入了logging
模块。然后定义了一个名为logger
的装饰器函数,它接受一个函数作为参数。
装饰器函数logger
内部定义了一个内部函数wrapper
。wrapper
函数使用logging.basicConfig
方法来设置日志记录器的基本配置,将日志记录到名为log.txt
的文件中,并将日志级别设置为INFO
。
在wrapper
函数中,我们首先记录了被装饰函数的名称,以及在函数执行之前的日志信息。然后,我们尝试执行被装饰的函数,并捕获任何可能的异常。如果函数执行成功,我们记录一个成功的日志消息,否则记录一个错误的日志消息。
最后,我们将被装饰函数example_function
用@logger
装饰,这样每次调用example_function
时,都会通过日志记录器记录相关信息。
运行以上示例代码,会在当前目录下生成一个名为log.txt
的日志文件,其中包含了每次调用example_function
时的日志记录信息。