From 5c28a8cb8ed19edda473d12e3838d5ca409a1c60 Mon Sep 17 00:00:00 2001 From: Vyacheslav Tverskoy Date: Thu, 18 Nov 2021 09:04:07 +0000 Subject: [PATCH] Support enums in flask-mongorest --- flask_mongorest/utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/flask_mongorest/utils.py b/flask_mongorest/utils.py index 272473d2..572c8643 100644 --- a/flask_mongorest/utils.py +++ b/flask_mongorest/utils.py @@ -1,3 +1,4 @@ +import enum import json import decimal import datetime @@ -26,6 +27,8 @@ def default(self, value, **kwargs): return value.strftime("%Y-%m-%d") if isinstance(value, decimal.Decimal): return str(value) + if isinstance(value, enum.Enum): + return value.value return super(MongoEncoder, self).default(value, **kwargs)