86 lines
2.1 KiB
Python
86 lines
2.1 KiB
Python
|
import logging
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from typing import Literal, Union
|
||
|
|
|
||
|
|
from bson import ObjectId
|
||
|
|
from discord import Thread
|
||
|
|
|
||
|
|
from classes.enums import MemberStatus
|
||
|
|
from classes.errors import MemberNotFoundError
|
||
|
|
from classes.pycordcheck import PycordCheck
|
||
|
|
from modules.database import col_members
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class PycordMember:
|
||
|
|
"""Dataclass of DB entry of a member"""
|
||
|
|
|
||
|
|
__slots__ = ("_id", "id", "guild", "status")
|
||
|
|
|
||
|
|
_id: ObjectId
|
||
|
|
id: int
|
||
|
|
guild: ObjectId
|
||
|
|
status: Literal[
|
||
|
|
MemberStatus.VERIFIED,
|
||
|
|
MemberStatus.UNVERIFIED,
|
||
|
|
MemberStatus.FAILED,
|
||
|
|
MemberStatus.ADDITIONAL,
|
||
|
|
]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
async def find(cls, id: Union[int, ObjectId], guild: ObjectId):
|
||
|
|
"""Find a member in a database.
|
||
|
|
|
||
|
|
### Args:
|
||
|
|
* id (`Union[int, ObjectId]`): Member's Discord ID
|
||
|
|
* guild (`ObjectId`): Discord guild's database ID
|
||
|
|
|
||
|
|
### Raises:
|
||
|
|
* `MemberNotFoundError`: Raised when member entry after insertion could not be found.
|
||
|
|
|
||
|
|
### Returns:
|
||
|
|
* `PycordMember`: Member with its database data.
|
||
|
|
"""
|
||
|
|
db_entry = await col_members.find_one(
|
||
|
|
(
|
||
|
|
{"id": id, "guild": guild}
|
||
|
|
if isinstance(id, int)
|
||
|
|
else {"_id": id, "guild": guild}
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
if db_entry is None:
|
||
|
|
raise MemberNotFoundError(id, guild)
|
||
|
|
|
||
|
|
return cls(**db_entry)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
async def create(
|
||
|
|
cls,
|
||
|
|
id: int,
|
||
|
|
guild: ObjectId,
|
||
|
|
status: Literal[
|
||
|
|
MemberStatus.VERIFIED,
|
||
|
|
MemberStatus.UNVERIFIED,
|
||
|
|
MemberStatus.FAILED,
|
||
|
|
MemberStatus.ADDITIONAL,
|
||
|
|
],
|
||
|
|
):
|
||
|
|
pass
|
||
|
|
|
||
|
|
async def get_check(self, guild: ObjectId) -> PycordCheck:
|
||
|
|
pass
|
||
|
|
|
||
|
|
async def set_status(
|
||
|
|
self,
|
||
|
|
status: Literal[
|
||
|
|
MemberStatus.VERIFIED,
|
||
|
|
MemberStatus.UNVERIFIED,
|
||
|
|
MemberStatus.FAILED,
|
||
|
|
MemberStatus.ADDITIONAL,
|
||
|
|
],
|
||
|
|
) -> Union[Thread, None]:
|
||
|
|
pass
|