Python怎么将两个功能相同的装饰器高阶函数，一个带参数一个不带参数的合并成一个装饰器？请前辈指点一二，十分感谢？

问题描述

问 题

Python的装饰器怎么将两个功能相同的函数，一个带参数一个不带参数的合并成一个函数？

获取本地时间方法

def get_now_local_time():
    return time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))

装饰器1

def log1(func):
    @functools.wraps(func)  # 保持传入的函数名称不被返回函数改变
    def wrapper(*args, **kwargs):
        print('%s %s():' % ("begin call", func.__name__))
        call_func_ref = func(*args, **kwargs)
        print('%s %s():' % ("end call", func.__name__))
        return call_func_ref
    return wrapper

@log1       # 为函数添加无参数装饰器
def now():
    print('Current local time：' + get_now_local_time())
now()

无参装饰器函数调用输出

begin call now():
Current local time：2016-10-12 23:51:11
end call now():
    

装饰器2

def log2(text=None):
    def decorator(func):
        @functools.wraps(func)  # 保持传入的函数名称不被返回函数改变
        def wrapper(*args, **kwargs):
            print('%s %s():' % (("begin " + text), func.__name__))
            call_func_ref = func(*args, **kwargs)
            print('%s %s():' % (("end " + text), func.__name__))
            return call_func_ref
        return wrapper
    return decorator

@log2('execute current the function')    # 为函数添加带参数装饰器
def now():
    print('Current local time：' + get_now_local_time())
now()

带有参装饰器函数调用输出

begin execute current the function now():
Current local time：2016-10-12 23:49:25
end execute current the function now():

想做完成这两个函数的合并编码，该怎么写？请大牛前辈给指点一下呗，谢谢~~~

解决方案

# -*- coding: utf-8 -*-
import functools
import time

def get_now_local_time():
    return time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))


def log(text):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*arg, **kwarg):
            print('%s, %s()' % ('begin call',func.__name__) if callable(text) else \
             '%s %s():' % (("begin " + text), func.__name__))
            func(*arg, **kwarg)
            print('%s, %s()' % ('end call', func.__name__) if callable(text) else \
             '%s %s():' % (("end " + text), func.__name__))
        return wrapper
    return decorator(text) if callable(text) else decorator

@log
def now():
    print('Current local time：' + get_now_local_time())
now()

print('==' * 50)

@log('execute current the function')    # 为函数添加带参数装饰器
def now():
    print('Current local time：' + get_now_local_time())
now()

需求是不是这样子的

callable（object）的文档解释

这篇关于Python怎么将两个功能相同的装饰器高阶函数，一个带参数一个不带参数的合并成一个装饰器？请前辈指点一二，十分感谢？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！