h0v1n8-web/website/rotbot/views/network.py

from django.shortcuts import render, get_object_or_404
from django.http import HttpResponseRedirect
from django.urls import reverse
from django.forms import inlineformset_factory
from django.contrib.auth.decorators import login_required, permission_required
from website.settings import APPLICATION_NAME
from rotbot.models import Network, Host, Channel, User, Message, Action, Notice, Join, Kick, Owner
from rotbot.forms import NetworkForm, HostForm, OwnerForm
from .common import default_keywords, shorten_number, total_messages

def networks(request):
    networks = Network.objects.all()
    hosts = Host.objects.all()
    channels = Channel.objects.all()
    users = User.objects.all()
    messages = Message.objects.all()
    actions = Action.objects.all()
    notices = Notice.objects.all()
    joins = Join.objects.all()
    kicks = Kick.objects.all()
    total_level, total_xp_spent, total_ap_spent, total_coin, total_coin_spent, total_coin_given = user_totals(users)

    context = {
        'title': 'RotBot',
        'icon': 'robot',
        'description': 'Index of RotBot, the IRC robot.',
        'keywords': default_keywords('index'),
        'networks': networks,
        'total_networks': shorten_number(networks.count()),
        'total_hosts': shorten_number(hosts.count()),
        'total_chanels': shorten_number(channels.count()),
        'total_users': shorten_number(users.count()),
        'total_messages': shorten_number(total_messages(messages)),
        'total_actions': shorten_number(total_messages(actions)),
        'total_notices': shorten_number(total_messages(notices)),
        'total_joins': shorten_number(joins.count()),
        'total_kicks': shorten_number(kicks.count()),
        'total_level': total_level,
        'total_xp_spent': total_xp_spent,
        'total_ap_spent': total_ap_spent,
        'total_coin': total_coin,
        'total_coin_spent': total_coin_spent,
        'total_coin_given': total_coin_given,
    }
    return render(request, 'rotbot/networks.html', context)

def user_totals(users):
    total_level = 0
    total_xp_spent = 0
    total_ap_spent = 0
    total_coin = 0
    total_coin_spent = 0
    total_coin_given = 0
    for user in users:
        total_level += user.level
        total_xp_spent += user.xp_spent
        total_ap_spent += user.ap_spent
        total_coin += user.coin
        total_coin_spent += user.coin_spent
        total_coin_given += user.coin_given
    return total_level, total_xp_spent, total_ap_spent, total_coin, total_coin_spent, total_coin_given

def network(request, network_slug):
    network = get_object_or_404(Network, slug=network_slug)
    hosts = Host.objects.filter(network=network)
    channels = Channel.objects.filter(network=network)
    users = User.objects.filter(network=network)
    messages = Message.objects.filter(network=network)
    actions = Action.objects.filter(network=network)
    notices = Notice.objects.filter(network=network)
    joins = Join.objects.all()
    kicks = Kick.objects.all()

    for host in hosts:
        host.connection_fails = host.connection_attempts - host.connection_succeeds
    total_level, total_xp_spent, total_ap_spent, total_coin, total_coin_spent, total_coin_given = user_totals(users)

    context = {
        'parent_title': 'Networks',
        'parent_url': reverse('rotbot:networks'),
        'parent_icon': 'robot',
        'title': network.name,
        'icon': 'sitemap',
        'description': 'Details of ' + network.name,
        'keywords': default_keywords() + 'network.name, display, details',
        'network': network,
        'hosts': hosts,
        'channels': channels,
        'total_hosts': shorten_number(hosts.count()),
        'total_chanels': shorten_number(channels.count()),
        'total_users': shorten_number(users.count()),
        'total_messages': shorten_number(total_messages(messages)),
        'total_actions': shorten_number(total_messages(actions)),
        'total_notices': shorten_number(total_messages(notices)),
        'total_joins': shorten_number(joins.count()),
        'total_kicks': shorten_number(kicks.count()),
        'total_level': total_level,
        'total_xp_spent': total_xp_spent,
        'total_ap_spent': total_ap_spent,
        'total_coin': total_coin,
        'total_coin_spent': total_coin_spent,
        'total_coin_given': total_coin_given,
    }
    return render(request, 'rotbot/network.html', context)


@login_required
@permission_required('rotbot.change_network', raise_exception=True)
def edit_network(request, network_slug):
    network = get_object_or_404(Network, slug=network_slug)
    title = network.name
    HostFormSet = inlineformset_factory(Network, Host, form=HostForm)
    OwnerFormSet = inlineformset_factory(Network, Owner, form=OwnerForm)
    if request.method == 'POST':
        form = NetworkForm(request.POST, instance=network)
        hostformset = HostFormSet(request.POST, instance=network)
        ownerformset = HostFormSet(request.POST, instance=network)
        if form.is_valid() and hostformset.is_valid() and ownerformset.is_valid():
            form.save()
            hostformset.save()
            ownerformset.save()
            return HttpResponseRedirect(reverse('rotbot:network', args=(network.slug,)))
    else:
        form = NetworkForm(instance=network)
        hostformset = HostFormSet(instance=network)
        ownerformset = OwnerFormSet(instance=network)
    context = {
        'parent_title': 'Networks',
        'parent_url': reverse('rotbot:networks'),
        'parent_icon': 'robot',
        'title': title,
        'icon': 'sitemap',
        'description': 'Edit ' + title,
        'keywords': default_keywords((title, 'edit')),
        'network_slug': network_slug,
        'form': form,
        'hostformset': hostformset,
        'ownerformset': ownerformset,
    }
    return render(request, 'rotbot/edit_network.html', context)

@login_required
@permission_required('rotbot.add_network', raise_exception=True)
def add_network(request):
    HostFormSet = inlineformset_factory(Network, Host, form=HostForm)
    if request.method == 'POST':
        form = NetworkForm(request.POST)
        if form.is_valid():
            network = form.save()
            formset = HostFormSet(request.POST, instance=network)
            if formset.is_valid():
                formset.save()
                return HttpResponseRedirect(reverse('rotbot:network', args=(form.cleaned_data['slug'],)))
        else:
            formset = HostFormSet(queryset=Host.objects.none())
    else:   # Not a POST request.
        form = NetworkForm()
        formset = HostFormSet(queryset=Host.objects.none())
    context = {
        'parent_title': 'Networks',
        'parent_url': reverse('rotbot:networks'),
        'parent_icon': 'robot',
        'title': 'Add network',
        'icon': 'sitemap',
        'description': 'Add new irc network to ' + APPLICATION_NAME,
        'keywords': default_keywords() + 'add, add network',
        'form': form,
        'hostformset': formset,
    }
    return render(request, 'rotbot/add_network.html', context)


@login_required
@permission_required('rotbot.change_network', raise_exception=True)
def delete_network(request, network_slug):
    network = get_object_or_404(Network, slug=network_slug)
    network.delete()
    return HttpResponseRedirect(reverse('rotbot:networks'))