I would like to add a method to my enum.
class Kerneltype(Enum):
tube = 0
subspace_KDE = 1
deltashift = 2
dist_sens_via_mass_1 = 3
def aslist(self):
return [self.tube, self.subspace_KDE, self.deltashift, self.dist_sens_via_mass_1]
def fromint(self, int):
return self.aslist()[int]
does not work. Instead of
Kerneltype.aslist()
I currently have to do
[kt[1] for kt in ob.Kerneltype.__members__.items()]
You created an instance method, so aslist only exists on instances of Kerneltype (i.e. the enum members themselves). You need a classmethod instead:
@classmethod
def aslist(cls):
return [cls.tube, cls.subspace_KDE, cls.deltashift, cls.dist_sens_via_mass_1]
>>> Kerneltype.aslist()
[<Kerneltype.tube: 0>, <Kerneltype.subspace_KDE: 1>, <Kerneltype.deltashift: 2>, <Kerneltype.dist_sens_via_mass_1: 3>]
tube and deltashift, are class variables that are stored on the class/type instead of an instance. If you want to access those, you will need to have a reference to the type. That reference is the cls that is passed to @classmethod methods but not @staticmethod methods.You should be defining your methods with the classmethod decorator as you are calling them from the class and not the Enum member.
@classmethod
def aslist(cls):
return [cls.tube, cls.subspace_KDE, cls.deltashift, cls.dist_sens_via_mass_1]
@classmethod
def fromint(cls, int):
return cls.aslist()[int]
As others have mentioned in comments, your aslist() method is not required and you can directly use list() on it and it by default preserves the order of definition. Only difference is that it doesn't return the aliases.
>>> list(Kerneltype)
[<Kerneltype.tube: 0>, <Kerneltype.subspace_KDE: 1>, <Kerneltype.deltashift: 2>, <Kerneltype.dist_sens_via_mass_1: 3>]
You missed the classmethod decorator:
@classmethod
def aslist(self):
return [self.tube, self.subspace_KDE, self.deltashift, self.dist_sens_via_mass_1]
The issue with what you do is trying to access enum values via self whereas you should access them as you would normally via the enum.
from enum import Enum
class V(Enum):
V1 = (1, 2)
V2 = (2, 3)
def getKey(v: float):
return [k for k in list(V) if v > k.value[0] and v < k.value[1]]
V.getKey(1.1) # [<V.V1: (1, 2)>]
V.getKey(1.1)[0].value # (1, 2)
aslistmethod does not do anything thatlist(Kerneltype)would not give you automatically.subspace_KDEto say 5 and then try again.