minecraft_manager/api/api.py

import socket, requests, logging, os, datetime, pytz, mcstatus, discord
from minecraft_manager.models import Alert
from django.contrib.auth.models import User
from django.conf import settings

logger = logging.getLogger(__name__)


def create_alert(message):
    users = User.objects.filter(is_active=True)
    for user in users:
        alert = Alert(user=user, message=message)
        alert.save()


PLUGIN_APPLICATION = 'application'
PLUGIN_ACCEPT = 'accept'
PLUGIN_DENY = 'deny'
PLUGIN_GLOBAL_CHAT = 'global'
PLUGIN_STAFF_CHAT = 'staff'
PLUGIN_DEMOTE = 'demote'


def plugin(key, command):
    host = '127.0.0.1'
    port = getattr(settings, 'PLUGIN_PORT', None)
    full_command = "{0} {1}".format(key, command)
    if port and plugin_exists():
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.connect((host, port))
        sock.sendall(full_command.encode('utf-8'))
        sock.close()


def discord_mcm(message='', embeds=None, ping=False):
    discord_mcm_webhook = getattr(settings, 'DISCORD_MCM_WEBHOOK', None)
    if discord_mcm_webhook:
        ping_list = getattr(settings, 'DISCORD_PING_LIST', [])
        if ping and ping_list:
            ping_list = ["<@&{0}>".format(ping) for ping in ping_list]
            message = "{0}\n{1}".format(" ".join(ping_list), message)
        data = {}
        if message:
            data['content'] = message
        if embeds:
            data['embeds'] = embeds
        return requests.post(discord_mcm_webhook, json=data)
    return None


def discord_notification(message='', embeds=None, ping=False):
    discord_notification_webhook = getattr(settings, 'DISCORD_NOTIFICATION_WEBHOOK', None)
    if discord_notification_webhook:
        ping_list = getattr(settings, 'DISCORD_PING_LIST', [])
        if ping and ping_list:
            ping_list = ["<@&{0}>".format(ping) for ping in ping_list]
            message = "{0}\n{1}".format(" ".join(ping_list), message)
        data = {}
        if message:
            data['content'] = message
        if embeds:
            data['embeds'] = embeds
        return requests.post(discord_notification_webhook, json=data)
    return None


def discord_webhook(webhook_name=None, message=None, embed=None, ping=False):
    webhook = getattr(settings, 'DISCORD_{}_WEBHOOK'.format(webhook_name.upper()), getattr(settings, 'DISCORD_WEBHOOK', None))
    if webhook and (message or embed):
        ping_list = getattr(settings, 'DISCORD_PING_LIST', [])
        if ping and ping_list:
            ping_list = ["<@&{0}>".format(ping) for ping in ping_list]
            message = "{0}\n{1}".format(" ".join(ping_list), message)
        data = {}
        if message:
            data['content'] = message
        if embed:
            data['embeds'] = embed
        return requests.post(webhook, json=data)
    return None



def strip_format(message):
    return message.replace("§0", "").replace("§1", "").replace("§2", "").replace("§3", "").replace("§4", "") \
        .replace("§5", "").replace("§6", "").replace("§7", "").replace("§8", "").replace("§9", "").replace("§a", "") \
        .replace("§b", "").replace("§c", "").replace("§d", "").replace("§e", "").replace("§f", "").replace("§r", "") \
        .replace("§k", "").replace("§l", "").replace("§m", "").replace("§n", "").replace("§o", "")


def plugin_exists():
    return os.path.exists(os.path.join(getattr(settings, 'MINECRAFT_BASE_DIR', ''), 'plugins/MinecraftManager'))


def get_chats(as_timezone):
    global_log = getattr(settings, 'GLOBAL_LOG', "")
    staff_log = getattr(settings, 'STAFF_LOG', "")
    log_time_format = getattr(settings, 'LOG_TIME_FORMAT', '%m/%d/%y %I:%M %p')
    if plugin_exists():
        try:
            with open(global_log, encoding='utf-8') as g:
                g_chat = g.readlines()[-100:]
        except OSError:
            g_chat = []
        for idx, gc in enumerate(g_chat):
            try:
                date_str = gc.split("] ")[0].replace("[", "")
                msg = gc.split("] ", 1)[1]
                dt = datetime.datetime.strptime(date_str, log_time_format)
                dt = pytz.timezone('US/Central').localize(dt)
                dt = dt.astimezone(pytz.timezone(as_timezone))
                g_chat[idx] = {"date": dt.strftime(log_time_format), "text": msg}
            except:
                g_chat[idx] = {"date": "01/01/17 00:00 AM", "name": "Error", "text": "Parsing Line"}
        try:
            with open(staff_log, encoding='utf-8') as s:
                s_chat = s.readlines()[-100:]
        except OSError:
            s_chat = []
        for idx, sc in enumerate(s_chat):
            try:
                date_str = sc.split("] ")[0].replace("[", "")
                msg = sc.split("] ", 1)[1]
                dt = datetime.datetime.strptime(date_str, log_time_format)
                dt = pytz.timezone('US/Central').localize(dt)
                dt = dt.astimezone(pytz.timezone(as_timezone))
                s_chat[idx] = {"date": dt.strftime(log_time_format), "text": msg}
            except:
                s_chat[idx] = {"date": "01/01/17 00:00 AM", "name": "Error", "text": "Parsing Line"}

        return {'global': g_chat, 'staff': s_chat}
    return None


def get_query():
    try:
        server = mcstatus.MinecraftServer.lookup(getattr(settings, 'SERVER_QUERY_IP', None))
        query = server.query()
        return {'max': str(query.players.max), 'online': str(query.players.online),
                'players': sorted(query.players.names)}
    except:
        return {'max': 0, 'online': 0,
                'players': []}