mirror of
https://github.com/TotalFreedomMC/TotalFreedomBot.git
synced 2024-12-22 15:44:57 +00:00
111 lines
2.8 KiB
Python
111 lines
2.8 KiB
Python
import json
|
|
import requests
|
|
import os
|
|
import itertools
|
|
|
|
from discord.ext import commands
|
|
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 = [f'{l[i]}' for i in range(len(l))]
|
|
l_names = [name.replace('_', '\_') for name in l_names]
|
|
|
|
em = embed_entry(
|
|
name=name,
|
|
value=", ".join(l_names),
|
|
playercount = len(l)
|
|
)
|
|
return em
|
|
|
|
|
|
def get_prefix(client, message):
|
|
#prefixes = ['TF!', 'Tf!', 'tF!', 'tf!']
|
|
prefix = os.getenv('prefix')
|
|
prefixes = map(''.join, itertools.product(*((letter.upper(), letter.lower()) for letter in prefix)))
|
|
return commands.when_mentioned_or(*prefixes)(client, message)
|
|
|
|
|
|
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, server=1):
|
|
config_file = read_json('config')
|
|
if server == 1:
|
|
ip = config_file['SERVER_IP']
|
|
pw = config_file['ENDPOINTS_PW']
|
|
else:
|
|
ip = config_file['SERVER_IP_2']
|
|
pw = config_file['ENDPOINTS_PW_2']
|
|
port = config_file['ENDPOINTS_PORT']
|
|
url = f"http://{ip}:{port}?password={pw}&command={command}"
|
|
payload = {}
|
|
headers = {}
|
|
try:
|
|
response = json.loads(requests.request(
|
|
"GET", url, headers=headers, data=payload, timeout=5).text)
|
|
except:
|
|
response = {'response': 'Connection Error.'}
|
|
return response['response']
|
|
|
|
def get_server_status(server=1):
|
|
config_file = read_json('config')
|
|
if server == 1:
|
|
ip = config_file['SERVER_IP']
|
|
else:
|
|
ip = config_file['SERVER_IP_2']
|
|
port = config_file['PLAYERLIST_PORT']
|
|
try:
|
|
requests.get(f"http://{ip}:{port}/list?json=true", timeout=5).json()
|
|
except:
|
|
return False
|
|
else:
|
|
return True
|