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

WIP: Refactor Aurora (3.0.0) #29

Draft
cswimr wants to merge 347 commits from aurora-pydantic into main
pull from: aurora-pydantic
merge into: cswimr:main
Conversation 1 Commits 347 Files changed 23 +6 -3
2 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit 2dfc9d9824 - Show all commits

5
aurora/models.py
View file

@ -5,7 +5,10 @@ from typing import Dict, List, Optional
from pydantic import BaseModel from pydantic import BaseModel
class Moderation(BaseModel):
class AuroraBaseModel(BaseModel):
"""Base class for all models in Aurora."""
class Moderation(AuroraBaseModel):
moderation_id: int moderation_id: int
guild_id: int guild_id: int
timestamp: datetime timestamp: datetime

4
aurora/utilities/json.py
View file

@ -1,7 +1,7 @@
import json import json
from datetime import datetime, timedelta from datetime import datetime, timedelta
from pydantic import BaseModel from aurora.models import AuroraBaseModel
class JSONEncoder(json.JSONEncoder): class JSONEncoder(json.JSONEncoder):
@ -10,7 +10,7 @@ class JSONEncoder(json.JSONEncoder):
return int(o.timestamp()) return int(o.timestamp())
if isinstance(o, timedelta): if isinstance(o, timedelta):
return str(o) return str(o)
if isinstance(o, BaseModel): if isinstance(o, AuroraBaseModel):
return o.model_dump() return o.model_dump()
return super().default(o) return super().default(o)