functions.py

import asyncio
import random
from variables import DEFAULT_PREFIX
from database import PREFIXES, LEVEL_DATABASE, PLUGINS, PERMISSIONS


async def get_prefix(ctx):
    prefix = await PREFIXES.find_one({"_id": ctx.guild.id})

    if prefix is not None:
        return prefix["prefix"]
    else:
        return DEFAULT_PREFIX


async def get_xp_range(guild_id):
    collection = LEVEL_DATABASE.get_collection(str(guild_id))
    settings = await collection.find_one({"_id": 0})
    return settings["xprange"]


async def get_random_text(typing_time):
    f = open("resources/words.txt").read()
    words = f.split("\n")

    if typing_time == 10:
        r = range(15)

    elif typing_time == 15:
        r = range(25)

    elif typing_time == 30:
        r = range(40)

    elif typing_time == 60:
        r = range(60)

    else:
        r = range(1)

    return " ".join(random.choice(words) for i in r)


def get_code(amount):
    return ''.join(
        random.choices(
            "ABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890", k=amount
        )
    )


"""
Some functions to counter errors and warnings while working locally :p

To get everything work properly database needs to be updates even if it's 
working locally on a single guild, this is because lots of places have major 
database dependencies.

First function simply updates all plugin and permissions documents with a new
plugin, only used when some new plugin is added, not required to use this 
function to fix any errors or warnings.

Second function does the main job, it checks for all plugin, permission, 
leveling (if enabled) and prefix documents, then updates/adds them if they
aren't there.

I would have loved to say that I did this intentionally to avoid people from 
stealing code but it was just me writing bad code which ended up benefiting
¯\_(ツ)_/¯
"""


# Adding new plugin and permissions
async def update_plugins_and_permissions(plugin):
    await PLUGINS.update_many(
        {plugin: {"$exists": False}},
        {
            "$set": {plugin: True}
        }
    )
    await PERMISSIONS.update_many(
        {plugin: {"$exists": False}},
        {
            "$set": {plugin: {}}
        }

    )


# updating leveling, plugin, prefix and permission data
async def update_db(guild_ids):
    plugins_update = []
    permissions_update = []
    leveling_update = []

    for guild_id in guild_ids:
        guild_plugins = await PLUGINS.find_one({"_id": guild_id})

        if not await PLUGINS.count_documents({"_id": guild_id}, limit=1):
            PLUGINS.insert_one({

                "_id": guild_id,
                "Leveling": False,
                "Moderation": True,
                "ReactionRoles": True,
                "Welcome": False,
                "Verification": False,
                "Chatbot": True,
                "AutoMod": False,
                "Karma": False,
                "Fun": True,
                "Giveaway": True
            })
            plugins_update.append(guild_id)
            print(f"✅{guild_id} - Plugins 🔧")

        if not await PERMISSIONS.count_documents({"_id": guild_id}, limit=1):
            PERMISSIONS.insert_one({

                "_id": guild_id,
                "Leveling": {},
                "Moderation": {},
                "ReactionRoles": {},
                "Welcome": {},
                "Verification": {},
                "Chatbot": {},
                "Commands": {},
                "AutoMod": {},
                "Karma": {},
                "Fun": {},
                "Giveaway": {}
            })
            permissions_update.append(guild_id)
            print(f"✅{guild_id} - Permissions 🔨")

        if (
            guild_plugins["Leveling"]
            and str(guild_id)
            not in await LEVEL_DATABASE.list_collection_names()
        ):
            guild_level_db = await LEVEL_DATABASE.create_collection(
                str(guild_id))
            await guild_level_db.insert_one({

                "_id": 0,
                "xprange": [15, 25],
                "alertchannel": None,
                "blacklistedchannels": [],
                "alerts": True
            })
            leveling_update.append(guild_id)
            print(f"✅{guild_id} - Leveling 📊")

        # Only use this when working locally
        try:
            await PREFIXES.insert_one({
                "_id": guild_id,
                "prefix": "ax"
            })
            print(f"✅{guild_id} - Prefix ⚪")

        except:
            print(f"❌{guild_id} - Prefix ⚪")

    print(
        f"Update results"
        f"\n{len(plugins_update)} plugins\n"
        f"{len(permissions_update)} permissions\n"
        f"{len(leveling_update)} leveling"
    )

# serveridlist = [843516084266729512, 751491708465840159]
# loop = asyncio.get_event_loop()
# loop.run_until_complete(updatedb(serveridlist))

# update_plugins_and_permissions("Giveaway")