TotalFreedomBot/functions.py
2020-12-18 18:20:05 +01:00

87 lines
2.1 KiB
Python

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 = [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 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