Класс Decorator для проверки необходимых переменных класса - PullRequest
Новая
       17

Класс Decorator для проверки необходимых переменных класса

0 голосов
/ 10 марта 2010

Прежде всего, я не знаю, правильный ли это подход. Я хочу написать класс декоратора, который будет использоваться с методами другого класса. Перед запуском метода я хотел бы проверить, все ли переменные класса инициализированы. В идеальном случае было бы что-то похожее на это: 

class Test(object):
    def __init__(self, f):
        self.f = f
        # some magic

    def __call__(self):
        self.f.magically_get_variable(required)
        # do some checks and execute method or throw an exception

class Data(object):
    def __init__(self, a, b):
        self.a = a

    @test
    def sum(self):
        required('self.a', 'self.b')
        return self.a + self.b

Если это не так, пожалуйста, посоветуйте мне, как это сделать правильно.

Ответы [ 2 ]

1 голос
/ 10 марта 2010

Я бы сказал, что декораторы здесь не годятся для проверки, существует ли переменная.

Подумайте о том, что вы планируете делать, если необходимые переменные не указаны: выведите исключение (это в вашем комментарии).

Я бы сказал (на основе кода выше): 

def sum(self):
    return self.a + self.b

И пусть произойдет сбой , если self.a или self.b не предоставлены (это вызовет исключение NameError, вы можете поймать его, если хотите, и вы можете поднять свой собственный, если хочешь.)

0 голосов
/ 10 марта 2010

Если вы настаиваете на проверке перед запуском тела декорированного метода, один из разумных способов сделать это ...: 

import functools

class test(object):
    def __init__(self, requiredvars):
        self.reqs = requiredvars.split()

    def __call__(self, f):
        @functools.wraps(f)
        def wrapper(wself, *a, **k):
          missing = set()
          for v in self.reqs:
            if not hasattr(wself, v):
              missing.add(v)
          if missing:
            msg = 'missing fields: %s' % ','.join(sorted(missing))
            raise valueerror, msg
          return f(wself, *a, **k)
        return wrapper

class data(object):
    def __init__(self, a, b):
        self.a = a

    @test('a b')
    def sum(self):
        return self.a + self.b

d = data(23, 42)
d.sum()

Это излучает, как вы, очевидно, хотите, 

Traceback (most recent call last):
  File "rev.py", line 29, in <module>
    d.sum()
  File "rev.py", line 16, in wrapper
    raise ValueError, msg
ValueError: Missing fields: b
...