misc(aurora): change the JSONEncoder subclass to use match/case instead of if statements
This commit is contained in:
parent
21fa3d9eb0
commit
8ac735dafe
1 changed files with 13 additions and 10 deletions
|
@ -1,5 +1,6 @@
|
|||
import json
|
||||
from datetime import datetime, timedelta
|
||||
from typing import Any
|
||||
|
||||
from redbot.core.bot import Red
|
||||
|
||||
|
@ -7,16 +8,18 @@ from ..models.base import AuroraBaseModel
|
|||
|
||||
|
||||
class JSONEncoder(json.JSONEncoder):
|
||||
def default(self, o):
|
||||
if isinstance(o, datetime):
|
||||
return int(o.timestamp())
|
||||
if isinstance(o, timedelta):
|
||||
return str(o)
|
||||
if isinstance(o, AuroraBaseModel):
|
||||
return o.dump()
|
||||
if isinstance(o, Red):
|
||||
return None
|
||||
return super().default(o)
|
||||
def default(self, o) -> Any:
|
||||
match o:
|
||||
case datetime():
|
||||
return int(o.timestamp())
|
||||
case timedelta():
|
||||
return str(o)
|
||||
case AuroraBaseModel():
|
||||
return o.dump()
|
||||
case Red():
|
||||
return None
|
||||
case _:
|
||||
return super().default(o)
|
||||
|
||||
|
||||
# This is a wrapper around the json module's dumps function that uses our custom JSONEncoder class
|
||||
|
|
Loading…
Reference in a new issue