misc(aurora): change the JSONEncoder subclass to use match/case instead of if statements

This commit is contained in:
Seaswimmer 2024-06-04 14:20:01 -04:00
parent 21fa3d9eb0
commit 8ac735dafe
Signed by untrusted user: cswimr
GPG key ID: 5D671B5D03D65A7F

View file

@ -1,5 +1,6 @@
import json import json
from datetime import datetime, timedelta from datetime import datetime, timedelta
from typing import Any
from redbot.core.bot import Red from redbot.core.bot import Red
@ -7,16 +8,18 @@ from ..models.base import AuroraBaseModel
class JSONEncoder(json.JSONEncoder): class JSONEncoder(json.JSONEncoder):
def default(self, o): def default(self, o) -> Any:
if isinstance(o, datetime): match o:
return int(o.timestamp()) case datetime():
if isinstance(o, timedelta): return int(o.timestamp())
return str(o) case timedelta():
if isinstance(o, AuroraBaseModel): return str(o)
return o.dump() case AuroraBaseModel():
if isinstance(o, Red): return o.dump()
return None case Red():
return super().default(o) return None
case _:
return super().default(o)
# This is a wrapper around the json module's dumps function that uses our custom JSONEncoder class # This is a wrapper around the json module's dumps function that uses our custom JSONEncoder class