How to use private enum values that are not accessible by Enum.__members__?

user_610569 0
Viewed 205

With an Enum object as such:

from enum import Enum

class Fruits(str, Enum):
    apple = "apple"
    banana = "banana"
    orange = "orange"
    lemon = "lemon"

    _citrus = {orange, lemon}

    def is_citrus(self):
        return self in self.citrus

when we try to check its member:

>>> Fruits.__members__.keys()
dict_keys(['apple', 'banana', 'orange', 'lemon', '_citrus'])

we get all possible Enum values. Is there a way to declare private vs public enums where some specific values are private and not accessible by Enum.__members__?

It's possible to do this:

from enum import Enum

class Fruits(str, Enum):
    apple = "apple"
    banana = "banana"
    orange = "orange"
    lemon = "lemon"

    _citrus = {orange, lemon}

    def is_citrus(self):
        return self in self.citrus
    
    @classmethod
    def values(cls):
        return set(k for k in cls.__members__.keys() if not k.startswith('_'))

Usage:

>>> Fruits.values()
{'apple', 'banana', 'lemon', 'orange'}

Are there some other way to have restricted members access?

Are there examples of private vs public members in Python's Enum?

private enums python
1 Answers

You can use a property for this problem, like this:

class Fruits(str, Enum):
    apple = "apple"
    banana = "banana"
    orange = "orange"
    lemon = "lemon"

    @property
    def _citrus(self):
        return {self.__class__.orange, self.__class__.lemon}

    def is_citrus(self):
        return self in self._citrus
    

print(Fruits.orange.is_citrus())
# True

print(Fruits.__members__.keys())
dict_keys(['apple', 'banana', 'orange', 'lemon'])
user_116470250
Related