如何验证namedtuple值?

时间:2019-06-26 19:43:53

标签: python oop namedtuple

我有namedtuple变量,它代表应用程序的版本(其编号和类型)。但是我想要对值进行一些限制:

Version = namedtuple("Version", ["app_type", "number"])
version = Version("desktop") # i want only "desktop" and "web" are valid app types
version = Version("deskpop") # i want to protect from such mistakes

我目前的解决方案是没有方法的原始类:

class Version:
    def __init__(self, app_type, number):
        assert app_type in ('desktop', 'web')

        self.app_type = app_type
        self.number = number

它是pythonic吗?是过度杀伤力吗?

1 个答案:

答案 0 :(得分:4)

您可以使用enum.Enumtyping.NamedTuple代替collections.namedtuple

也许是这样的:

from typing import NamedTuple
import enum

class AppType(enum.Enum):
    desktop = 0
    web = 1

class Version(NamedTuple):
    app: AppType


v0 = Version(app=AppType.desktop)
v1 = Version(app=AppType.web)

print(v0, v1)

输出:

Version(app=<AppType.desktop: 0>) Version(app=<AppType.web: 1>)

未定义的AppType引发AttributeError

v2 = Version(app=AppType.deskpoop)

输出:

AttributeError: deskpoop