utils.py

"""
Created by Epic at 10/13/20
"""
from cache import CacheElement

from logging import getLogger
from discord import Guild, User, Member

logger = getLogger("AQue.utils")


def get_object_id(args):
    total = 0
    for arg in args:
        if type(arg) in [Guild, User, Member]:
            total += arg.id
    return str(total)


def cacheable(cache_id, *, expire=10 * 60):
    def outer(func):
        def inner(*args, **kwargs):
            object_id = cache_id + "." + get_object_id(args)
            cache = CacheElement(object_id, expire=expire)
            cache_value = cache.get()
            if cache_value is None:
                cache_value = func(*args, **kwargs)
                cache.set(cache_value)
            return cache_value

        return inner

    return outer


def remove_cache(cache_id, args):
    object_id = str(sum([hash(arg) for arg in args]))
    cache_path = cache_id + "." + object_id
    cache = CacheElement(cache_path, expire=1)
    cache.set(None)


def get_matchmaking_type_by_id(channel_id, config):
    for match_type, voice_id in config.items():
        if voice_id == channel_id:
            return match_type