spacebot2.py

from discord.ext import commands, tasks
from discord import Intents, User, Reaction, Message, TextChannel, NotFound
from discord.ext.commands import CommandNotFound, Context, CommandInvokeError
from emojis import decode
from discord import Embed

from discord.utils import get

from persistent import server_config, sql
from util.logging import log, LogLevel
import asyncio
import simplejson as json

# settings.py
from dotenv import load_dotenv
load_dotenv()

import os
from datetime import datetime, timedelta

LIVETOKEN = os.getenv("LIVETOKEN")
DEVTOKEN = os.getenv("DEVTOKEN")

# dev token, please start using environment variables...
#Different token
DISCORD_TOKEN = LIVETOKEN


TOKEN = DISCORD_TOKEN
VERSION = "0.2.7"
COMMAND_PREFIX = "."

intents = Intents.all()

client = commands.Bot(command_prefix=COMMAND_PREFIX, intents=intents)
client.remove_command('help')

@tasks.loop(seconds = 120)
async def st():
    with open("suspend.log", 'r+') as sl:
        data = json.loads(sl.read())
        guild = get(client.guilds, id=522815906376843274)
        suspended_role = get(guild.roles, id=776320653581877278)
        mr = get(guild.roles, id=522817975091462147)
        dellist = []
        for uid in data:
            try:
                member = await guild.fetch_member(uid)
                if data[uid]['dur'] <= 0:
                    print("unsupending user")
                    if member is None:
                        print("Error Null Member")
                    else:
                        try:
                            await member.remove_roles(suspended_role)
                            if suspended_role is not None:
                                await member.add_roles(mr)
                                logchannel = get(guild.channels, id=761788719685435404)
                                suschannel = get(guild.channels, id=763644055536009216)
                                embed=Embed(title="User Unsuspended", description="**%s**, He has served his time \n Unsuspender: **SpaceBot**" % (member.display_name))

                                await suschannel.send(embed=embed)
                                await logchannel.send(embed=embed)
                        except Exception as e:
                            await log("An unexpected Error has occured during unsuspend while removing user roles.")
                    dellist.append(uid)
                else:
                    data[uid]['dur'] = data[uid]['dur'] - 2
            except NotFound:
                dellist.append(uid)
            except Exception as e:
                await log("An unexpected Error has occured during unsuspend tread.")
                await log(e)
        for uid in dellist:
            data.pop(uid, None)
        sl.seek(0)
        json.dump(data, sl)
        sl.truncate()
        sl.close()
        
'''
@tasks.loop(seconds = 86400)
async def pt():
    await log("rollover thread %s" %  (datetime.now().weekday()))
    if datetime.now().weekday() == 0:
        guild = get(client.guilds, id=522815906376843274)
        logchannel = get(guild.channels, id=761788719685435404)
        staffinfo  = get(guild.channels, id=816025151178670090)
        
        embed = Embed(title="Monday Quota Bot", description="=========================")
        
        embed.add_field(name = "%s: Quota has come are you ready to parse staff?" % (datetime.now()),
        value = "This will start the rollover, (STAFF WILL NOT BE AUTO DM'ED)",
        inline=False)
        msg = await staffinfo.send(content = "<@&761211992438472744> <@&522816654091223051>", embed=embed)

        await msg.add_reaction("✔")
        await msg.add_reaction("❌")

        def check(obj, user = 0):
            if obj.message != msg:
                return False
            elif user.bot:
                return False
            else:
                return True


        contract = await client.wait_for('reaction_add', check=check)

        await msg.clear_reactions()

        if contract[0].emoji == "✔":
            data = await sql.rollover()
            
            for user in data:
                try:
                    member = guild.get_member(int(user[0]))
                    e = Embed(
                        title="Inactivity Alert", 
                        description=''
                        **%s did not meet quota.**
                        Their weekly quota was: **%s** points, but they only had **%s** points.\n
                        Their roles are: %s \n 
                        If they shouldn't have a quota copy the following command \n
                        `.noquota %s` \n
                        When you are done with this card react with ❌
                        ' % (member.display_name,
                        40,
                        user[1], 
                        ", ".join([(role.name) for role in member.roles]),
                        member.id), 
                        color=0xdb021c)
                    e.set_footer(text="Space Travel Dungeons", icon_url = "https://cdn.discordapp.com/attachments/751589431441490082/764948382912479252/SPACE.gif")
                    await logchannel.send(embed = e)
                    k = await staffinfo.send(embed = e)

                    def kcheck(obj, user = 0):
                        if obj.message != k:
                            return False
                        elif user.bot:
                            return False
                        else:
                            return True
                    await msg.add_reaction("❌")

                    contract = await client.wait_for('reaction_add', check=kcheck)

                    await k.delete()

                    e = Embed(
                        title="Quota Notice", 
                        description='<@!%s> has completed quota check on <@!%s>',
                        color=0xdb021c)
                    await logchannel.send( embed = e)
                except Exception:
                    print("Opps something went wrong")
                
            await log("Rolling over")
        else:
            embed = Embed(title="Monday Quota Bot", description="=========================")
        
            embed.add_field(name = "%s: %s?" % (datetime.now(), contract[1].display_name),
            value = "You have canceled rollover this week, dm @Arceye if this was a mistake",
            inline=False)
            await msg.edit(embed=embed)
'''
        
        

@client.event
async def on_ready():
    await log(f'{client.user.name} v{VERSION} has connected to Discord!')
    await log("Attempting connection to SQL server...")
    await sql.connect()
    # Loads config for guilds the bot is currently a member of.
    await log("Loading config for connected guilds.")
    st.start()
    # pt.start()
    for guild in client.guilds:
        cfg = await server_config.get_config(guild)
        if cfg is not None:
            await log("Loaded config for guild " + guild.name + " with id " + str(guild.id) + ".")


@client.event
async def on_message(message: Message):
    user: str = f"{message.author.name}#{message.author.discriminator}"
    channel = "DMs"
    if message.guild is not None:
        channel = f"(guild {message.guild.name} with id {message.guild.id})"
    msg: str = message.content
    if msg.startswith(COMMAND_PREFIX):
        await log(f"{user} issued command {msg} in {channel}.")
        await client.process_commands(message)


@client.event
async def on_command_error(ctx: Context, error):
    if isinstance(error, CommandNotFound):
        pass
        #await ctx.send(f":x: **Command `{ctx.message.content.split()[0].replace(COMMAND_PREFIX, '')}` not found.**")
    #elif isinstance(error, CommandInvokeError):
    #    await ctx.send(":x: **Unable to invoke command.** Ensure you allow direct messages for server members.", delete_after=10)
    else:
        await log("An exception occurred while executing this command.", LogLevel.WARN)
        await log(error.__str__())


@client.event
async def on_member_join(member):
    pass


for filename in os.listdir('./cogs'):
    if filename.endswith('.py'):
        if filename != '__init__.py':
            print(f"Loading cog from {filename}.")
            client.load_extension(f'cogs.{filename[:-3]}')

client.run(TOKEN)