TotalFreedomBot/main.py

98 lines
3.5 KiB
Python
Raw Normal View History

2020-10-28 11:47:39 +00:00
import discord
import os
import time
2020-11-21 12:38:59 +00:00
import logscript
import re
2020-10-28 11:47:39 +00:00
2020-11-21 13:44:46 +00:00
from unicode import *
2020-10-28 12:23:49 +00:00
from datetime import datetime
2020-10-28 11:47:39 +00:00
from discord.ext import commands
from dotenv import load_dotenv
2020-11-21 12:38:59 +00:00
from checks import *
from functions import *
print = logscript.logging.getLogger().critical
2020-10-28 11:47:39 +00:00
load_dotenv()
botToken = os.getenv('botToken')
intents = discord.Intents.all()
2021-01-04 23:43:48 +00:00
bot = commands.Bot(command_prefix=get_prefix, case_insensitive=True, description='TotalFreedom bot help command',
intents=intents)
2020-11-21 12:38:59 +00:00
2020-10-28 11:47:39 +00:00
extensions = [
2020-10-28 12:23:49 +00:00
"commands.moderation",
"commands.server_commands",
2020-10-28 11:47:39 +00:00
"commands.help",
2020-10-28 12:23:49 +00:00
"commands.miscellaneous",
"events"
2020-10-28 11:47:39 +00:00
]
if __name__ == '__main__':
for extension in extensions:
try:
bot.load_extension(extension)
print(
f"[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [Extensions] {extension} loaded successfully")
2020-10-28 11:47:39 +00:00
except Exception as e:
print(
f"[{str(datetime.utcnow().replace(microsecond=0))[11:]} INFO]: [Extensions] {extension} didn't load {e}")
2020-10-28 11:47:39 +00:00
2020-11-21 12:38:59 +00:00
@bot.event
async def on_message(message):
2020-12-19 15:54:24 +00:00
if isinstance(message.channel, discord.channel.DMChannel):
print(f'{message.author} DM: {message.content}')
return
2021-01-04 23:43:48 +00:00
elif message.guild and message.author is message.guild.me and message.channel.id == reports_channel_id:
2020-11-21 12:38:59 +00:00
await message.add_reaction(clipboard)
2021-01-04 23:43:48 +00:00
elif message.type == discord.MessageType.new_member:
if re.search('discord\.gg\/[a-zA-z0-9\-]{1,16}', message.author.name.lower()) or re.search(
'discordapp\.com\/invite\/[a-z0-9]+/ig', message.author.name.lower()):
2020-11-21 12:38:59 +00:00
await message.author.ban(reason="Name is an invite link.")
await message.delete()
bypass_roles = [discord_admin, discord_mod]
bypass = False
if message.author != bot.user:
for role in message.author.roles:
if role.id in bypass_roles:
bypass = True
else:
2021-01-04 23:43:48 +00:00
server_chats = {1: server_chat, 2: server_chat_2}
for server in range(1, len(server_chats)):
if message.channel.id == server_chats[server]:
if not get_server_status(server):
hit_endpoint('start', server)
if 'Server has started' in message.content: # Telnet reconnect script
2020-11-21 12:38:59 +00:00
try:
bot.telnet_object.connect()
except Exception as e:
print(f'Failed to reconnect telnet: {e}')
time.sleep(5)
try:
bot.telnet_object.connect()
except Exception as fuckup:
print(
f'Second attempt failed to reconnect telnet: {fuckup}')
2021-01-04 23:43:48 +00:00
try:
bot.telnet_object_2.connect()
except Exception as e:
print(f'Failed to reconnect telnet 2: {e}')
time.sleep(5)
try:
bot.telnet_object_2.connect()
except Exception as fuckup:
print(
f'Second attempt failed to reconnect telnet 2: {fuckup}')
2020-11-21 12:38:59 +00:00
if not bypass:
2021-01-04 23:43:48 +00:00
if re.search('discord\.gg\/[a-zA-z0-9\-]{1,16}', message.content) or re.search(
'discordapp\.com\/invite\/[a-z0-9]+/ig', message.content):
2020-11-21 12:38:59 +00:00
await message.delete()
await message.channel.send(f"{message.author.mention} do not post invite links to other discord servers.")
2020-12-19 15:54:24 +00:00
await bot.process_commands(message)
2020-11-21 12:38:59 +00:00
2021-01-04 23:43:48 +00:00
2020-10-28 12:23:49 +00:00
bot.run(botToken)