Передать в декоратор метода класса сам класс

Question

[alhimik45]

Вот простой декоратор:

def deco(clazz):
	def dec(fn):
		print clazz,fn
		return fn
	return dec

Пытаемся его использовать:

class Foo(object):
	@deco(Foo)
	def Bar():
		print "bar"

При запуске пишет:

Traceback (most recent call last):
  File "<pyshell#12>", line 1, in <module>
    class Foo(object):
  File "<pyshell#12>", line 2, in Foo
    @deco(Foo)
NameError: name 'Foo' is not defined

Собственно, почему он не видит класс и как это можно обойти?

Comments

[iroln]

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

Answer 1

[Алексей Акулович]

Потому что в момент выполнения тела класса, он еще не сформирован.
Можно например так:

import traceback

def deco():
    clazz = traceback.extract_stack()[1][2]
    def dec(fn):
        print clazz,fn
        return fn
    return dec

class Foo(object):
    @deco()
    def Bar():
        print "bar"

Может кто что получше предложит, что-то голова не варит.

Comments

[alhimik45]

Этот код мне выводит:

main <function Bar at 0x0134E8F0>

[Алексей Акулович]

У вас выходит первый индекс тут
traceback.extract_stack()[ 1 ][2]
не подошел.

Корректнее будет (не с той стороны брал)
clazz = traceback.extract_stack()[ -2 ][2]

[alhimik45]

Спасибо, так работает.
Появилась такая идея: нельзя ли как-то узнать какому классу принадлежит метод:

def dec(fn):
        print # тут что-то, чтобы получить класс метода
        print fn
        return fn

[Алексей Акулович]

Так сейчас clazz и получает имя класса, в котором используется декоратор.

Вы бы написали что в итоге хотите получить :) «Вот код, вот тут он должен как-то получить строку XXX».

[alhimik45]

В общем, мне нужно получить объект класса в декораторе и добавить ему пару свойств.

А

clazz = traceback.extract_stack()[ -2 ][2]

Дает только имя класса, а при попытке eval'а дает то же самое:

NameError: name 'Foo' is not defined

[Алексей Акулович]

Я ж написал, что пока выполнение кода находится внутри описания класса — его еще не существует.
Как-то вам нужно свою задачу изменить.

[alhimik45]

Ладно, спасибо. Пойду другим путем

[JustAMan]

Возможно, вам нужен декоратор уровня класса, а не уровня метода.
Ну или метакласс :)

Answer 2

[xSus]

Декоратор уровня метода — решает вопросы самого метода. И не корректно пытаться расширить его на уровень класса. Поймите что метод может быть задекларирован где угодно и в любом другом методе присвоен любому классу. Что в этом случае должен выдать декоратор? Такой уж питон. Надо попробовать переформулировать задачу, возможно решение вовсе не в декораторах?

Comments

[JustAMan]

При чём здесь эти возможности? Как часто вообще метод класса подсовывается снаружи? Я понимаю, как всё это работает, более того, я этим пользуюсь. Но понимание кода это очень сильно затрудняет.
А вообще декоратор метода вполне может брать первый аргумент, пришедший в функцию, и считать его объектом self.

Тут на самом деле проблема в том, что такая форма (декоратор внутри декларации класса) оборачивает не метод, а функцию. Которая про класс, в принципе, ни сном ни духом (помимо аргумента self).

Answer 3

[JustAMan]

Похоже, то, что реализует ответ на поставленный вопрос — это вот этот пример:
wiki.python.org/moin/PythonDecoratorLibrary#CA-1f74040af898602b483f2d7691d29b4a5644cce2_1

Answer 4

[homm]

> мне нужно получить объект класса в декораторе и добавить ему пару свойств. Для этого больше подойдут метаклассы. В любом случае вы не получите объект класса в момент вызова декоратора его метода — класс еще не создан.