6 Commits
v1.0.0 ... main

5 changed files with 18 additions and 9 deletions

View File

@@ -27,7 +27,7 @@ logger: Logger = get_logger(__name__)
class PycordBot(LibPycordBot): class PycordBot(LibPycordBot):
__version__ = "1.0.0" __version__ = "1.0.1"
started: datetime started: datetime
cache: CacheMemcached | CacheRedis | None = None cache: CacheMemcached | CacheRedis | None = None

View File

@@ -11,7 +11,6 @@ from libbot.cache.classes import Cache
from pymongo import DESCENDING from pymongo import DESCENDING
from pymongo.results import InsertOneResult from pymongo.results import InsertOneResult
from classes.abstract import Cacheable
from classes.base.base_cacheable import BaseCacheable from classes.base.base_cacheable import BaseCacheable
from classes.errors import EventNotFoundError from classes.errors import EventNotFoundError
from modules.database import col_events from modules.database import col_events
@@ -86,7 +85,7 @@ class PycordEvent(BaseCacheable):
if cached_entry is not None: if cached_entry is not None:
return cls(**cls._entry_from_cache(cached_entry)) return cls(**cls._entry_from_cache(cached_entry))
db_entry = await cls.__collection__.find_one( db_entry: Dict[str, Any] | None = await cls.__collection__.find_one(
{"_id": event_id if isinstance(event_id, ObjectId) else ObjectId(event_id)} {"_id": event_id if isinstance(event_id, ObjectId) else ObjectId(event_id)}
) )

View File

@@ -64,7 +64,7 @@ class PycordEventStage(BaseCacheable):
if cached_entry is not None: if cached_entry is not None:
return cls(**cls._entry_from_cache(cached_entry)) return cls(**cls._entry_from_cache(cached_entry))
db_entry = await cls.__collection__.find_one( db_entry: Dict[str, Any] | None = await cls.__collection__.find_one(
{"_id": stage_id if isinstance(stage_id, ObjectId) else ObjectId(stage_id)} {"_id": stage_id if isinstance(stage_id, ObjectId) else ObjectId(stage_id)}
) )
@@ -155,7 +155,11 @@ class PycordEventStage(BaseCacheable):
""" """
return { return {
"_id": self._id if not json_compatible else str(self._id), "_id": self._id if not json_compatible else str(self._id),
"event_id": self.event_id if not json_compatible else str(self.event_id), "event_id": (
self.event_id
if not json_compatible
else (None if self.event_id is None else str(self.event_id))
),
"guild_id": self.guild_id, "guild_id": self.guild_id,
"sequence": self.sequence, "sequence": self.sequence,
"created": self.created if not json_compatible else self.created.isoformat(), "created": self.created if not json_compatible else self.created.isoformat(),

View File

@@ -60,7 +60,7 @@ class PycordGuild(BaseCacheable):
if cached_entry is not None: if cached_entry is not None:
return cls(**cls._entry_from_cache(cached_entry)) return cls(**cls._entry_from_cache(cached_entry))
db_entry = await cls.__collection__.find_one({"id": guild_id}) db_entry: Dict[str, Any] | None = await cls.__collection__.find_one({"id": guild_id})
if db_entry is None: if db_entry is None:
if not allow_creation: if not allow_creation:

View File

@@ -86,7 +86,9 @@ class PycordUser(BaseCacheable):
if cached_entry is not None: if cached_entry is not None:
return cls(**cls._entry_from_cache(cached_entry)) return cls(**cls._entry_from_cache(cached_entry))
db_entry = await cls.__collection__.find_one({"id": user_id, "guild_id": guild_id}) db_entry: Dict[str, Any] | None = await cls.__collection__.find_one(
{"id": user_id, "guild_id": guild_id}
)
if db_entry is None: if db_entry is None:
if not allow_creation: if not allow_creation:
@@ -119,10 +121,14 @@ class PycordUser(BaseCacheable):
"event_channels": self.event_channels, "event_channels": self.event_channels,
"is_jailed": self.is_jailed, "is_jailed": self.is_jailed,
"current_event_id": ( "current_event_id": (
self.current_event_id if not json_compatible else str(self.current_event_id) self.current_event_id
if not json_compatible
else (None if self.current_event_id is None else str(self.current_event_id))
), ),
"current_stage_id": ( "current_stage_id": (
self.current_stage_id if not json_compatible else str(self.current_stage_id) self.current_stage_id
if not json_compatible
else (None if self.current_stage_id is None else str(self.current_stage_id))
), ),
"registered_event_ids": ( "registered_event_ids": (
self.registered_event_ids self.registered_event_ids