Как получить ссылку на функцию из класса?

Question

[FOUREX]

class Aliases:
    @property
    def commands(self) -> tuple:
        return (
            self.new_alias,
        )

    @command(
        command="новый алиас",
        args={"алиас": "str"}
    )
    def new_alias(self, message: types.Message):
        ...

Есть класс Aliases, а в нём функция commands которая возвращает некоторые функции из класса. Если функция new_alias будет без декоратора command, то функция commands возвращает ссылку на функцию (то что мне нужно), но если функция new_alias будет с декоратором command, то функция commands возвращает аргументы декоратора ({'command': 'новый алиас', 'args': {'алиас': 'str'}}), без ссылки на функцию. Как я могу получить таким образом ссылку на функцию и аргументы из декоратора command?

Comments

[User]

Можно код декоратора?

[FOUREX]

User, Ничего особенного.

def command(**kwargs):
    def wrapper(*args):
        return kwargs

    return wrapper

Просто для того то бы получать из него информацию

[User]

Либо я чего-то не понимаю, либо вы не договариваете, либо делаете что-то странное. У вас тут внутри декоратора не происходит вызов или даже возврат decorated функции...

[FOUREX]

User, Да всё я сказал, можете у себя попробовать посмотреть.

[User]

FOUREX, Не знаю какую проблему вы на самом деле пытаетесь решить (почитатйе про xy problem чтоль...) но ответ я вам дал. любителям поизвращаться

[shurshur]

FOUREX, код делает то, что в нём написано.

self.new_alias будет ссылкой не на исходную функцию new_alias, а на функцию wrapper из декоратора. Эта функция возвращает те значения, которые переданы в параметрах декоратора. Исходную функцию она не вызывает. При вызове декорированной функции будут, конечно, возвращены значения из декоратора - что и наблюдается в реальности.

Если грубо представить, то декоратор заменяет исходную функцию другой функцией wrapper, которая может вызвать, а может и не вызвать исходную. Сама функция декоратора при этом вызывается один раз - в момент описания класса.

Например, можно завести в классе статический список Aliases.commands_list (общий на все экземпляры класса), а декоратор будет в него добавлять. Примерно так:

def command(function, *, name:str, args:dict):
    def wrapper(*args, **kwargs):
        return function(*args, **kwargs)
    Aliases.commands_list.append({"name":name,"args":args})
    return wrapper

class Aliases:
    commands_list = []

    @property
    def commands(self) -> tuple:
        return tuple(self.commands_list)

    @command(
        command="новый алиас",
        args={"алиас": "str"}
    )
    def new_alias(self, message: types.Message):
        ...

Я бы посоветовал посмотреть, как подобную задачу решают взрослые библиотеки с декораторами такого типа (обработчики команд для telegram, для discord...).

[FOUREX]

shurshur, Ваш ответ подтолкнул меня на вот такое решение

def command(**kwargs):
    def wrapper(*args):
        return args[0], kwargs

    return wrapper


class Aliases:
    def __init__(self, bot: Bot):
        self.bot = bot

    @property
    def commands(self) -> tuple:
        return (
            self.new_alias,
        )

    @command(
        name="новый алиас",
        args={"алиас": "str"}
    )
    def new_alias(self, message: types.Message = None):
        ...


def setup(bot: Bot, dp: Dispatcher):
    aliases = Aliases(bot)

    print(aliases.commands)

Это мне позволило мне получить и ссылку на функцию, и аргументы декоратора

((Тут ссылка, но она почему-то не отображается на сайте, {'name': 'новый алиас', 'args': {'алиас': 'str'}}),)

Но теперь другая проблема, при вызове функции пишет в консоль что пропущен первый аргумент "self".

File "/home/fourex/PycharmProjects/pythonProject/cogs/aliases.py", line 30, in setup
function[0]()
TypeError: Aliases.new_alias() missing 1 required positional argument: 'self

Изменит немного код функции setup

def setup(bot: Bot, dp: Dispatcher):
    aliases = Aliases(bot)

    for function in aliases.commands:
        function[0]()

Answer 1

[FOUREX]

def command(**kwargs):
    def wrapper(func_ref):
        func_ref.command_kwargs = kwargs
        return func_ref

    return wrapper


class Aliases:
    def __init__(self, bot: Bot):
        self.bot = bot

    @property
    def commands(self) -> tuple:
        return (
            self.new_alias,
        )

    @command(
        name="новый алиас",
        args={"алиас": "str"}
    )
    def new_alias(self, message: types.Message = None):
        ...


def setup(bot: Bot, dp: Dispatcher):
    aliases = Aliases(bot)

    for function in aliases.commands:
        print(function, function.command_kwargs)

Вывод в консоль:

<\bound method Aliases.new_alias of <\aliases.py.Aliases object at 0x7f4b0d0c2560>> {'name': 'новый алиас', 'args': {'алиас': 'str'}}

Answer 2

[User]

осторожно

from functools import wraps, partial

def command(**cmdargs):
    def _run(func):
        @wraps(func)
        def wrapper(self, *args):
            return partial(func, self, args, cmdargs)
        return wrapper
    return _run

class Aliases:
    @property
    def commands(self) -> tuple:
        return (
            self.new_alias(),
        )

    @command(
        command="новый алиас",
        args={"алиас": "str"}
    )
    def new_alias(self, *args):
        for arg in args:
            if isinstance(arg, str):  # заменить на types.Message):
                return arg
        return args

>>> aliases = Aliases()
>>> aliases.commands
(functools.partial(<function Aliases.new_alias at 0x778154c550>, <__main__.Aliases object at 0x778205af50>, (), {'command': 'новый алиас', 'args': {'алиас': 'str'}}),)
>>> aliases.commands[0]
functools.partial(<function Aliases.new_alias at 0x778154c550>, <__main__.Aliases object at 0x778205af50>, (), {'command': 'новый алиас', 'args': {'алиас': 'str'}})
>>> aliases.commands[0].args[2]
{'command': 'новый алиас', 'args': {'алиас': 'str'}}
>>> aliases.commands[0]("types.Message object")
'types.Message object'