在python中使用lambda或helper方法避免for循环重复

2024-03-29 09:11:28 发布

您现在位置:Python中文网/ 问答频道 /正文

我有下面这个简单的类,我想知道是否有一个简单的方法来使用lambdadecorator,或helper method,等等。。。要避免在CODENAMESALL_DEFAULTS的每个方法中出现重复的for循环?你知道吗

class classproperty(object):
    """
    When used to decorate a method in a class, that method will behave
    like as a class property.
    """
    def __init__(self, f):
        # f - the func that's being decorated
        self.f = f

    def __get__(self, obj, cls):
        # call the func on the class
        return self.f(cls)

class PermissionInfo(object):

    MODELS = ['ticket', 'person', 'role']
    PERMS = ['view', 'add', 'change', 'delete']

    @classproperty
    def CODENAMES(cls):
        codenames = []
        for p in cls.PERMS:
            for m in cls.MODELS:
                codenames.append('{}_{}'.format(p, m))
        return codenames

    @classproperty
    def ALL_DEFAULTS(cls):
        ret = {}

        for p in cls.PERMS:
            for m in cls.MODELS:
                ret["{}_{}".format(p, m)] = False

        return ret

重复for循环是每个方法的此部分:

# ...
for p in cls.PERMS:
    for m in cls.MODELS:
#...

Tags: the方法inselfforreturnmodelsdef
3条回答
from itertools import product


class PermissionInfo(object):

    MODELS = ['ticket', 'person', 'role']
    PERMS = ['view', 'add', 'change', 'delete']
    PRODUCT = product(PERMS, MODELS)
    CODENAMES = ['{}_{}'.format(p, m) for (p, m) in PRODUCT]
    ALL_DEFAULTS = {codename: False for codename in CODENAMES}

哦,闪亮的:-}

可以在helper方法中使用^{}来生成名称:

from itertools import product

def names(cls):
    yield from ('_'.join(x) for x in product(cls.PERMS, cls.MODELS))

然后你可以改变你的类来利用它:

@classproperty
def CODENAMES(cls):
    return list(names(cls))

@classproperty
def ALL_DEFAULTS(cls):
    return dict.fromkeys(names(cls), False)

你也许可以用这样的方法:

MODEL_PERMS = list(map(lambda t: "{}_{}".format(t[0],t[1]), itertools.product(MODELS, PERMS)))

相关问题 更多 >