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'))