h/blog
1
Fork 0
mirror of https://github.com/lavafroth/lavafroth.github.io.git synced 2026-06-01 13:01:22 -03:00
Code Issues Packages Projects Releases Wiki Activity

feat: algebraic python enums (draft 5)

Browse Source
This commit is contained in:
Himadri Bhattacharjee
2025-11-03 08:42:42 +05:30
parent e65caed1b1
commit c201f3dace
1 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
content/post/algebraic-python-enums.md
View File

@@ -231,12 +231,10 @@ We can define a decorator that takes all of the nested dataclasses and makes the
{{< collapsable-explanation >}}
```python
import inspect
def AlgebraicEnum(cls):
for subclass_name, subclass in inspect.getmembers(cls, predicate=inspect.isclass):
if subclass_name != "__class__":
setattr(cls, subclass_name, type(subclass_name, (cls, subclass), {}))
for name, nested in cls.__dict__.items():
if isinstance(nested, type):
setattr(cls, name, type(name, (cls, nested), {}))
return cls
```
@@ -292,14 +290,13 @@ print(empty.is_empty())
```
```python
import inspect
from dataclasses import dataclass
def AlgebraicEnum(cls):
for subclass_name, subclass in inspect.getmembers(cls, predicate=inspect.isclass):
if subclass_name != "__class__":
setattr(cls, subclass_name, type(subclass_name, (cls, subclass), {}))
for name, nested in cls.__dict__.items():
if isinstance(nested, type):
setattr(cls, name, type(name, (cls, nested), {}))
return cls