Added stubs for classes and cogs, performed a cleanup, added more database indexes
This commit is contained in:
@@ -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
|
||||
|
@@ -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
17
classes/pycord_event.py
Normal 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]
|
18
classes/pycord_event_stage.py
Normal file
18
classes/pycord_event_stage.py
Normal 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]
|
Reference in New Issue
Block a user