Added stubs for classes and cogs, performed a cleanup, added more database indexes

This commit is contained in:
2025-04-18 00:48:09 +02:00
parent 82c57a3dd1
commit 691dd1c958
9 changed files with 104 additions and 23 deletions

View File

@@ -1,2 +1,4 @@
from .pycord_event import PycordEvent
from .pycord_event_stage import PycordEventStage
from .pycord_guild import PycordGuild
from .pycord_user import PycordUser

View File

@@ -2,7 +2,6 @@ import logging
from logging import Logger
from typing import Any
from aiohttp import ClientSession
from discord import User
from libbot.cache.manager import create_cache_client
from libbot.pycord.classes import PycordBot as LibPycordBot
@@ -12,17 +11,12 @@ from classes import PycordUser
logger: Logger = logging.getLogger(__name__)
# from modules.tracking.dhl import update_tracks_dhl
class PycordBot(LibPycordBot):
def __init__(self, *args, **kwargs) -> None:
super().__init__(*args, **kwargs)
self._set_cache_engine()
self.client_session = ClientSession()
if self.scheduler is None:
return
@@ -63,8 +57,6 @@ class PycordBot(LibPycordBot):
await super().start(*args, **kwargs)
async def close(self, **kwargs) -> None:
await self.client_session.close()
if self.scheduler is not None:
self.scheduler.shutdown()

17
classes/pycord_event.py Normal file
View File

@@ -0,0 +1,17 @@
from dataclasses import dataclass
from datetime import datetime
from typing import List
from bson import ObjectId
@dataclass
class PycordEvent:
_id: ObjectId
id: int
guild_id: int
created: datetime
creator_id: int
starts: datetime
ends: datetime
stage_ids: List[int]

View File

@@ -0,0 +1,18 @@
from dataclasses import dataclass
from datetime import datetime
from typing import List
from bson import ObjectId
@dataclass
class PycordEventStage:
_id: ObjectId
id: int
event_id: int
guild_id: int
sequence: int
created: datetime
creator_id: int
text: str | None
media: List[str]