TotalFreedomBot/functions.py

import json
import requests

from checks import *


class embed_entry:
    def __init__(self, name, value, *, playercount):
        self.name = name
        self.value = value
        if playercount:
            self.playercount = playercount

def format_list_entry(embed, l, name):
    l_names = []
    [l_names.append(f'`{l[i]}`') for i in range(len(l))]
    em = embed_entry(name=name,
                    value=", ".join(l_names),
                    playercount = len(l)
                    )
    return em


def did_mention_other_user(users, author):
    for user in users:
        if user is not author:
            return True
    return False


def removed_user_mentions(old, new):
    users = []
    for user in old:
        if user not in new:
            users.append(user)
    return users


def removed_role_mentions(old, new):
    roles = []
    for role in old:
        if role not in new:
            roles.append(role)
    return roles


def get_avatar(user, animate=True):
    if user.avatar_url:
        avatar = str(user.avatar_url).replace(".webp", ".png")
    else:
        avatar = str(user.default_avatar_url)
    if not animate:
        avatar = avatar.replace(".gif", ".png")
    return avatar


def read_json(file_name):
    with open(f'/root/totalfreedom/{file_name}.json', 'r') as file:
        data = json.load(file)
    return data


def write_json(file_name, data):
    with open(f'/root/totalfreedom/{file_name}.json', 'w') as file:
        json.dump(data, file, indent=4)
    return data


def hit_endpoint(command):
    url = f"http://play.totalfreedom.me:3000?password=CENSORED&command={command}"
    payload = {}
    headers = {}
    try:
        response = json.loads(requests.request(
            "GET", url, headers=headers, data=payload, timeout=100).text)
    except:
        response = 'Connection Error.'
    return response['response']

def get_server_status():
    try:
        requests.get("http://play.totalfreedom.me:28966/list?json=true", timeout=5).json()
    except:
        return False
    else:
        return True