1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
# --------------------------------------------------------------------------------------
# Copyright (c) 2013-2025, Nucleic Development Team.
#
# Distributed under the terms of the Modified BSD License.
#
# The full license is in the file LICENSE, distributed with this software.
# --------------------------------------------------------------------------------------
from .catom import DefaultValue, Member, Validate
class Enum(Member):
"""A member where the value can be one in a sequence of items."""
__slots__ = ()
def __init__(self, *items):
"""Initialize an Enum.
Parameters
----------
*items
The allowed values which can be assigned to the enum.
"""
if len(items) == 0:
raise ValueError("an Enum requires at least 1 item")
self.set_default_value_mode(DefaultValue.Static, items[0])
self.set_validate_mode(Validate.Enum, items)
@property
def items(self):
"""A readonly property which returns the items in the enum."""
return self.validate_mode[1]
def added(self, *items):
"""Create a clone of the Enum with added items.
Parameters
----------
*items
Additional items to include in the Enum.
Returns
-------
result : Enum
A new enum object which contains all of the original items
plus the new items.
"""
olditems = self.items
newitems = olditems + items
clone = self.clone()
clone.set_validate_mode(Validate.Enum, newitems)
return clone
def removed(self, *items):
"""Create a clone of the Enum with some items removed.
Parameters
----------
*items
The items to remove remove from the new enum.
Returns
-------
result : Enum
A new enum object which contains all of the original items
but with the given items removed.
"""
newitems = tuple(i for i in self.items if i not in items)
if len(newitems) == 0:
raise ValueError("an Enum requires at least 1 item")
clone = self.clone()
clone.set_default_value_mode(DefaultValue.Static, newitems[0])
clone.set_validate_mode(Validate.Enum, newitems)
return clone
def __call__(self, item):
"""Create a clone of the Enum item with a new default.
Parameters
----------
item : object
The item to use as the Enum default. The item must be one
of the valid enum items.
"""
if item not in self.items:
raise TypeError("invalid enum value")
clone = self.clone()
clone.set_default_value_mode(DefaultValue.Static, item)
return clone
|
