h0v1n8-web/website/rotbot/views.py

from django.shortcuts import render, get_object_or_404
from django.http import HttpResponseRedirect
from django.urls import reverse
#from django.views import generic
from django.contrib.auth.decorators import login_required, permission_required
from django.forms import modelformset_factory, inlineformset_factory

from website.settings import APPLICATION_NAME
from .models import Network, Host, Channel, User, Message, Action, Notice
from .forms import NetworkForm, HostForm

def default_keywords(additional_keywords=None):
    default_keywords = 'RotBot, robot, bot, irc, irc bot, irc robot, '
    if additional_keywords:
        additional_keywords = ', '.join(map(str, additional_keywords))
        return (default_keywords + additional_keywords)
    return (default_keywords)


def networks(request):
    networks = Network.objects.all()
    channels = Channel.objects.all()
    users = User.objects.all()
    messages = Message.objects.all()
    actions = Action.objects.all()
    notices = Notice.objects.all()

    total_messages = 0
    for message in messages:
        total_messages += message.amount
    for action in actions:
        total_messages += action.amount
    for notice in notices:
        total_messages += notice.amount

    context = {
        'title': 'RotBot',
        'icon': 'robot',
        'description': 'Index of RotBot, the IRC robot.',
        'keywords': default_keywords('index'),
        'networks': networks,
        'network_amount': networks.count(),
        'channel_amount': channels.count(),
        'user_amount': users.count(),
        'message_amount': total_messages,
    }
    return render(request, 'rotbot/networks.html', context)


def network(request, network_slug):
    network = get_object_or_404(Network, slug=network_slug)
    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)

    total_messages = 0
    for message in messages:
        print(message.amount)
        total_messages += message.amount
    for action in actions:
        total_messages += action.amount
    for notice in notices:
        total_messages += notice.amount

    context = {
        'parent_title': 'Networks',
        'parent_url': 'rotbot:networks',
        'parent_icon': 'robot',
        'title': network.name,
        'icon': 'sitemap',
        'description': 'Details of ' + network.name,
        'keywords': default_keywords() + 'network.name, display, details',
        'network': network,
        'channel_amount': channels.count(),
        'user_amount': users.count(),
        'message_amount': total_messages,
    }
    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)
    if request.method == 'POST':
        form = NetworkForm(request.POST, instance=network)
        formset = HostFormSet(request.POST, instance=network)
        if form.is_valid() and formset.is_valid():
            form.save()
            formset.save()
            return HttpResponseRedirect(reverse('rotbot:network', args=(network.slug,)))
    else:
        form = NetworkForm(instance=network)
        formset = HostFormSet(instance=network)
    context = {
        'parent_title': 'Networks',
        'parent_url': 'rotbot:networks',
        'parent_icon': 'robot',
        'title': title,
        'icon': 'sitemap',
        'description': 'Edit ' + title,
        'keywords': default_keywords((title, 'edit')),
        'network_slug': network_slug,
        'form': form,
        'hostformset': formset,
    }
    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': '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'))