cswimr/SeaCogs
cswimr/SeaCogs
1
1
Fork 1
Code Issues 10 Pull requests 4 Projects 1 Packages 1 Wiki Activity Actions

misc(aurora): change the JSONEncoder subclass to use match/case instead of if statements
All checks were successful
Actions / Build Documentation (MkDocs) (pull_request) Successful in 40s
Details
Actions / Lint Code (Ruff & Pylint) (pull_request) Successful in 1m4s
Details

Browse source
This commit is contained in:
Seaswimmer 2024-06-04 14:20:01 -04:00
parent 21fa3d9eb0
commit 8ac735dafe
Signed by: cswimr
GPG key ID: 5D671B5D03D65A7F
1 changed files with 13 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
aurora/utilities/json.py
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