Можно ли сделать декоратор, работающий и с асинхронными, и с синхронными функциями?

Question

[Max Payne]

Допустим, у есть просто декоратор:

def if_no_data(f):
    def wrapper(self, *args, **kwargs):
        if not self.data:
            return f(self, *args, **kwargs)

    return wrapper

Суть его работы проста - проверить на пустоту свойство объекта. Отлично, а если нужно чтобы декоратор работал как с асинхронными, так и с синхронными функциями?
Что-то такое выглядит очень странно, да и вряд ли работать будет.

def if_no_data(f):
    async def wrapper(self, *args, **kwargs):
        if not self.data:
            result = f(self, *args, **kwargs)

            try:
                return await result
            except ValueError:
                return result

    return wrapper

Answer 1

[Александр Брагин]

Что-то типа того

def if_no_data(f):
    async def wrapper(self, *args, **kwargs):
        if self.data:
            return None

        if asyncio.iscoroutinefunction(f):
            return await f(self, *args, **kwargs)
        else:
            return f(self, *args, **kwargs)

    return wrapper

Comments

[Max Payne]

А, про iscoroutinefunction не знал, думал как же извернуться. Но тут второй вопрос возникает -- как вызывать декорированную функцию?

допустим:

@if_no_data
def a(): pass

@if_no_data
async def b(): pass

всегда будет

await a()
await b()

?

[Александр Брагин]

Max Payne, ага, т.к. враппер в декораторе - корутина