python:使一个类对象可迭代

时间:2018-04-11 07:43:41

标签: python list object iterable class-method

以下是我要做的事情:

我创建了一个classmethod,如果没有将params添加到此方法,则返回dict的对象,或者如果添加了任何参数,则返回由dicts列表组成的对象,如下面的代码所示:

@classmethod
def make_attr(cls, **kwargs):
    """
    code: do something, generate either a tuple 
    """ 
    if :
        data = {}
        """
        add k, v in data
        """
        return cls(**data)
    else:
        data = []
        for k, v in kwargs.items():
            _d = {}
            _d[k] = v
            data.append(_d)
        return cls(*data)


代码如上,我遇到的问题是如果classmethod返回dict的对象,我可以使用object.attr来获取它的值,但如果classmethod返回list的对象,我发现无法解压缩它或得到它属性或值,例如:

>> result = Class.make_attr(login_name="test")
>> print("age: ", result.age)
>> age: 25

>> result = Class.make_attr()
>> print(result)
>> <model._class.Class object at 0x104020358>

>> for i in result:
>>     print(i)
>> TypeError: 'Class' object is not iterable

那么,我怎样才能使这个对象可迭代?

1 个答案:

答案 0 :(得分:1)

我终于通过在我的类中添加内置方法找到解决问题的方法:

class Class:
    def __init__(self, *args, **kwargs):
        self.start = 0
        self.end = args

    def __iter__(self):
        return self

    def __next__(self):
        if self.start >= len(self.end):
            raise StopIteration
        else:
            self.start += 1
            return self.end[self.start-1]

再次感谢@ducminh