Added logger thats adds users to DB
This commit is contained in:
parent
b3a78816f7
commit
32c7ec7d44
49
cogs/logger.py
Normal file
49
cogs/logger.py
Normal file
@ -0,0 +1,49 @@
|
|||||||
|
from discord import Member, Message
|
||||||
|
from discord import utils as ds_utils
|
||||||
|
from discord.ext import commands
|
||||||
|
|
||||||
|
from modules.database import col_users
|
||||||
|
from modules.utils import config_get
|
||||||
|
|
||||||
|
|
||||||
|
class Logger(commands.Cog):
|
||||||
|
|
||||||
|
def __init__(self, client):
|
||||||
|
self.client = client
|
||||||
|
|
||||||
|
@commands.Cog.listener()
|
||||||
|
async def on_message(self, message: Message):
|
||||||
|
|
||||||
|
if (message.author != self.client.user) and (message.author.bot == False) and (message.author.system == False):
|
||||||
|
|
||||||
|
if col_users.find_one({"user": message.author.id}) is None:
|
||||||
|
user = {}
|
||||||
|
defaults = await config_get("user", "defaults")
|
||||||
|
|
||||||
|
user["user"] = message.author.id
|
||||||
|
|
||||||
|
for key in defaults:
|
||||||
|
user[key] = defaults[key]
|
||||||
|
|
||||||
|
col_users.insert_one(document=user)
|
||||||
|
|
||||||
|
@commands.Cog.listener()
|
||||||
|
async def on_member_join(self, member: Member):
|
||||||
|
|
||||||
|
welcome_chan = ds_utils.get(self.client.get_guild(await config_get("guild")).channels, id=await config_get("logging", "channels", "text"))
|
||||||
|
rules_chan = ds_utils.get(self.client.get_guild(await config_get("guild")).channels, id=await config_get("rules", "channels", "text"))
|
||||||
|
|
||||||
|
if (member != self.client.user) and (member.bot == False) and (member.system == False):
|
||||||
|
|
||||||
|
await welcome_chan.send(content=(await config_get("welcome", "messages")).format(mention=member.mention, rules=rules_chan.mention))
|
||||||
|
|
||||||
|
if col_users.find_one({"user": member.id}) is None:
|
||||||
|
col_users.insert_one(document=user)
|
||||||
|
|
||||||
|
user = {}
|
||||||
|
defaults = await config_get("user", "defaults")
|
||||||
|
|
||||||
|
user["user"] = member.id
|
||||||
|
|
||||||
|
for key in defaults:
|
||||||
|
user[key] = defaults[key]
|
2
main.py
2
main.py
@ -9,6 +9,7 @@ from modules.utils import config_get
|
|||||||
from modules.utils_sync import config_get_sync
|
from modules.utils_sync import config_get_sync
|
||||||
|
|
||||||
from cogs.custom_channels import CustomChannels
|
from cogs.custom_channels import CustomChannels
|
||||||
|
from cogs.logger import Logger
|
||||||
|
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
level=logging.INFO,
|
level=logging.INFO,
|
||||||
@ -36,6 +37,7 @@ async def on_ready():
|
|||||||
|
|
||||||
def main():
|
def main():
|
||||||
client.add_cog(CustomChannels(client))
|
client.add_cog(CustomChannels(client))
|
||||||
|
client.add_cog(Logger(client))
|
||||||
|
|
||||||
try:
|
try:
|
||||||
scheduler.start()
|
scheduler.start()
|
||||||
|
Loading…
Reference in New Issue
Block a user