如何使这些计数器装饰器对我的类方法起作用

时间:2018-08-22 23:57:01

标签: python decorator class-method

我对装饰者是陌生的,并且相信我大部分时间都了解他们的目的/总体思路。不幸的是,我在使用此装饰器及其在类中的使用时遇到了一些问题。我正在尝试用_profile_dict = {}创建的新键填充字典profile + str(<counter value>)。我宁愿不实例化该类并将其设置为我的if __name__ == '__main'语句中的变量,因此我尝试使用create_profiles函数执行该类。这是我的代码,

_profile_dict = {}


class Profile:
    def __init__(self):
        self.first_name = input('first name: ')
        self.last_name = input('last name: ')
        self.email = input('email: ')
        self.address1 = input('primary address: ')
        self.address2 = input('secondary address: ')
        self.city = input('city: ')
        self.country = input('country: ')
        self.province = input('province: ')
        self.zip = input('zip: ')
        self.phone = input('phone number:')
        self.cc_name = input('Name on credit card: ')
        self.cc_num = input('Credit card number: ')
        self.exp_month = input('card expiration month: ')
        self.exp_year = input('card expiration year: ')
        self.cvv = input('cvv: ')

        self.json_obj = {"utf8": "✓",
                          "_method": "",
                          "authenticity_token": "",
                          "previous_step": "",
                          "checkout[email]": self.email,
                          "checkout[buyer_accepts_marketing]": 1,
                          "checkout[shipping_address][first_name]": self.first_name,
                          "checkout[shipping_address][last_name]": self.last_name,
                          "checkout[shipping_address][address1]": self.address1,
                          "checkout[shipping_address][address2]": self.address2,
                          "checkout[shipping_address][city]": self.city,
                          "checkout[shipping_address][country]": self.country,
                          "checkout[shipping_address][province]": self.province,
                          "checkout[shipping_address][zip]": self.zip,
                          "checkout[shipping_address][phone]": self.phone,
                          "step": ""}

    def counter(self, func):
        def wrapper():
            wrapper.counter += 1
            return func(wrapper.counter)

        wrapper.counter = 0
        return wrapper

    @counter
    def _populate_profile_dict(self, count):
        profile = 'profile'
        _profile_dict[profile + str(count)] = self.json_obj
        print(_profile_dict)


def create_profiles():
    Profile()._populate_profile_dict()

if __name__ == "__main__":
    create_profiles()

此刻,我遇到以下错误

Traceback (most recent call last):
  File "C:/Users/Carsten/Profile_Gen/src/config.py", line 6, in <module>
class Profile:
  File "C:/Users/Carsten/Profile_gen/src/config.py", line 49, in Profile
    @counter
TypeError: counter() missing 1 required positional argument: 'func'

尽管根据我刚才在这里读到的内容,Python decorators in classes,我认为这是不可能的。是否有可能完全通过@classmethod完成我要达到的目标?

仅通过将最后3个函数/方法更改为此,仍然不会产生任何输出:

@classmethod
def counter(cls, func):
    def wrapper():
        wrapper.counter += 1
        return func(wrapper.counter)

    wrapper.counter = 0
    return wrapper

def _populate_profile_dict(self, count):
    profile = 'profile'
    _profile_dict[profile + str(count)] = self.json_obj
    print(_profile_dict)

#def create_profiles():
#    Profile()._populate_profile_dict()


if __name__ == "__main__":
    Profile()._populate_profile_dict = 
    Profile.counter(Profile._populate_profile_dict)

任何帮助将不胜感激。感谢您的时间。我觉得装饰器以及我日程安排中的其他一些事项肯定会帮助我进入python的中间阶段。

2 个答案:

答案 0 :(得分:1)

如果您不需要使用装饰器,则只需测量 您的字典:

def _populate_profile_dict(self):
        _profile_dict['profile' + str(len(_profile_dict))] = self.json_obj
        print(_profile_dict)

答案 1 :(得分:1)

现在让我忽略您的第二次尝试,因为第一次尝试更接近于工作:

def counter(self, func):
    def wrapper():
        wrapper.counter += 1
        return func(wrapper.counter)

    wrapper.counter = 0
    return wrapper

@counter
def _populate_profile_dict(self, count):
    # ...

这不起作用的原因是装饰器语法正在尝试这样做:

_populate_profile_dict = counter(_populate_profile_dict)

…但是counter被定义为接受两个参数,而不是self和一个函数。显然这里没有self。实际上,甚至类本身还不存在,更不用说任何方法了。您只用函数来调用它,所以这就是需要的功能。

如果仅删除该self参数,则装饰器部分将正常工作:

def counter(func):
    def wrapper():
        wrapper.counter += 1
        return func(wrapper.counter)

    wrapper.counter = 0
    return wrapper

现在,counter不再用作类实例上的方法,但这很好,因为您永远不会打算那样使用它。

同时,当仍在定义类时,它作为常规函数非常有用。这正是您要使用它的方式。

counter最终看起来像是Profile实例的公共方法,这可能有点令人困惑,即使实际上不是以这种方式可以调用,这也很令人困惑。但是,您可以通过将其重命名为_counter或在类定义的最后执行del counter来解决该问题。


无论如何,装饰器部分都可以工作,但是还有第二个问题:

def wrapper():

这定义了没有参数的函数。但是您正在尝试创建可以作为普通实例方法调用的对象,因此将传递一个self参数。您需要接受该self参数,还需要将其传递给包装的函数。所以:

def counter(func):
    def wrapper(self):
        wrapper.counter += 1
        return func(self, wrapper.counter)
    # etc.

现在,一切都按我的预期进行。


在使用该代码时,这段代码有点奇怪:

def create_profiles():
    Profile()._populate_profile_dict()

这将创建一个Profile实例,在其上调用一个私有方法,然后丢弃该实例。我不确定为什么要这么做。当然,它会将profile1条目添加到私有全局_profile_dict中,但是那能为您带来什么呢?


现在是您的第二次尝试:

@classmethod
def counter(cls, func):
    def wrapper():
        wrapper.counter += 1
        return func(wrapper.counter)

    wrapper.counter = 0
    return wrapper

现在,您成功创建了一个公共classmethod,您可以尝试将其明确地称为classmethod,这很完美:

Profile()._populate_profile_dict = 
Profile.counter(Profile._populate_profile_dict)

但是它还有其他多个问题:

  • 像这样SyntaxError尝试将作业分成两行。
  • Profile.counter(Profile._populate_profile_dict)返回一个新函数,您不会调用它,因此它只是将函数本身分配为值。
  • 当您进行调用时,它期望有两个参数,一个self和一个count,但是您只传递了count。在这里,您不清楚要从哪里获得self。也许您想在Profile的实例上调用它,以便得到self并可以通过它?
  • 您正在将值分配给刚刚创建的Profile实例的实例属性,该实例属性将被丢弃,这没有可见效果。
  • 该instance属性与您要调用的方法具有相同的名称-这是合法的,并且在这里不会引起任何特定的问题,但这很令人困惑。

与您要尝试执行的操作相比,我能看到的最接近的合理方法是使counter返回可调用的内容作为方法,如上所述:

@classmethod
def counter(cls, func):
    def wrapper(self):
        wrapper.counter += 1
        return func(self, wrapper.counter)

    wrapper.counter = 0
    return wrapper

…然后用结果猴子Profile进行补丁,然后在Profile实例上调用该方法,如下所示:

if __name__ == "__main__":
    Profile._populate_profile_dict = Profile.counter(Profile._populate_profile_dict)
    Profile()._populate_profile_dict()

现在它可以工作了,但是我不确定这是否是您想要的。