Communities

Writing
Writing
Codidact Meta
Codidact Meta
The Great Outdoors
The Great Outdoors
Photography & Video
Photography & Video
Scientific Speculation
Scientific Speculation
Cooking
Cooking
Electrical Engineering
Electrical Engineering
Judaism
Judaism
Languages & Linguistics
Languages & Linguistics
Software Development
Software Development
Mathematics
Mathematics
Christianity
Christianity
Code Golf
Code Golf
Music
Music
Physics
Physics
Linux Systems
Linux Systems
Power Users
Power Users
Tabletop RPGs
Tabletop RPGs
Community Proposals
Community Proposals
tag:snake search within a tag
answers:0 unanswered questions
user:xxxx search by author id
score:0.5 posts with 0.5+ score
"snake oil" exact phrase
votes:4 posts with 4+ votes
created:<1w created < 1 week ago
post_type:xxxx type of post
Search help
Notifications
Mark all as read See all your notifications »
Q&A

Welcome to Software Development on Codidact!

Will you help us build our independent community of developers helping developers? We're small and trying to grow. We welcome questions about all aspects of software development, from design to code to QA and more. Got questions? Got answers? Got code you'd like someone to review? Please join us.

Post History

77%
+5 −0
Q&A Is it possible to write protocols for enumerations in Python?

There's one big difficulty with the proposed pattern. Enum instances are singleton instances of their particular class, and in general two enums from different classes are not equal even if they wr...

posted 2y ago by r~~‭

Answer
#1: Initial revision by user avatar r~~‭ · 2022-05-11T16:57:55Z (almost 2 years ago)
There's one big difficulty with the proposed pattern. Enum instances are singleton instances of their particular class, and in general two enums from different classes are not equal even if they wrap the same int value. So an expression like `foobar == EnumProtocol.FOO` is never going to be true if `foobar` is any enum instance.

If you're okay with adapting that to `foobar.value == EnumProtocol.FOO`, then you can do this:

```python
from typing import Literal, Protocol

from enum import Enum

class EnumProtocol(Protocol):
    FOO: Literal[1] = 1
    BAR: Literal[2] = 2

    # You could move these properties into a superprotocol
    @property
    def name(self) -> str:
        pass
    
    @property
    def value(self) -> int:
        pass

def useit(foobar: EnumProtocol) -> None:
    if foobar.value == EnumProtocol.FOO:
        print("foo")
    elif foobar.value == EnumProtocol.BAR:
        print("bar")
    else:
        print(f"{foobar.name} = {foobar.value}")

# The following class fulfils the protocol
class GoodEnum(Enum):
    FOO = 1
    BAR = 2
    BAZ = 3

useit(GoodEnum.FOO) # ok, prints "foo"
useit(GoodEnum.BAZ) # ok, prints "BAZ = 3"

# The following class violates the protocol by having wrong values
class BadEnum(Enum):
    FOO = 2
    BAR = 3

useit(BadEnum.FOO) # type checking gives error

# The following class violates the protocol by missing BAR
class UglyEnum(Enum):
    FOO = 1
    BAZ = 3

useit(UglyEnum.FOO) # type checking gives error
```