Compare commits
15 Commits
a753918432
...
v0.1.0
Author | SHA1 | Date | |
---|---|---|---|
b9aeaf5c86 | |||
|
eb8019ccfe | ||
|
ce57755eee | ||
|
7a64e334d2 | ||
6087349622 | |||
|
9417951f55 | ||
|
6060a3df83 | ||
eed084cd91
|
|||
7b64f6938b
|
|||
c54586940e
|
|||
0195706e92
|
|||
162898f5eb
|
|||
3010dc02bc | |||
4afcbc93d5 | |||
72ccaa04a4 |
@@ -10,7 +10,7 @@
|
|||||||
## Installation from release
|
## Installation from release
|
||||||
|
|
||||||
1. Install MongoDB using the [official installation manual](https://www.mongodb.com/docs/manual/installation)
|
1. Install MongoDB using the [official installation manual](https://www.mongodb.com/docs/manual/installation)
|
||||||
2. Install Python 3.9+ (3.11+ is recommended)
|
2. Install Python 3.11+
|
||||||
3. Download the [latest release](https://git.end-play.xyz/HoloUA/Discord/releases/latest)'s archive
|
3. Download the [latest release](https://git.end-play.xyz/HoloUA/Discord/releases/latest)'s archive
|
||||||
4. Extract the archive
|
4. Extract the archive
|
||||||
5. Navigate to the extracted folder and subfolder `Discord` in it
|
5. Navigate to the extracted folder and subfolder `Discord` in it
|
||||||
@@ -19,14 +19,14 @@
|
|||||||
7. Activate the virtual environment:
|
7. Activate the virtual environment:
|
||||||
Windows: `.venv\Scripts\activate.bat`
|
Windows: `.venv\Scripts\activate.bat`
|
||||||
Linux/macOS: `.venv/bin/activate`
|
Linux/macOS: `.venv/bin/activate`
|
||||||
8. Install dependencies:
|
8. Install the dependencies:
|
||||||
`python -m pip install -r requirements.txt`
|
`python -m pip install -r requirements.txt`
|
||||||
9. Run the bot with `python main.py` after completing the [configuration](#Configuration)
|
9. Run the bot with `python main.py` after completing the [configuration](#Configuration)
|
||||||
|
|
||||||
## Installation with Git
|
## Installation with Git
|
||||||
|
|
||||||
1. Install MongoDB using the [official installation manual](https://www.mongodb.com/docs/manual/installation)
|
1. Install MongoDB using the [official installation manual](https://www.mongodb.com/docs/manual/installation)
|
||||||
2. Install Python 3.9+ (3.11+ is recommended)
|
2. Install Python 3.11+
|
||||||
3. Clone the repository:
|
3. Clone the repository:
|
||||||
`git clone https://git.end-play.xyz/HoloUA/Discord.git`
|
`git clone https://git.end-play.xyz/HoloUA/Discord.git`
|
||||||
4. `cd Discord`
|
4. `cd Discord`
|
||||||
|
@@ -4,15 +4,3 @@ from libbot.pycord.classes import PycordBot
|
|||||||
class HoloBot(PycordBot):
|
class HoloBot(PycordBot):
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
super().__init__(*args, **kwargs)
|
super().__init__(*args, **kwargs)
|
||||||
|
|
||||||
async def start(self, *args, **kwargs) -> None:
|
|
||||||
if self.scheduler is not None:
|
|
||||||
self.scheduler.start()
|
|
||||||
|
|
||||||
await super().start(*args, **kwargs)
|
|
||||||
|
|
||||||
async def close(self) -> None:
|
|
||||||
if self.scheduler is not None:
|
|
||||||
self.scheduler.shutdown()
|
|
||||||
|
|
||||||
await super().close()
|
|
||||||
|
@@ -1,22 +1,23 @@
|
|||||||
import logging
|
import logging
|
||||||
from typing import Any, Union, Dict
|
from logging import Logger
|
||||||
|
from typing import Any, Dict
|
||||||
|
|
||||||
from bson import ObjectId
|
from bson import ObjectId
|
||||||
from discord import User, Member
|
from discord import User, Member
|
||||||
from libbot import config_get
|
from libbot.utils import config_get
|
||||||
|
|
||||||
from errors import UserNotFoundError
|
from errors import UserNotFoundError
|
||||||
from modules.database import col_warnings, sync_col_users, sync_col_warnings, col_users
|
from modules.database import col_warnings, sync_col_users, sync_col_warnings, col_users
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger: Logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class HoloUser:
|
class HoloUser:
|
||||||
def __init__(self, user: Union[User, Member, int]) -> None:
|
def __init__(self, user: User | Member | int) -> None:
|
||||||
"""Get an object that has a proper binding between Discord ID and database
|
"""Get an object that has a proper binding between Discord ID and database
|
||||||
|
|
||||||
### Args:
|
### Args:
|
||||||
* `user` (Union[User, Member, int]): Object from which ID can be extracted
|
* `user` (User | Member | int): Object from which ID can be extracted
|
||||||
|
|
||||||
### Raises:
|
### Raises:
|
||||||
* `UserNotFoundError`: User with such ID does not seem to exist in database
|
* `UserNotFoundError`: User with such ID does not seem to exist in database
|
||||||
@@ -24,17 +25,15 @@ class HoloUser:
|
|||||||
|
|
||||||
self.id: int = user if not hasattr(user, "id") else user.id
|
self.id: int = user if not hasattr(user, "id") else user.id
|
||||||
|
|
||||||
jav_user: Union[Dict[str, Any], None] = sync_col_users.find_one(
|
jav_user: Dict[str, Any] | None = sync_col_users.find_one({"user": self.id})
|
||||||
{"user": self.id}
|
|
||||||
)
|
|
||||||
|
|
||||||
if jav_user is None:
|
if jav_user is None:
|
||||||
raise UserNotFoundError(user=user, user_id=self.id)
|
raise UserNotFoundError(user=user, user_id=self.id)
|
||||||
|
|
||||||
self.db_id: ObjectId = jav_user["_id"]
|
self.db_id: ObjectId = jav_user["_id"]
|
||||||
|
|
||||||
self.customrole: Union[int, None] = jav_user["customrole"]
|
self.customrole: int | None = jav_user["customrole"]
|
||||||
self.customchannel: Union[int, None] = jav_user["customchannel"]
|
self.customchannel: int | None = jav_user["customchannel"]
|
||||||
self.warnings: int = self.warns()
|
self.warnings: int = self.warns()
|
||||||
|
|
||||||
def warns(self) -> int:
|
def warns(self) -> int:
|
||||||
@@ -43,9 +42,7 @@ class HoloUser:
|
|||||||
### Returns:
|
### Returns:
|
||||||
* `int`: Number of warnings
|
* `int`: Number of warnings
|
||||||
"""
|
"""
|
||||||
warns: Union[Dict[str, Any], None] = sync_col_warnings.find_one(
|
warns: Dict[str, Any] | None = sync_col_warnings.find_one({"user": self.id})
|
||||||
{"user": self.id}
|
|
||||||
)
|
|
||||||
|
|
||||||
return 0 if warns is None else warns["warns"]
|
return 0 if warns is None else warns["warns"]
|
||||||
|
|
||||||
@@ -55,9 +52,7 @@ class HoloUser:
|
|||||||
### Args:
|
### Args:
|
||||||
* `count` (int, optional): Count of warnings to be added. Defaults to 1.
|
* `count` (int, optional): Count of warnings to be added. Defaults to 1.
|
||||||
"""
|
"""
|
||||||
warns: Union[Dict[str, Any], None] = await col_warnings.find_one(
|
warns: Dict[str, Any] | None = await col_warnings.find_one({"user": self.id})
|
||||||
{"user": self.id}
|
|
||||||
)
|
|
||||||
|
|
||||||
if warns is not None:
|
if warns is not None:
|
||||||
await col_warnings.update_one(
|
await col_warnings.update_one(
|
||||||
@@ -88,11 +83,11 @@ class HoloUser:
|
|||||||
logger.info("Set attribute %s of user %s to %s", key, self.id, value)
|
logger.info("Set attribute %s of user %s to %s", key, self.id, value)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def is_moderator(member: Union[User, Member]) -> bool:
|
async def is_moderator(member: User | Member) -> bool:
|
||||||
"""Check if user is moderator or council member
|
"""Check if user is moderator or council member
|
||||||
|
|
||||||
### Args:
|
### Args:
|
||||||
* `member` (Union[User, Member]): Member object
|
* `member` (User | Member): Member object
|
||||||
|
|
||||||
### Returns:
|
### Returns:
|
||||||
`bool`: `True` if member is a moderator or member of council and `False` if not
|
`bool`: `True` if member is a moderator or member of council and `False` if not
|
||||||
@@ -100,8 +95,8 @@ class HoloUser:
|
|||||||
if isinstance(member, User):
|
if isinstance(member, User):
|
||||||
return False
|
return False
|
||||||
|
|
||||||
moderator_role: Union[int, None] = await config_get("moderators", "roles")
|
moderator_role: int | None = await config_get("moderators", "roles")
|
||||||
council_role: Union[int, None] = await config_get("council", "roles")
|
council_role: int | None = await config_get("council", "roles")
|
||||||
|
|
||||||
for role in member.roles:
|
for role in member.roles:
|
||||||
if role.id in (moderator_role, council_role):
|
if role.id in (moderator_role, council_role):
|
||||||
@@ -110,11 +105,11 @@ class HoloUser:
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def is_council(member: Union[User, Member]) -> bool:
|
async def is_council(member: User | Member) -> bool:
|
||||||
"""Check if user is a member of council
|
"""Check if user is a member of council
|
||||||
|
|
||||||
### Args:
|
### Args:
|
||||||
* `member` (Union[User, Member]): Member object
|
* `member` (User | Member): Member object
|
||||||
|
|
||||||
### Returns:
|
### Returns:
|
||||||
`bool`: `True` if member is a member of council and `False` if not
|
`bool`: `True` if member is a member of council and `False` if not
|
||||||
@@ -129,8 +124,3 @@ class HoloUser:
|
|||||||
return True
|
return True
|
||||||
|
|
||||||
return False
|
return False
|
||||||
|
|
||||||
# def purge(self) -> None:
|
|
||||||
# """Completely remove data from database. Will not remove transactions logs and warnings."""
|
|
||||||
# col_users.delete_one(filter={"_id": self.db_id})
|
|
||||||
# self.unauthorize()
|
|
||||||
|
110
cogs/admin.py
110
cogs/admin.py
@@ -1,6 +1,6 @@
|
|||||||
import logging
|
import logging
|
||||||
import sys
|
import sys
|
||||||
from typing import Union
|
from logging import Logger
|
||||||
|
|
||||||
from discord import (
|
from discord import (
|
||||||
ApplicationContext,
|
ApplicationContext,
|
||||||
@@ -13,8 +13,7 @@ from discord import (
|
|||||||
)
|
)
|
||||||
from discord import utils as ds_utils
|
from discord import utils as ds_utils
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from libbot import config_get
|
from libbot.utils import config_get
|
||||||
from libbot.sync import config_get as sync_config_get
|
|
||||||
|
|
||||||
from classes.holo_bot import HoloBot
|
from classes.holo_bot import HoloBot
|
||||||
from enums import Color
|
from enums import Color
|
||||||
@@ -22,7 +21,7 @@ from modules.scheduler import scheduler
|
|||||||
from modules.utils_sync import guild_name
|
from modules.utils_sync import guild_name
|
||||||
from modules.waifu_pics import waifu_pics
|
from modules.waifu_pics import waifu_pics
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger: Logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class Admin(commands.Cog):
|
class Admin(commands.Cog):
|
||||||
@@ -31,93 +30,10 @@ class Admin(commands.Cog):
|
|||||||
def __init__(self, client: HoloBot):
|
def __init__(self, client: HoloBot):
|
||||||
self.client: HoloBot = client
|
self.client: HoloBot = client
|
||||||
|
|
||||||
# Disabled because warning functionality is temporarily not needed
|
|
||||||
# @slash_command(
|
|
||||||
# name="warning",
|
|
||||||
# description="Попередити юзера про порушення правил",
|
|
||||||
# guild_ids=[config_get_sync("guild")],
|
|
||||||
# )
|
|
||||||
# @option("user", description="Користувач")
|
|
||||||
# @option("reason", description="Причина")
|
|
||||||
# async def warn_cmd(
|
|
||||||
# self,
|
|
||||||
# ctx: ApplicationContext,
|
|
||||||
# user: User,
|
|
||||||
# reason: str = "Не вказана",
|
|
||||||
# ):
|
|
||||||
# logging.info(f"User {ctx.user.id} warned {user.id} for {reason}")
|
|
||||||
# await ctx.defer()
|
|
||||||
# jav_user = HoloUser(user)
|
|
||||||
# if ctx.user.id in await config_get("admins"):
|
|
||||||
# logging.info(
|
|
||||||
# f"Moderator {guild_name(ctx.user)} warned {guild_name(user)} for {reason} (has {jav_user.warnings} warns)"
|
|
||||||
# )
|
|
||||||
# if jav_user.warnings >= 5:
|
|
||||||
# logging.info(
|
|
||||||
# f"User {guild_name(user)} was banned due to a big amount of warns ({jav_user.warnings})"
|
|
||||||
# )
|
|
||||||
# await user.send(
|
|
||||||
# embed=Embed(
|
|
||||||
# title="Перманентне блокування",
|
|
||||||
# description=f"Вас було заблоковано за неодноразове порушення правил сервера.",
|
|
||||||
# color=Color.fail,
|
|
||||||
# )
|
|
||||||
# )
|
|
||||||
# await user.ban(reason=reason)
|
|
||||||
# elif jav_user.warnings >= 2:
|
|
||||||
# logging.info(
|
|
||||||
# f"User {guild_name(user)} was muted due to a big amount of warns ({jav_user.warnings})"
|
|
||||||
# )
|
|
||||||
# jav_user.warn(reason=reason)
|
|
||||||
# await user.send(
|
|
||||||
# embed=Embed(
|
|
||||||
# title="Тимчасове блокування",
|
|
||||||
# description=f"Причина: `{reason}`\n\nНа вашому рахунку вже {jav_user.warnings} попереджень. Вас було тимчасово заблоковано на **1 годину**.\n\nЯкщо Ви продовжите порушувати правила сервера – згодом Вас заблокують.",
|
|
||||||
# color=0xDED56B,
|
|
||||||
# )
|
|
||||||
# )
|
|
||||||
# await user.timeout_for(timedelta(hours=1), reason=reason)
|
|
||||||
# else:
|
|
||||||
# jav_user.warn()
|
|
||||||
|
|
||||||
# await ctx.respond(
|
|
||||||
# embed=Embed(
|
|
||||||
# title="Попередження",
|
|
||||||
# description=f"{user.mention} Будь ласка, не порушуйте правила. Ви отримали попередження з причини `{reason}`.\n\nЯкщо Ви продовжите порушувати правила – це може призвести до блокування в спільноті.",
|
|
||||||
# color=0xDED56B,
|
|
||||||
# )
|
|
||||||
# )
|
|
||||||
# else:
|
|
||||||
# logging.warning(
|
|
||||||
# f"User {guild_name(ctx.user)} tried to use /warn but permission denied"
|
|
||||||
# )
|
|
||||||
# await ctx.respond(
|
|
||||||
# embed=Embed(
|
|
||||||
# title="Відмовлено в доступі",
|
|
||||||
# description="Здається, це команда лише для модераторів",
|
|
||||||
# color=Color.fail,
|
|
||||||
# )
|
|
||||||
# )
|
|
||||||
# mod_role = ds_utils.get(
|
|
||||||
# ctx.user.guild.roles, id=await config_get("moderators", "roles")
|
|
||||||
# )
|
|
||||||
# admin_chan = ds_utils.get(
|
|
||||||
# ctx.user.guild.channels,
|
|
||||||
# id=await config_get("adminchat", "channels", "text"),
|
|
||||||
# )
|
|
||||||
# await admin_chan.send(
|
|
||||||
# content=f"{mod_role.mention}",
|
|
||||||
# embed=Embed(
|
|
||||||
# title="Неавторизований запит",
|
|
||||||
# description=f"Користувач {ctx.user.mention} запитав у каналі {ctx.channel.mention} команду, до якої не повинен мати доступу/бачити.",
|
|
||||||
# color=Color.fail,
|
|
||||||
# ),
|
|
||||||
# )
|
|
||||||
|
|
||||||
@slash_command(
|
@slash_command(
|
||||||
name="clear",
|
name="clear",
|
||||||
description="Видалити деяку кількість повідомлень в каналі",
|
description="Видалити деяку кількість повідомлень в каналі",
|
||||||
guild_ids=[sync_config_get("guild")],
|
guild_ids=[config_get("guild")],
|
||||||
)
|
)
|
||||||
@option("amount", description="Кількість")
|
@option("amount", description="Кількість")
|
||||||
@option("user", description="Користувач", default=None)
|
@option("user", description="Користувач", default=None)
|
||||||
@@ -127,6 +43,10 @@ class Admin(commands.Cog):
|
|||||||
amount: int,
|
amount: int,
|
||||||
user: User,
|
user: User,
|
||||||
) -> None:
|
) -> None:
|
||||||
|
"""Command /clear <amount> [<user>]
|
||||||
|
|
||||||
|
Removes last <amount> messages in the current channel. Optionally from a specific user.
|
||||||
|
"""
|
||||||
if ctx.user.id in self.client.owner_ids:
|
if ctx.user.id in self.client.owner_ids:
|
||||||
logging.info(
|
logging.info(
|
||||||
"User %s removed %s message(s) in %s",
|
"User %s removed %s message(s) in %s",
|
||||||
@@ -161,10 +81,10 @@ class Admin(commands.Cog):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
mod_role: Union[Role, None] = ds_utils.get(
|
mod_role: Role | None = ds_utils.get(
|
||||||
ctx.user.guild.roles, id=await config_get("moderators", "roles")
|
ctx.user.guild.roles, id=await config_get("moderators", "roles")
|
||||||
)
|
)
|
||||||
admin_chan: Union[TextChannel, None] = ds_utils.get(
|
admin_chan: TextChannel | None = ds_utils.get(
|
||||||
ctx.user.guild.channels,
|
ctx.user.guild.channels,
|
||||||
id=await config_get("adminchat", "channels", "text"),
|
id=await config_get("adminchat", "channels", "text"),
|
||||||
)
|
)
|
||||||
@@ -182,9 +102,13 @@ class Admin(commands.Cog):
|
|||||||
@slash_command(
|
@slash_command(
|
||||||
name="reboot",
|
name="reboot",
|
||||||
description="Перезапустити бота",
|
description="Перезапустити бота",
|
||||||
guild_ids=[sync_config_get("guild")],
|
guild_ids=[config_get("guild")],
|
||||||
)
|
)
|
||||||
async def reboot_cmd(self, ctx: ApplicationContext) -> None:
|
async def reboot_cmd(self, ctx: ApplicationContext) -> None:
|
||||||
|
"""Command /reboot
|
||||||
|
|
||||||
|
Stops the bot. Is called "reboot" because it's assumed that the bot has automatic restart.
|
||||||
|
"""
|
||||||
await ctx.defer(ephemeral=True)
|
await ctx.defer(ephemeral=True)
|
||||||
|
|
||||||
if ctx.user.id in self.client.owner_ids:
|
if ctx.user.id in self.client.owner_ids:
|
||||||
@@ -217,10 +141,10 @@ class Admin(commands.Cog):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
mod_role: Union[Role, None] = ds_utils.get(
|
mod_role: Role | None = ds_utils.get(
|
||||||
ctx.user.guild.roles, id=await config_get("moderators", "roles")
|
ctx.user.guild.roles, id=await config_get("moderators", "roles")
|
||||||
)
|
)
|
||||||
admin_chan: Union[TextChannel, None] = ds_utils.get(
|
admin_chan: TextChannel | None = ds_utils.get(
|
||||||
ctx.user.guild.channels,
|
ctx.user.guild.channels,
|
||||||
id=await config_get("adminchat", "channels", "text"),
|
id=await config_get("adminchat", "channels", "text"),
|
||||||
)
|
)
|
||||||
|
@@ -1,4 +1,5 @@
|
|||||||
import logging
|
import logging
|
||||||
|
from logging import Logger
|
||||||
from typing import Dict, List, Any
|
from typing import Dict, List, Any
|
||||||
|
|
||||||
from discord import Cog, Message
|
from discord import Cog, Message
|
||||||
@@ -7,7 +8,7 @@ from discord.ext import commands
|
|||||||
from classes.holo_bot import HoloBot
|
from classes.holo_bot import HoloBot
|
||||||
from modules.database import col_analytics
|
from modules.database import col_analytics
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger: Logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class Analytics(commands.Cog):
|
class Analytics(commands.Cog):
|
||||||
@@ -16,11 +17,13 @@ class Analytics(commands.Cog):
|
|||||||
|
|
||||||
@Cog.listener()
|
@Cog.listener()
|
||||||
async def on_message(self, message: Message) -> None:
|
async def on_message(self, message: Message) -> None:
|
||||||
|
"""Listener that collects analytical data (stickers, attachments, messages)."""
|
||||||
if (
|
if (
|
||||||
(message.author != self.client.user)
|
(message.author != self.client.user)
|
||||||
and (message.author.bot is False)
|
and (message.author.bot is False)
|
||||||
and (message.author.system is False)
|
and (message.author.system is False)
|
||||||
):
|
):
|
||||||
|
# Handle stickers
|
||||||
stickers: List[Dict[str, Any]] = []
|
stickers: List[Dict[str, Any]] = []
|
||||||
|
|
||||||
for sticker in message.stickers:
|
for sticker in message.stickers:
|
||||||
@@ -33,6 +36,7 @@ class Analytics(commands.Cog):
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Handle attachments
|
||||||
attachments: List[Dict[str, Any]] = []
|
attachments: List[Dict[str, Any]] = []
|
||||||
|
|
||||||
for attachment in message.attachments:
|
for attachment in message.attachments:
|
||||||
@@ -49,6 +53,7 @@ class Analytics(commands.Cog):
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Insert entry into the database
|
||||||
await col_analytics.insert_one(
|
await col_analytics.insert_one(
|
||||||
{
|
{
|
||||||
"user": message.author.id,
|
"user": message.author.id,
|
||||||
|
@@ -1,13 +1,13 @@
|
|||||||
import logging
|
import logging
|
||||||
from typing import Any, Dict, Union
|
from logging import Logger
|
||||||
|
from typing import Any, Dict
|
||||||
|
|
||||||
from discord import ApplicationContext, Embed, option, TextChannel, Role
|
from discord import ApplicationContext, Embed, option, TextChannel, Role
|
||||||
from discord import utils as ds_utils
|
from discord import utils as ds_utils
|
||||||
from discord.abc import GuildChannel
|
from discord.abc import GuildChannel
|
||||||
from discord.commands import SlashCommandGroup
|
from discord.commands import SlashCommandGroup
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from libbot import config_get
|
from libbot.utils import config_get
|
||||||
from libbot.sync import config_get as sync_config_get
|
|
||||||
|
|
||||||
from classes.holo_bot import HoloBot
|
from classes.holo_bot import HoloBot
|
||||||
from classes.holo_user import HoloUser
|
from classes.holo_user import HoloUser
|
||||||
@@ -15,7 +15,7 @@ from enums import Color
|
|||||||
from modules.database import col_users
|
from modules.database import col_users
|
||||||
from modules.utils_sync import guild_name
|
from modules.utils_sync import guild_name
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger: Logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class CustomChannels(commands.Cog):
|
class CustomChannels(commands.Cog):
|
||||||
@@ -35,7 +35,7 @@ class CustomChannels(commands.Cog):
|
|||||||
@custom_channel_group.command(
|
@custom_channel_group.command(
|
||||||
name="get",
|
name="get",
|
||||||
description="Отримати персональний текстовий канал",
|
description="Отримати персональний текстовий канал",
|
||||||
guild_ids=[sync_config_get("guild")],
|
guild_ids=[config_get("guild")],
|
||||||
)
|
)
|
||||||
@option("name", description="Назва каналу")
|
@option("name", description="Назва каналу")
|
||||||
@option("reactions", description="Дозволити реакції")
|
@option("reactions", description="Дозволити реакції")
|
||||||
@@ -43,6 +43,10 @@ class CustomChannels(commands.Cog):
|
|||||||
async def custom_channel_get_cmd(
|
async def custom_channel_get_cmd(
|
||||||
self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool
|
self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool
|
||||||
) -> None:
|
) -> None:
|
||||||
|
"""Command /customchannel get <name> <reactions> <threads>
|
||||||
|
|
||||||
|
Command to create a custom channel for a user.
|
||||||
|
"""
|
||||||
holo_user_ctx: HoloUser = HoloUser(ctx.user)
|
holo_user_ctx: HoloUser = HoloUser(ctx.user)
|
||||||
|
|
||||||
# Return if the user is using the command outside of a guild
|
# Return if the user is using the command outside of a guild
|
||||||
@@ -109,9 +113,7 @@ class CustomChannels(commands.Cog):
|
|||||||
bots: Dict[str, Any] = await config_get("bots")
|
bots: Dict[str, Any] = await config_get("bots")
|
||||||
|
|
||||||
for bot in bots:
|
for bot in bots:
|
||||||
role: Union[Role, None] = ds_utils.get(
|
role: Role | None = ds_utils.get(ctx.user.guild.roles, id=bots[bot]["role"])
|
||||||
ctx.user.guild.roles, id=bots[bot]["role"]
|
|
||||||
)
|
|
||||||
|
|
||||||
if role is not None:
|
if role is not None:
|
||||||
await created_channel.set_permissions(
|
await created_channel.set_permissions(
|
||||||
@@ -122,7 +124,7 @@ class CustomChannels(commands.Cog):
|
|||||||
@custom_channel_group.command(
|
@custom_channel_group.command(
|
||||||
name="edit",
|
name="edit",
|
||||||
description="Змінити параметри особистого каналу",
|
description="Змінити параметри особистого каналу",
|
||||||
guild_ids=[sync_config_get("guild")],
|
guild_ids=[config_get("guild")],
|
||||||
)
|
)
|
||||||
@option("name", description="Назва каналу")
|
@option("name", description="Назва каналу")
|
||||||
@option("reactions", description="Дозволити реакції")
|
@option("reactions", description="Дозволити реакції")
|
||||||
@@ -130,9 +132,13 @@ class CustomChannels(commands.Cog):
|
|||||||
async def custom_channel_edit_cmd(
|
async def custom_channel_edit_cmd(
|
||||||
self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool
|
self, ctx: ApplicationContext, name: str, reactions: bool, threads: bool
|
||||||
) -> None:
|
) -> None:
|
||||||
|
"""Command /customchannel edit <name> <reactions> <threads>
|
||||||
|
|
||||||
|
Command to change properties of a custom channel.
|
||||||
|
"""
|
||||||
holo_user_ctx: HoloUser = HoloUser(ctx.user)
|
holo_user_ctx: HoloUser = HoloUser(ctx.user)
|
||||||
|
|
||||||
custom_channel: Union[TextChannel, None] = ds_utils.get(
|
custom_channel: TextChannel | None = ds_utils.get(
|
||||||
ctx.guild.channels, id=holo_user_ctx.customchannel
|
ctx.guild.channels, id=holo_user_ctx.customchannel
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -167,12 +173,15 @@ class CustomChannels(commands.Cog):
|
|||||||
@custom_channel_group.command(
|
@custom_channel_group.command(
|
||||||
name="remove",
|
name="remove",
|
||||||
description="Відібрати канал, знищуючи його, та частково повернути кошти",
|
description="Відібрати канал, знищуючи його, та частково повернути кошти",
|
||||||
guild_ids=[sync_config_get("guild")],
|
guild_ids=[config_get("guild")],
|
||||||
)
|
)
|
||||||
@option("confirm", description="Підтвердження операції")
|
@option("confirm", description="Підтвердження операції")
|
||||||
async def custom_channel_remove_cmd(
|
async def custom_channel_remove_cmd(
|
||||||
self, ctx: ApplicationContext, confirm: bool = False
|
self, ctx: ApplicationContext, confirm: bool = False
|
||||||
) -> None:
|
) -> None:
|
||||||
|
"""Command /customchannel remove [<confirm>]
|
||||||
|
|
||||||
|
Command to remove a custom channel. Requires additional confirmation."""
|
||||||
holo_user_ctx: HoloUser = HoloUser(ctx.user)
|
holo_user_ctx: HoloUser = HoloUser(ctx.user)
|
||||||
|
|
||||||
# Return if the user does not have a custom channel
|
# Return if the user does not have a custom channel
|
||||||
@@ -189,7 +198,7 @@ class CustomChannels(commands.Cog):
|
|||||||
|
|
||||||
await ctx.defer()
|
await ctx.defer()
|
||||||
|
|
||||||
custom_channel: Union[TextChannel, None] = ds_utils.get(
|
custom_channel: TextChannel | None = ds_utils.get(
|
||||||
ctx.guild.channels, id=holo_user_ctx.customchannel
|
ctx.guild.channels, id=holo_user_ctx.customchannel
|
||||||
)
|
)
|
||||||
|
|
||||||
|
30
cogs/data.py
30
cogs/data.py
@@ -2,16 +2,14 @@ import logging
|
|||||||
from logging import Logger
|
from logging import Logger
|
||||||
from os import makedirs
|
from os import makedirs
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Union, List, Dict, Any
|
from typing import List, Dict, Any
|
||||||
from uuid import uuid4
|
from uuid import uuid4
|
||||||
|
|
||||||
from discord import ApplicationContext, Embed, File, option, Role, TextChannel
|
from discord import ApplicationContext, Embed, File, option, Role, TextChannel
|
||||||
from discord import utils as ds_utils
|
from discord import utils as ds_utils
|
||||||
from discord.commands import SlashCommandGroup
|
from discord.commands import SlashCommandGroup
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from libbot import config_get
|
from libbot.utils import config_get, json_write
|
||||||
from libbot.sync import config_get as sync_config_get
|
|
||||||
from libbot.sync import json_write as sync_json_write
|
|
||||||
|
|
||||||
from classes.holo_bot import HoloBot
|
from classes.holo_bot import HoloBot
|
||||||
from classes.holo_user import HoloUser
|
from classes.holo_user import HoloUser
|
||||||
@@ -31,12 +29,15 @@ class Data(commands.Cog):
|
|||||||
@data.command(
|
@data.command(
|
||||||
name="export",
|
name="export",
|
||||||
description="Експортувати дані",
|
description="Експортувати дані",
|
||||||
guild_ids=[sync_config_get("guild")],
|
guild_ids=[config_get("guild")],
|
||||||
)
|
)
|
||||||
@option(
|
@option(
|
||||||
"kind", description="Тип даних, які треба експортувати", choices=["Користувачі"]
|
"kind", description="Тип даних, які треба експортувати", choices=["Користувачі"]
|
||||||
)
|
)
|
||||||
async def data_export_cmd(self, ctx: ApplicationContext, kind: str) -> None:
|
async def data_export_cmd(self, ctx: ApplicationContext, kind: str) -> None:
|
||||||
|
"""Command /data export <kind>
|
||||||
|
|
||||||
|
Command to export specific kind of data."""
|
||||||
await ctx.defer()
|
await ctx.defer()
|
||||||
|
|
||||||
# Return if the user is not an owner and not in the council
|
# Return if the user is not an owner and not in the council
|
||||||
@@ -56,10 +57,10 @@ class Data(commands.Cog):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
mod_role: Union[Role, None] = ds_utils.get(
|
mod_role: Role | None = ds_utils.get(
|
||||||
ctx.user.guild.roles, id=await config_get("moderators", "roles")
|
ctx.user.guild.roles, id=await config_get("moderators", "roles")
|
||||||
)
|
)
|
||||||
admin_chan: Union[TextChannel, None] = ds_utils.get(
|
admin_chan: TextChannel | None = ds_utils.get(
|
||||||
ctx.user.guild.channels,
|
ctx.user.guild.channels,
|
||||||
id=await config_get("adminchat", "channels", "text"),
|
id=await config_get("adminchat", "channels", "text"),
|
||||||
)
|
)
|
||||||
@@ -94,19 +95,26 @@ class Data(commands.Cog):
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
sync_json_write(users, Path(f"tmp/{uuid}"))
|
# Temporary file must be written synchronously,
|
||||||
|
# otherwise it will not be there when ctx.respond() is be called
|
||||||
|
json_write(users, Path(f"tmp/{uuid}"))
|
||||||
|
|
||||||
await ctx.respond(file=File(Path(f"tmp/{uuid}"), filename="users.json"))
|
await ctx.respond(file=File(Path(f"tmp/{uuid}"), filename="users.json"))
|
||||||
|
|
||||||
@data.command(
|
@data.command(
|
||||||
name="migrate",
|
name="migrate",
|
||||||
description="Мігрувати всіх користувачів до бази",
|
description="Мігрувати всіх користувачів до бази",
|
||||||
guild_ids=[sync_config_get("guild")],
|
guild_ids=[config_get("guild")],
|
||||||
)
|
)
|
||||||
@option(
|
@option(
|
||||||
"kind", description="Тип даних, які треба експортувати", choices=["Користувачі"]
|
"kind", description="Тип даних, які треба експортувати", choices=["Користувачі"]
|
||||||
)
|
)
|
||||||
async def data_migrate_cmd(self, ctx: ApplicationContext, kind: str) -> None:
|
async def data_migrate_cmd(self, ctx: ApplicationContext, kind: str) -> None:
|
||||||
|
"""Command /migrate <kind>
|
||||||
|
|
||||||
|
Command to migrate specific kind of data.
|
||||||
|
|
||||||
|
Migration of users in this case means creation of their DB entries."""
|
||||||
await ctx.defer()
|
await ctx.defer()
|
||||||
|
|
||||||
# Return if the user is not an owner and not in the council
|
# Return if the user is not an owner and not in the council
|
||||||
@@ -126,10 +134,10 @@ class Data(commands.Cog):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
mod_role: Union[Role, None] = ds_utils.get(
|
mod_role: Role | None = ds_utils.get(
|
||||||
ctx.user.guild.roles, id=await config_get("moderators", "roles")
|
ctx.user.guild.roles, id=await config_get("moderators", "roles")
|
||||||
)
|
)
|
||||||
admin_chan: Union[TextChannel, None] = ds_utils.get(
|
admin_chan: TextChannel | None = ds_utils.get(
|
||||||
ctx.user.guild.channels,
|
ctx.user.guild.channels,
|
||||||
id=await config_get("adminchat", "channels", "text"),
|
id=await config_get("adminchat", "channels", "text"),
|
||||||
)
|
)
|
||||||
|
13
cogs/fun.py
13
cogs/fun.py
@@ -1,15 +1,15 @@
|
|||||||
import logging
|
import logging
|
||||||
|
from logging import Logger
|
||||||
|
|
||||||
from discord import ApplicationContext, Embed, User, option, slash_command
|
from discord import ApplicationContext, Embed, User, option, slash_command
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from libbot import config_get
|
from libbot.utils import config_get
|
||||||
from libbot.sync import config_get as sync_config_get
|
|
||||||
|
|
||||||
from classes.holo_bot import HoloBot
|
from classes.holo_bot import HoloBot
|
||||||
from modules.utils_sync import guild_name
|
from modules.utils_sync import guild_name
|
||||||
from modules.waifu_pics import waifu_pics
|
from modules.waifu_pics import waifu_pics
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger: Logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class Fun(commands.Cog):
|
class Fun(commands.Cog):
|
||||||
@@ -19,15 +19,18 @@ class Fun(commands.Cog):
|
|||||||
@slash_command(
|
@slash_command(
|
||||||
name="action",
|
name="action",
|
||||||
description="Провести над користувачем РП дію",
|
description="Провести над користувачем РП дію",
|
||||||
guild_ids=[sync_config_get("guild")],
|
guild_ids=[config_get("guild")],
|
||||||
)
|
)
|
||||||
@option(
|
@option(
|
||||||
"type",
|
"type",
|
||||||
description="Тип дії, яку хочете провести з користувачем",
|
description="Тип дії, яку хочете провести з користувачем",
|
||||||
choices=sync_config_get("actions").keys(),
|
choices=config_get("actions").keys(),
|
||||||
)
|
)
|
||||||
@option("user", description="Користувач")
|
@option("user", description="Користувач")
|
||||||
async def action_cmd(self, ctx: ApplicationContext, type: str, user: User) -> None:
|
async def action_cmd(self, ctx: ApplicationContext, type: str, user: User) -> None:
|
||||||
|
"""Command /action <type> <user>
|
||||||
|
|
||||||
|
Command to perform some RP action on a user and send them a GIF."""
|
||||||
await ctx.defer()
|
await ctx.defer()
|
||||||
|
|
||||||
action: str = await config_get("category", "actions", type)
|
action: str = await config_get("category", "actions", type)
|
||||||
|
@@ -1,13 +1,17 @@
|
|||||||
from typing import Dict, Any, Union
|
import logging
|
||||||
|
from logging import Logger
|
||||||
|
from typing import Dict, Any
|
||||||
|
|
||||||
from discord import Member, Message, TextChannel
|
from discord import Member, Message, TextChannel, MessageType
|
||||||
from discord import utils as ds_utils
|
from discord import utils as ds_utils
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from libbot import config_get
|
from libbot.utils import config_get
|
||||||
|
|
||||||
from classes.holo_bot import HoloBot
|
from classes.holo_bot import HoloBot
|
||||||
from modules.database import col_users
|
from modules.database import col_users
|
||||||
|
|
||||||
|
logger: Logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class Logger(commands.Cog):
|
class Logger(commands.Cog):
|
||||||
def __init__(self, client: HoloBot):
|
def __init__(self, client: HoloBot):
|
||||||
@@ -15,6 +19,7 @@ class Logger(commands.Cog):
|
|||||||
|
|
||||||
@commands.Cog.listener()
|
@commands.Cog.listener()
|
||||||
async def on_message(self, message: Message):
|
async def on_message(self, message: Message):
|
||||||
|
"""Message listener. All actions on messages remain here for now."""
|
||||||
if (
|
if (
|
||||||
(message.author != self.client.user)
|
(message.author != self.client.user)
|
||||||
and (message.author.bot is False)
|
and (message.author.bot is False)
|
||||||
@@ -31,17 +36,42 @@ class Logger(commands.Cog):
|
|||||||
|
|
||||||
await col_users.insert_one(document=user)
|
await col_users.insert_one(document=user)
|
||||||
|
|
||||||
|
if (
|
||||||
|
(message.type == MessageType.thread_created)
|
||||||
|
and (message.channel is not None)
|
||||||
|
and (
|
||||||
|
await col_users.count_documents({"customchannel": message.channel.id})
|
||||||
|
> 0
|
||||||
|
)
|
||||||
|
):
|
||||||
|
try:
|
||||||
|
logger.info(
|
||||||
|
"Deleting the thread creation message in a custom channel %s",
|
||||||
|
message.channel.id,
|
||||||
|
)
|
||||||
|
await message.delete()
|
||||||
|
except Exception as exc:
|
||||||
|
logger.warning(
|
||||||
|
"Could not delete the thread creation message in a custom channel %s due to %s",
|
||||||
|
message.channel.id,
|
||||||
|
exc,
|
||||||
|
)
|
||||||
|
|
||||||
@commands.Cog.listener()
|
@commands.Cog.listener()
|
||||||
async def on_member_join(self, member: Member) -> None:
|
async def on_member_join(self, member: Member) -> None:
|
||||||
welcome_chan: Union[TextChannel, None] = ds_utils.get(
|
"""Member join handler. All actions on member join remain here for now."""
|
||||||
|
welcome_chan: TextChannel | None = ds_utils.get(
|
||||||
self.client.get_guild(await config_get("guild")).channels,
|
self.client.get_guild(await config_get("guild")).channels,
|
||||||
id=await config_get("welcome", "channels", "text"),
|
id=await config_get("welcome", "channels", "text"),
|
||||||
)
|
)
|
||||||
rules_chan: Union[TextChannel, None] = ds_utils.get(
|
rules_chan: TextChannel | None = ds_utils.get(
|
||||||
self.client.get_guild(await config_get("guild")).channels,
|
self.client.get_guild(await config_get("guild")).channels,
|
||||||
id=await config_get("rules", "channels", "text"),
|
id=await config_get("rules", "channels", "text"),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if welcome_chan is None:
|
||||||
|
logger.warning("Could not find a welcome channel by its id")
|
||||||
|
|
||||||
if (
|
if (
|
||||||
(member != self.client.user)
|
(member != self.client.user)
|
||||||
and (member.bot is False)
|
and (member.bot is False)
|
||||||
|
@@ -3,7 +3,7 @@ from logging import Logger
|
|||||||
|
|
||||||
from discord import Activity, ActivityType
|
from discord import Activity, ActivityType
|
||||||
from discord.ext import commands
|
from discord.ext import commands
|
||||||
from libbot import config_get
|
from libbot.utils import config_get
|
||||||
|
|
||||||
from classes.holo_bot import HoloBot
|
from classes.holo_bot import HoloBot
|
||||||
|
|
||||||
@@ -16,6 +16,7 @@ class Utility(commands.Cog):
|
|||||||
|
|
||||||
@commands.Cog.listener()
|
@commands.Cog.listener()
|
||||||
async def on_ready(self) -> None:
|
async def on_ready(self) -> None:
|
||||||
|
"""Listener for the event when bot connects to Discord and becomes "ready"."""
|
||||||
logger.info("Logged in as %s", self.client.user)
|
logger.info("Logged in as %s", self.client.user)
|
||||||
|
|
||||||
activity_type: str = await config_get("type", "status")
|
activity_type: str = await config_get("type", "status")
|
||||||
|
4
main.py
4
main.py
@@ -4,7 +4,7 @@ from logging import Logger
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
from discord import LoginFailure, Intents
|
from discord import LoginFailure, Intents
|
||||||
from libbot.sync import config_get as sync_config_get
|
from libbot.utils import config_get
|
||||||
|
|
||||||
from classes.holo_bot import HoloBot
|
from classes.holo_bot import HoloBot
|
||||||
from modules.scheduler import scheduler
|
from modules.scheduler import scheduler
|
||||||
@@ -38,7 +38,7 @@ def main() -> None:
|
|||||||
client.load_extension("cogs")
|
client.load_extension("cogs")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
client.run(sync_config_get("bot_token", "bot"))
|
client.run(config_get("bot_token", "bot"))
|
||||||
except LoginFailure as exc:
|
except LoginFailure as exc:
|
||||||
logger.error("Provided bot token is invalid: %s", exc)
|
logger.error("Provided bot token is invalid: %s", exc)
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
|
@@ -1,12 +1,12 @@
|
|||||||
from typing import Dict, Any
|
from typing import Dict, Any
|
||||||
|
|
||||||
from async_pymongo import AsyncClient, AsyncCollection, AsyncDatabase
|
from async_pymongo import AsyncClient, AsyncCollection, AsyncDatabase
|
||||||
from libbot.sync import config_get as sync_config_get
|
from libbot.utils import config_get
|
||||||
from pymongo import MongoClient
|
from pymongo import MongoClient
|
||||||
from pymongo.synchronous.collection import Collection
|
from pymongo.synchronous.collection import Collection
|
||||||
from pymongo.synchronous.database import Database
|
from pymongo.synchronous.database import Database
|
||||||
|
|
||||||
db_config: Dict[str, Any] = sync_config_get("database")
|
db_config: Dict[str, Any] = config_get("database")
|
||||||
|
|
||||||
con_string: str = (
|
con_string: str = (
|
||||||
"mongodb://{0}:{1}/{2}".format(
|
"mongodb://{0}:{1}/{2}".format(
|
||||||
|
@@ -1,9 +1,7 @@
|
|||||||
from typing import Union
|
|
||||||
|
|
||||||
from discord import Member, User
|
from discord import Member, User
|
||||||
|
|
||||||
|
|
||||||
def guild_name(member: Union[Member, User]) -> str:
|
def guild_name(member: Member | User) -> str:
|
||||||
if isinstance(member, User):
|
if isinstance(member, User):
|
||||||
return member.name
|
return member.name
|
||||||
|
|
||||||
|
@@ -5,6 +5,7 @@ requests>=2.32.2
|
|||||||
aiofiles~=24.1.0
|
aiofiles~=24.1.0
|
||||||
apscheduler>=3.10.0
|
apscheduler>=3.10.0
|
||||||
async_pymongo==0.1.11
|
async_pymongo==0.1.11
|
||||||
libbot[speed,pycord]==3.3.1
|
libbot[speed,pycord]==4.0.0
|
||||||
|
typing-extensions~=4.12.2
|
||||||
ujson~=5.10.0
|
ujson~=5.10.0
|
||||||
WaifuPicsPython==0.2.0
|
WaifuPicsPython==0.2.0
|
Reference in New Issue
Block a user