mirror of
https://github.com/neoarz/Syntrel.git
synced 2025-12-25 11:40:12 +01:00
34b034bdcd
Removed template copyright headers from moderation cogs and database files. Improved embed styling and consistency in purge and warnings commands, including unified colors, author fields, and permission checks. Added a helper for sending embeds in warnings cog and replaced decorator-based permission checks with manual checks for better feedback. Updated comments in database files.
92 lines
3.1 KiB
Python
92 lines
3.1 KiB
Python
# Copyright © Krypton 2019-Present - https://github.com/kkrypt0nn/Python-Discord-Bot-Template/blob/main/database/__init__.py
|
|
# Used by neoarz
|
|
|
|
import aiosqlite
|
|
|
|
|
|
class DatabaseManager:
|
|
def __init__(self, *, connection: aiosqlite.Connection) -> None:
|
|
self.connection = connection
|
|
|
|
async def add_warn(
|
|
self, user_id: int, server_id: int, moderator_id: int, reason: str
|
|
) -> int:
|
|
"""
|
|
This function will add a warn to the database.
|
|
|
|
:param user_id: The ID of the user that should be warned.
|
|
:param reason: The reason why the user should be warned.
|
|
"""
|
|
rows = await self.connection.execute(
|
|
"SELECT id FROM warns WHERE user_id=? AND server_id=? ORDER BY id DESC LIMIT 1",
|
|
(
|
|
user_id,
|
|
server_id,
|
|
),
|
|
)
|
|
async with rows as cursor:
|
|
result = await cursor.fetchone()
|
|
warn_id = result[0] + 1 if result is not None else 1
|
|
await self.connection.execute(
|
|
"INSERT INTO warns(id, user_id, server_id, moderator_id, reason) VALUES (?, ?, ?, ?, ?)",
|
|
(
|
|
warn_id,
|
|
user_id,
|
|
server_id,
|
|
moderator_id,
|
|
reason,
|
|
),
|
|
)
|
|
await self.connection.commit()
|
|
return warn_id
|
|
|
|
async def remove_warn(self, warn_id: int, user_id: int, server_id: int) -> int:
|
|
"""
|
|
This function will remove a warn from the database.
|
|
|
|
:param warn_id: The ID of the warn.
|
|
:param user_id: The ID of the user that was warned.
|
|
:param server_id: The ID of the server where the user has been warned
|
|
"""
|
|
await self.connection.execute(
|
|
"DELETE FROM warns WHERE id=? AND user_id=? AND server_id=?",
|
|
(
|
|
warn_id,
|
|
user_id,
|
|
server_id,
|
|
),
|
|
)
|
|
await self.connection.commit()
|
|
rows = await self.connection.execute(
|
|
"SELECT COUNT(*) FROM warns WHERE user_id=? AND server_id=?",
|
|
(
|
|
user_id,
|
|
server_id,
|
|
),
|
|
)
|
|
async with rows as cursor:
|
|
result = await cursor.fetchone()
|
|
return result[0] if result is not None else 0
|
|
|
|
async def get_warnings(self, user_id: int, server_id: int) -> list:
|
|
"""
|
|
This function will get all the warnings of a user.
|
|
|
|
:param user_id: The ID of the user that should be checked.
|
|
:param server_id: The ID of the server that should be checked.
|
|
:return: A list of all the warnings of the user.
|
|
"""
|
|
rows = await self.connection.execute(
|
|
"SELECT user_id, server_id, moderator_id, reason, strftime('%s', created_at), id FROM warns WHERE user_id=? AND server_id=?",
|
|
(
|
|
user_id,
|
|
server_id,
|
|
),
|
|
)
|
|
async with rows as cursor:
|
|
result = await cursor.fetchall()
|
|
result_list = []
|
|
for row in result:
|
|
result_list.append(row)
|
|
return result_list
|