fixing enum serialization for python 3.10

This commit is contained in:
Mose Müller 2024-01-16 14:13:36 +01:00
parent 616a5cea21
commit 484b5131e9
2 changed files with 6 additions and 1 deletions

View File

@ -1,5 +1,6 @@
import inspect
import logging
import sys
from collections.abc import Callable
from enum import Enum
from typing import Any
@ -68,6 +69,8 @@ class Serializer:
value = obj.name
readonly = False
doc = obj.__doc__
if sys.version_info < (3, 11) and doc == "An enumeration.":
doc = None
if type(obj).__base__.__name__ == "ColouredEnum":
obj_type = "ColouredEnum"
else:

View File

@ -100,6 +100,8 @@ def test_enum_serialize() -> None:
def test_ColouredEnum_serialize() -> None:
class Status(ColouredEnum):
"""Status description."""
PENDING = "#FFA500"
RUNNING = "#0000FF80"
PAUSED = "rgb(169, 169, 169)"
@ -121,7 +123,7 @@ def test_ColouredEnum_serialize() -> None:
"RUNNING": "#0000FF80",
},
"readonly": False,
"doc": None,
"doc": "Status description.",
}