allianceauth/allianceauth/optimer/views.py

import logging

from django.contrib import messages
from django.contrib.auth.decorators import login_required, permission_required
from django.shortcuts import get_object_or_404, redirect, render
from django.template.loader import render_to_string
from django.utils import timezone
from django.utils.translation import gettext_lazy as _

from .form import OpForm
from .models import OpTimer, OpTimerType

logger = logging.getLogger(__name__)

OPS_VIEW_PERMISSION = 'auth.optimer_view'
OPS_MANAGE_PERMISSION = 'auth.optimer_management'

@login_required
@permission_required(OPS_VIEW_PERMISSION)
def optimer_view(request):
    """
    View for the optimer management page

    :param request:
    :type request:
    :return:
    :rtype:
    """

    logger.debug(f"optimer_view called by user {request.user}")
    base_query = OpTimer.objects.select_related('eve_character', 'type')
    render_items = {'optimer': base_query.all(),
                    'future_timers': base_query.filter(
                        start__gte=timezone.now()),
                    'past_timers': base_query.filter(
                        start__lt=timezone.now()).order_by('-start')}

    return render(request, 'optimer/management.html', context=render_items)


@login_required
@permission_required(OPS_MANAGE_PERMISSION)
def add_optimer_view(request):
    """
    View for the add optimer page

    :param request:
    :type request:
    :return:
    :rtype:
    """

    logger.debug(f"add_optimer_view called by user {request.user}")

    if request.method == 'POST':
        form = OpForm(request.POST, data_list=OpTimerType.objects.all())
        logger.debug(f"Request type POST contains form valid: {form.is_valid()}")
        if form.is_valid():
            optimer_type = None

            if form.cleaned_data['type'] != '':
                try:
                    optimer_type = OpTimerType.objects.get(
                        type__iexact=form.cleaned_data['type']
                    )
                except OpTimerType.DoesNotExist:
                    optimer_type = OpTimerType.objects.create(
                        type=form.cleaned_data['type']
                    )

            # Get Current Time
            post_time = timezone.now()
            # Get character
            character = request.user.profile.main_character
            # handle valid form
            op = OpTimer()
            op.doctrine = form.cleaned_data['doctrine']
            op.system = form.cleaned_data['system']
            op.start = form.cleaned_data['start']
            op.duration = form.cleaned_data['duration']
            op.operation_name = form.cleaned_data['operation_name']
            op.fc = form.cleaned_data['fc']
            op.create_time = post_time
            op.eve_character = character
            op.type = optimer_type
            op.description = form.cleaned_data['description']
            op.save()
            logger.info(f"User {request.user} created op timer with name {op.operation_name}")
            messages.success(request, _('Created operation timer for %(opname)s.') % {"opname": op.operation_name})
            return redirect("optimer:view")
    else:
        logger.debug("Returning new opForm")
        form = OpForm(data_list=OpTimerType.objects.all())

    render_items = {'form': form}

    return render(request, 'optimer/add.html', context=render_items)


@login_required
@permission_required(OPS_MANAGE_PERMISSION)
def remove_optimer(request, optimer_id):
    """
    Remove optimer

    :param request:
    :type request:
    :param optimer_id:
    :type optimer_id:
    :return:
    :rtype:
    """

    logger.debug(f"remove_optimer called by user {request.user} for operation id {optimer_id}")
    op = get_object_or_404(OpTimer, id=optimer_id)
    op.delete()
    logger.info(f"Deleting optimer id {optimer_id} by user {request.user}")
    messages.success(request, _('Removed operation timer for %(opname)s.') % {"opname": op.operation_name})

    return redirect("optimer:view")


@login_required
@permission_required(OPS_MANAGE_PERMISSION)
def edit_optimer(request, optimer_id):
    """
    Edit optimer

    :param request:
    :type request:
    :param optimer_id:
    :type optimer_id:
    :return:
    :rtype:
    """

    logger.debug(f"edit_optimer called by user {request.user} for optimer id {optimer_id}")
    op = get_object_or_404(OpTimer, id=optimer_id)

    if request.method == 'POST':
        form = OpForm(request.POST, data_list=OpTimerType.objects.all())
        logger.debug(f"Received POST request containing update optimer form, is valid: {form.is_valid()}")
        if form.is_valid():
            character = request.user.profile.main_character

            optimer_type = None

            if form.cleaned_data['type'] != '':
                try:
                    optimer_type = OpTimerType.objects.get(
                        type__iexact=form.cleaned_data['type']
                    )
                except OpTimerType.DoesNotExist:
                    optimer_type = OpTimerType.objects.create(
                        type=form.cleaned_data['type']
                    )

            op.doctrine = form.cleaned_data['doctrine']
            op.system = form.cleaned_data['system']
            op.start = form.cleaned_data['start']
            op.duration = form.cleaned_data['duration']
            op.operation_name = form.cleaned_data['operation_name']
            op.fc = form.cleaned_data['fc']
            op.eve_character = character
            op.type = optimer_type
            op.description = form.cleaned_data['description']
            logger.info(f"User {request.user} updating optimer id {optimer_id} ")
            op.save()
            messages.success(request, _('Saved changes to operation timer for %(opname)s.') % {"opname": op.operation_name})
            return redirect("optimer:view")
    else:
        data = {
            'doctrine': op.doctrine,
            'system': op.system,
            'start': op.start,
            'duration': op.duration,
            'operation_name': op.operation_name,
            'fc': op.fc,
            'description': op.description,
            'type': op.type
        }
        form = OpForm(initial=data, data_list=OpTimerType.objects.all())
    return render(request, 'optimer/update.html', context={'form': form})


def dashboard_ops(request):
    """
    Returns the next five upcoming ops for the dashboard

    :param request:
    :type request:
    :return:
    :rtype:
    """
    if request.user.has_perm(OPS_VIEW_PERMISSION):
        base_query = OpTimer.objects.select_related('eve_character', 'type')
        timers = base_query.filter(
            start__gte=timezone.now()
        )[:5]

        if timers.count():
            context = {
                'timers': timers,
            }
            return render_to_string(
                'optimer/dashboard.ops.html',
                context=context,
                request=request
            )
        else:
            return ""
    else:
        return ""