mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2025-07-09 12:30:15 +02:00
Update timerboard views
Adds basic unit tests Changed to class based views
This commit is contained in:
parent
d993a299ef
commit
fff2cd32d5
@ -1,9 +1,37 @@
|
||||
import logging
|
||||
import datetime
|
||||
from django import forms
|
||||
from django.utils import timezone
|
||||
from django.core.validators import MaxValueValidator, MinValueValidator
|
||||
from django.utils.translation import ugettext_lazy as _
|
||||
|
||||
from .models import Timer
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class TimerForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = Timer
|
||||
fields = ('details', 'system', 'planet_moon', 'structure', 'objective', 'important', 'corp_timer')
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
self.user = kwargs.pop('user', None)
|
||||
if 'instance' in kwargs and kwargs['instance'] is not None:
|
||||
# Do conversion from db datetime to days/hours/minutes
|
||||
# for appropriate fields
|
||||
current_time = timezone.now()
|
||||
td = kwargs['instance'].eve_time - current_time
|
||||
initial = kwargs.pop('initial', dict())
|
||||
if 'days_left' not in initial:
|
||||
initial.update({'days_left': td.days})
|
||||
if 'hours_left' not in initial:
|
||||
initial.update({'hours_left': td.seconds // 3600})
|
||||
if 'minutes_left' not in initial:
|
||||
initial.update({'minutes_left': td.seconds // 60 % 60})
|
||||
kwargs.update({'initial': initial})
|
||||
super(TimerForm, self).__init__(*args, **kwargs)
|
||||
|
||||
class TimerForm(forms.Form):
|
||||
structure_choices = [('POCO', 'POCO'), ('I-HUB', 'I-HUB'), ('POS[S]', 'POS[S]'),
|
||||
('POS[M]', 'POS[M]'), ('POS[L]', 'POS[L]'), ('Citadel[M]', 'Citadel[M]'),
|
||||
('Citadel[L]', 'Citadel[L]'), ('Citadel[XL]', 'Citadel[XL]'),
|
||||
@ -25,3 +53,27 @@ class TimerForm(forms.Form):
|
||||
validators=[MinValueValidator(0), MaxValueValidator(59)])
|
||||
important = forms.BooleanField(label=_("Important"), required=False)
|
||||
corp_timer = forms.BooleanField(label=_("Corp-Restricted"), required=False)
|
||||
|
||||
def save(self, commit=True):
|
||||
timer = super(TimerForm, self).save(commit=False)
|
||||
|
||||
# Get character
|
||||
character = self.user.profile.main_character
|
||||
corporation = character.corporation
|
||||
logger.debug("Determined timer save request on behalf "
|
||||
"of character {} corporation {}".format(character, corporation))
|
||||
# calculate future time
|
||||
future_time = datetime.timedelta(days=self.cleaned_data['days_left'], hours=self.cleaned_data['hours_left'],
|
||||
minutes=self.cleaned_data['minutes_left'])
|
||||
current_time = timezone.now()
|
||||
eve_time = current_time + future_time
|
||||
logger.debug(
|
||||
"Determined timer eve time is %s - current time %s, adding %s" % (eve_time, current_time, future_time))
|
||||
|
||||
timer.eve_time = eve_time
|
||||
timer.eve_character = character
|
||||
timer.eve_corp = corporation
|
||||
timer.user = self.user
|
||||
if commit:
|
||||
timer.save()
|
||||
return timer
|
||||
|
@ -22,4 +22,4 @@ class Timer(models.Model):
|
||||
user = models.ForeignKey(User)
|
||||
|
||||
def __str__(self):
|
||||
return str(self.system) + ' ' + str(self.objective)
|
||||
return str(self.system) + ' ' + str(self.details)
|
||||
|
@ -1,29 +0,0 @@
|
||||
{% extends "allianceauth/base.html" %}
|
||||
{% load bootstrap %}
|
||||
{% load staticfiles %}
|
||||
{% load i18n %}
|
||||
{% get_current_language as LANGUAGE_CODE %}
|
||||
|
||||
{% block title %}{% trans "Alliance Auth - Structure Timer Create" %}{% endblock %}
|
||||
|
||||
{% block page_title %}{% trans "Timer Create" %}{% endblock page_title %}
|
||||
|
||||
{% block content %}
|
||||
<div class="col-lg-12">
|
||||
<h1 class="page-header text-center">{% trans "Create Structure Timer" %}</h1>
|
||||
|
||||
<div class="container-fluid">
|
||||
<div class="col-md-4 col-md-offset-4">
|
||||
<div class="row">
|
||||
<form class="form-signin" role="form" action="" method="POST">
|
||||
{% csrf_token %}
|
||||
{{ form|bootstrap }}
|
||||
<br/>
|
||||
<button class="btn btn-lg btn-primary btn-block" type="submit">{% trans "Create Timer" %}</button>
|
||||
</form>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{% endblock content %}
|
33
allianceauth/timerboard/templates/timerboard/form.html
Normal file
33
allianceauth/timerboard/templates/timerboard/form.html
Normal file
@ -0,0 +1,33 @@
|
||||
{% extends "allianceauth/base.html" %}
|
||||
{% load bootstrap %}
|
||||
{% load staticfiles %}
|
||||
{% load i18n %}
|
||||
|
||||
{% block page_title %}
|
||||
{% endblock page_title %}
|
||||
|
||||
{% block content %}
|
||||
<div class="col-lg-12">
|
||||
<h1 class="page-header text-center">
|
||||
{% block page_header %}
|
||||
{% endblock %}
|
||||
{% include 'timerboard/index_button.html' %}
|
||||
</h1>
|
||||
<div class="container-fluid">
|
||||
<div class="col-md-4 col-md-offset-4">
|
||||
<div class="row">
|
||||
<form id="add-timer-form" class="form-signin" role="form" action="" method="POST">
|
||||
{% csrf_token %}
|
||||
{{ form|bootstrap }}
|
||||
<br/>
|
||||
<button class="btn btn-lg btn-primary btn-block" type="submit">
|
||||
{% block submit_button_text %}
|
||||
{% endblock %}
|
||||
</button>
|
||||
</form>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{% endblock content %}
|
@ -0,0 +1,4 @@
|
||||
{% load i18n %}
|
||||
<div class="text-right">
|
||||
<a href="{% url 'timerboard:view' %}" class="btn btn-default">Back</a>
|
||||
</div>
|
@ -0,0 +1,26 @@
|
||||
{% extends "allianceauth/base.html" %}
|
||||
{% load i18n %}
|
||||
|
||||
{% block page_title %}
|
||||
Delete Timer
|
||||
{% endblock page_title %}
|
||||
|
||||
{% block content %}
|
||||
<div class="col-lg-12">
|
||||
<h1 class="page-header text-center">
|
||||
{% trans "Delete Timer" %}
|
||||
{% include 'timerboard/index_button.html' %}
|
||||
</h1>
|
||||
<div class="container-fluid">
|
||||
<div class="col-md-2 col-md-offset-5">
|
||||
<div class="row">
|
||||
<form action="" method="post">
|
||||
{% csrf_token %}
|
||||
<p>{% blocktrans %}Are you sure you want to delete timer "{{ object }}"?{% endblocktrans %}</p>
|
||||
<input class="btn btn-danger btn-block" type="submit" value="Confirm" />
|
||||
</form>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
{% endblock content %}
|
@ -0,0 +1,14 @@
|
||||
{% extends "timerboard/form.html" %}
|
||||
{% load i18n %}
|
||||
|
||||
{% block page_title %}
|
||||
{% trans "Create Timer" %}
|
||||
{% endblock page_title %}
|
||||
|
||||
{% block page_header %}
|
||||
{% trans "Create Structure Timer" %}
|
||||
{% endblock %}
|
||||
|
||||
{% block submit_button_text %}
|
||||
{% trans "Create Timer" %}
|
||||
{% endblock %}
|
@ -0,0 +1,14 @@
|
||||
{% extends "timerboard/form.html" %}
|
||||
{% load i18n %}
|
||||
|
||||
{% block page_title %}
|
||||
{% trans "Update Structure Timer" %}
|
||||
{% endblock page_title %}
|
||||
|
||||
{% block page_header %}
|
||||
{% trans "Update Structure Timer" %}
|
||||
{% endblock %}
|
||||
|
||||
{% block submit_button_text %}
|
||||
{% trans "Update Structure Timer" %}
|
||||
{% endblock %}
|
@ -1,35 +0,0 @@
|
||||
{% extends "allianceauth/base.html" %}
|
||||
{% load bootstrap %}
|
||||
{% load staticfiles %}
|
||||
{% load i18n %}
|
||||
|
||||
{% block title %}Alliance Auth - Update Structure Timer {% endblock %}
|
||||
|
||||
{% block page_title %}{% trans "Update AAR Link" %}{% endblock page_title %}
|
||||
|
||||
{% block content %}
|
||||
|
||||
<div class="col-lg-12">
|
||||
<h1 class="page-header text-center">{% trans "Update Structure Timer" %}</h1>
|
||||
|
||||
<div class="container-fluid">
|
||||
<div class="col-md-4 col-md-offset-4">
|
||||
<div class="row">
|
||||
{% if no_fleet_id %}
|
||||
<div class="alert alert-danger" role="alert">{% trans "Structure Timer Does Not Exist" %}</div>
|
||||
{% else %}
|
||||
<form class="form-signin" role="form" action="" method="POST">
|
||||
{% csrf_token %}
|
||||
{{ form|bootstrap }}
|
||||
<br/>
|
||||
<button class="btn btn-lg btn-primary btn-block" type="submit">{% trans "Update Structure Timer" %}
|
||||
</button>
|
||||
</form>
|
||||
{% endif %}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
</div>
|
||||
|
||||
{% endblock content %}
|
@ -3,8 +3,6 @@
|
||||
{% load i18n %}
|
||||
{% get_current_language as LANGUAGE_CODE %}
|
||||
|
||||
{% block title %}Alliance Auth{% endblock %}
|
||||
|
||||
{% block page_title %}{% trans "Structure Timer Management" %}{% endblock page_title %}
|
||||
{% block extra_css %}{% endblock extra_css %}
|
||||
|
||||
@ -145,7 +143,7 @@
|
||||
<td class="text-center">{{ timer.eve_character.character_name }}</td>
|
||||
{% if perms.auth.timer_management %}
|
||||
<td class="text-center">
|
||||
<a href="{% url 'timerboard:remove' timer.id %}" class="btn btn-danger">
|
||||
<a href="{% url 'timerboard:delete' timer.id %}" class="btn btn-danger">
|
||||
<span class="glyphicon glyphicon-remove"></span>
|
||||
</a>
|
||||
<a href="{% url 'timerboard:edit' timer.id %}" class="btn btn-info">
|
||||
@ -279,7 +277,7 @@
|
||||
<td class="text-center">{{ timer.eve_character.character_name }}</td>
|
||||
{% if perms.auth.timer_management %}
|
||||
<td class="text-center">
|
||||
<a href="{% url 'timerboard:remove' timer.id %}" class="btn btn-danger">
|
||||
<a href="{% url 'timerboard:delete' timer.id %}" class="btn btn-danger">
|
||||
<span class="glyphicon glyphicon-remove"></span>
|
||||
</a>
|
||||
<a href="{% url 'timerboard:edit' timer.id %}" class="btn btn-info">
|
||||
@ -415,7 +413,7 @@
|
||||
<td class="text-center">{{ timer.eve_character.character_name }}</td>
|
||||
{% if perms.auth.timer_management %}
|
||||
<td class="text-center">
|
||||
<a href="{% url 'timerboard:remove' timer.id %}" class="btn btn-danger">
|
||||
<a href="{% url 'timerboard:delete' timer.id %}" class="btn btn-danger">
|
||||
<span class="glyphicon glyphicon-remove"></span>
|
||||
</a>
|
||||
<a href="{% url 'timerboard:edit' timer.id %}" class="btn btn-info">
|
232
allianceauth/timerboard/tests.py
Executable file → Normal file
232
allianceauth/timerboard/tests.py
Executable file → Normal file
@ -1 +1,231 @@
|
||||
# Create your tests here.
|
||||
from django_webtest import WebTest
|
||||
from django.utils import timezone
|
||||
from django.core.urlresolvers import reverse
|
||||
from django.contrib.auth.models import Permission, User
|
||||
from django.conf import settings
|
||||
|
||||
from datetime import timedelta
|
||||
|
||||
from allianceauth.tests.auth_utils import AuthUtils
|
||||
from allianceauth.eveonline.models import EveCorporationInfo
|
||||
|
||||
from .models import Timer
|
||||
from .form import TimerForm
|
||||
|
||||
|
||||
class TimerboardViewsTestCase(WebTest):
|
||||
csrf_checks = False
|
||||
|
||||
def setUp(self):
|
||||
corp = EveCorporationInfo.objects.create(corporation_id='2345', corporation_name='test corp',
|
||||
corporation_ticker='testc', member_count=24)
|
||||
other_corp = EveCorporationInfo.objects.create(corporation_id='9345', corporation_name='other test corp',
|
||||
corporation_ticker='testd', member_count=1)
|
||||
self.user = AuthUtils.create_user('test_user')
|
||||
AuthUtils.add_main_character(self.user, 'test character', '1234', '2345', 'test corp', 'testc')
|
||||
self.user = User.objects.get_by_natural_key('test_user')
|
||||
character = self.user.profile.main_character
|
||||
self.other_user = AuthUtils.create_user('other_test_user')
|
||||
AuthUtils.add_main_character(self.other_user, 'test character 2', '9234', '9345', 'other test corp', 'testd')
|
||||
self.other_user = User.objects.get_by_natural_key('other_test_user')
|
||||
other_character = self.other_user.profile.main_character
|
||||
|
||||
self.timer = Timer.objects.create(
|
||||
details='details',
|
||||
system='system',
|
||||
planet_moon='planet_moon',
|
||||
structure='structure',
|
||||
objective='objective',
|
||||
eve_time=timezone.now() + timedelta(days=30),
|
||||
important=True,
|
||||
corp_timer=False,
|
||||
eve_character=character,
|
||||
eve_corp=character.corporation,
|
||||
user=self.user,
|
||||
)
|
||||
self.corp_timer = Timer.objects.create(
|
||||
details='details',
|
||||
system='system',
|
||||
planet_moon='planet_moon',
|
||||
structure='structure',
|
||||
objective='objective',
|
||||
eve_time=timezone.now() + timedelta(days=30),
|
||||
important=False,
|
||||
corp_timer=True,
|
||||
eve_character=character,
|
||||
eve_corp=character.corporation,
|
||||
user=self.user,
|
||||
)
|
||||
self.other_corp_timer = Timer.objects.create(
|
||||
details='details',
|
||||
system='system',
|
||||
planet_moon='planet_moon',
|
||||
structure='structure',
|
||||
objective='objective',
|
||||
eve_time=timezone.now() + timedelta(days=30),
|
||||
important=False,
|
||||
corp_timer=True,
|
||||
eve_character=other_character,
|
||||
eve_corp=other_character.corporation,
|
||||
user=self.user,
|
||||
)
|
||||
self.expired_timer = Timer.objects.create(
|
||||
details='details',
|
||||
system='system',
|
||||
planet_moon='planet_moon',
|
||||
structure='structure',
|
||||
objective='objective',
|
||||
eve_time=timezone.now() - timedelta(days=30),
|
||||
important=True,
|
||||
corp_timer=False,
|
||||
eve_character=character,
|
||||
eve_corp=character.corporation,
|
||||
user=self.user,
|
||||
)
|
||||
|
||||
self.view_permission = Permission.objects.get(codename='timer_view')
|
||||
self.edit_permission = Permission.objects.get(codename='timer_management')
|
||||
|
||||
self.view_url = reverse('timerboard:view')
|
||||
self.add_url = reverse('timerboard:add')
|
||||
self.edit_url_name = 'timerboard:edit'
|
||||
self.delete_url_name = 'timerboard:delete'
|
||||
|
||||
def test_timer_view(self):
|
||||
self.user.user_permissions.add(self.view_permission)
|
||||
|
||||
self.app.set_user(self.user)
|
||||
|
||||
response = self.app.get(self.view_url)
|
||||
|
||||
context = response.context[-1]
|
||||
|
||||
timers = context['timers']
|
||||
corp_timers = context['corp_timers']
|
||||
future_timers = context['future_timers']
|
||||
past_timers = context['past_timers']
|
||||
|
||||
self.assertTemplateUsed(response, 'timerboard/view.html')
|
||||
|
||||
self.assertIn(self.timer, timers)
|
||||
self.assertIn(self.expired_timer, timers)
|
||||
self.assertNotIn(self.corp_timer, timers)
|
||||
self.assertNotIn(self.other_corp_timer, timers)
|
||||
|
||||
self.assertNotIn(self.timer, corp_timers)
|
||||
self.assertNotIn(self.expired_timer, corp_timers)
|
||||
self.assertIn(self.corp_timer, corp_timers)
|
||||
self.assertNotIn(self.other_corp_timer, corp_timers)
|
||||
|
||||
self.assertIn(self.timer, future_timers)
|
||||
self.assertNotIn(self.expired_timer, future_timers)
|
||||
self.assertNotIn(self.corp_timer, future_timers)
|
||||
self.assertNotIn(self.other_corp_timer, future_timers)
|
||||
|
||||
self.assertNotIn(self.timer, past_timers)
|
||||
self.assertIn(self.expired_timer, past_timers)
|
||||
self.assertNotIn(self.corp_timer, past_timers)
|
||||
self.assertNotIn(self.other_corp_timer, past_timers)
|
||||
|
||||
def test_timer_view_permission(self):
|
||||
self.client.force_login(self.user)
|
||||
response = self.app.get(self.view_url)
|
||||
self.assertRedirects(response, expected_url=reverse(settings.LOGIN_URL) + '?next=' + self.view_url)
|
||||
|
||||
def test_timer_view_login(self):
|
||||
response = self.app.get(self.view_url)
|
||||
self.assertRedirects(response, expected_url=reverse(settings.LOGIN_URL) + '?next=' + self.view_url)
|
||||
|
||||
def test_add_timer_get(self):
|
||||
self.user.user_permissions.add(self.edit_permission)
|
||||
self.app.set_user(self.user)
|
||||
|
||||
response = self.app.get(self.add_url)
|
||||
|
||||
self.assertTemplateUsed(response, 'timerboard/timer_create_form.html')
|
||||
|
||||
context = response.context[-1]
|
||||
|
||||
self.assertIs(TimerForm, type(context['form']))
|
||||
|
||||
def test_add_timer_form_error(self):
|
||||
self.user.user_permissions.add(self.edit_permission)
|
||||
self.app.set_user(self.user)
|
||||
page = self.app.post(self.add_url)
|
||||
page = page.forms['add-timer-form'].submit()
|
||||
self.assertContains(page, "This field is required.")
|
||||
|
||||
def test_add_timer_post(self):
|
||||
self.user.user_permissions.add(self.edit_permission)
|
||||
self.user.user_permissions.add(self.view_permission)
|
||||
|
||||
self.app.set_user(self.user)
|
||||
|
||||
page = self.app.post(self.add_url)
|
||||
form = page.forms['add-timer-form']
|
||||
|
||||
form['details'] = 'details'
|
||||
form['system'] = 'jita'
|
||||
form['planet_moon'] = '4-4'
|
||||
form['structure'] = TimerForm.structure_choices[0][0]
|
||||
form['objective'] = TimerForm.objective_choices[0][0]
|
||||
form['days_left'] = 1
|
||||
form['hours_left'] = 2
|
||||
form['minutes_left'] = 3
|
||||
form['important'] = True
|
||||
form['corp_timer'] = False
|
||||
|
||||
response = form.submit()
|
||||
|
||||
self.assertRedirects(response, self.view_url)
|
||||
|
||||
self.assertTrue(Timer.objects.filter(system='jita', details='details').exists())
|
||||
|
||||
def test_edit_timer_get(self):
|
||||
self.user.user_permissions.add(self.edit_permission)
|
||||
self.app.set_user(self.user)
|
||||
|
||||
response = self.app.get(reverse(self.edit_url_name, args=[self.timer.id]))
|
||||
context = response.context[-1]
|
||||
form = context['form']
|
||||
data = form.instance
|
||||
|
||||
self.assertTemplateUsed(response, 'timerboard/timer_update_form.html')
|
||||
self.assertIs(TimerForm, type(form))
|
||||
self.assertEqual(data, self.timer)
|
||||
|
||||
def test_edit_timer_post(self):
|
||||
self.user.user_permissions.add(self.edit_permission)
|
||||
self.user.user_permissions.add(self.view_permission)
|
||||
|
||||
self.app.set_user(self.user)
|
||||
|
||||
page = self.app.post(reverse(self.edit_url_name, args=[self.timer.id]))
|
||||
form = page.forms['add-timer-form']
|
||||
|
||||
form['details'] = 'detailsUNIQUE'
|
||||
form['system'] = 'jita'
|
||||
form['planet_moon'] = '4-4'
|
||||
form['structure'] = TimerForm.structure_choices[0][0]
|
||||
form['objective'] = TimerForm.objective_choices[0][0]
|
||||
form['days_left'] = 1
|
||||
form['hours_left'] = 2
|
||||
form['minutes_left'] = 3
|
||||
form['important'] = True
|
||||
form['corp_timer'] = False
|
||||
|
||||
response = form.submit()
|
||||
|
||||
self.assertRedirects(response, self.view_url)
|
||||
|
||||
self.assertTrue(Timer.objects.filter(system='jita', details='detailsUNIQUE').exists())
|
||||
|
||||
def test_delete_timer_get(self):
|
||||
self.user.user_permissions.add(self.edit_permission)
|
||||
self.app.set_user(self.user)
|
||||
|
||||
response = self.app.get(reverse(self.delete_url_name, args=[self.timer.id]))
|
||||
|
||||
self.assertTemplateUsed(response, 'timerboard/timer_confirm_delete.html')
|
||||
|
||||
self.assertContains(response, 'Are you sure you want to delete timer "'+str(self.timer))
|
||||
|
@ -5,8 +5,8 @@ from . import views
|
||||
app_name = 'timerboard'
|
||||
|
||||
urlpatterns = [
|
||||
url(r'^$', views.timer_view, name='view'),
|
||||
url(r'^add/$', views.add_timer_view, name='add'),
|
||||
url(r'^remove/(\w+)$', views.remove_timer, name='remove'),
|
||||
url(r'^edit/(\w+)$', views.edit_timer, name='edit'),
|
||||
url(r'^$', views.TimerView.as_view(), name='view'),
|
||||
url(r'^add/$', views.AddTimerView.as_view(), name='add'),
|
||||
url(r'^remove/(?P<pk>\w+)$', views.RemoveTimerView.as_view(), name='delete'),
|
||||
url(r'^edit/(?P<pk>\w+)$', views.EditTimerView.as_view(), name='edit'),
|
||||
]
|
||||
|
@ -2,13 +2,14 @@ import datetime
|
||||
import logging
|
||||
|
||||
from django.contrib import messages
|
||||
from django.contrib.auth.decorators import login_required
|
||||
from django.contrib.auth.decorators import permission_required
|
||||
from django.shortcuts import get_object_or_404
|
||||
from django.shortcuts import render, redirect
|
||||
from django.views import View
|
||||
from django.urls import reverse_lazy
|
||||
from django.views.generic import CreateView, UpdateView, DeleteView
|
||||
from django.contrib.auth.mixins import LoginRequiredMixin, PermissionRequiredMixin
|
||||
from django.utils import timezone
|
||||
from django.utils.translation import ugettext_lazy as _
|
||||
from allianceauth.eveonline.models import EveCorporationInfo
|
||||
|
||||
from .form import TimerForm
|
||||
from .models import Timer
|
||||
@ -16,136 +17,72 @@ from .models import Timer
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
@login_required
|
||||
@permission_required('auth.timer_view')
|
||||
def timer_view(request):
|
||||
logger.debug("timer_view called by user %s" % request.user)
|
||||
class BaseTimerView(LoginRequiredMixin, PermissionRequiredMixin, View):
|
||||
pass
|
||||
|
||||
|
||||
class TimerView(BaseTimerView):
|
||||
template_name = 'timerboard/view.html'
|
||||
permission_required = 'auth.timer_view'
|
||||
|
||||
def get(self, request):
|
||||
logger.debug("timer_view called by user {}".format(request.user))
|
||||
char = request.user.profile.main_character
|
||||
if char:
|
||||
corp = EveCorporationInfo.objects.get(corporation_id=char.corporation_id)
|
||||
corp = char.corporation
|
||||
else:
|
||||
corp = None
|
||||
if corp:
|
||||
corp_timers = Timer.objects.all().filter(corp_timer=True).filter(eve_corp=corp)
|
||||
else:
|
||||
corp_timers = []
|
||||
render_items = {'timers': Timer.objects.all().filter(corp_timer=False),
|
||||
'corp_timers': corp_timers,
|
||||
'future_timers': Timer.objects.all().filter(corp_timer=False).filter(
|
||||
eve_time__gte=datetime.datetime.now()),
|
||||
'past_timers': Timer.objects.all().filter(corp_timer=False).filter(
|
||||
eve_time__lt=datetime.datetime.now())}
|
||||
|
||||
return render(request, 'timerboard/management.html', context=render_items)
|
||||
|
||||
|
||||
@login_required
|
||||
@permission_required('auth.timer_management')
|
||||
def add_timer_view(request):
|
||||
logger.debug("add_timer_view called by user %s" % request.user)
|
||||
if request.method == 'POST':
|
||||
form = TimerForm(request.POST)
|
||||
logger.debug("Request type POST contains form valid: %s" % form.is_valid())
|
||||
if form.is_valid():
|
||||
# Get character
|
||||
character = request.user.profile.main_character
|
||||
corporation = EveCorporationInfo.get_corporation_info_by_id(character.corporation_id)
|
||||
logger.debug(
|
||||
"Determined timer add request on behalf of character %s corporation %s" % (character, corporation))
|
||||
# calculate future time
|
||||
future_time = datetime.timedelta(days=form.cleaned_data['days_left'], hours=form.cleaned_data['hours_left'],
|
||||
minutes=form.cleaned_data['minutes_left'])
|
||||
current_time = timezone.now()
|
||||
eve_time = current_time + future_time
|
||||
logger.debug(
|
||||
"Determined timer eve time is %s - current time %s, adding %s" % (eve_time, current_time, future_time))
|
||||
# handle valid form
|
||||
timer = Timer()
|
||||
timer.details = form.cleaned_data['details']
|
||||
timer.system = form.cleaned_data['system']
|
||||
timer.planet_moon = form.cleaned_data['planet_moon']
|
||||
timer.structure = form.cleaned_data['structure']
|
||||
timer.objective = form.cleaned_data['objective']
|
||||
timer.eve_time = eve_time
|
||||
timer.important = form.cleaned_data['important']
|
||||
timer.corp_timer = form.cleaned_data['corp_timer']
|
||||
timer.eve_character = character
|
||||
timer.eve_corp = corporation
|
||||
timer.user = request.user
|
||||
timer.save()
|
||||
logger.info("Created new timer in %s at %s by user %s" % (timer.system, timer.eve_time, request.user))
|
||||
messages.success(request, _('Added new timer in %(system)s at %(time)s.') % {"system": timer.system, "time": timer.eve_time})
|
||||
return redirect("timerboard:view")
|
||||
else:
|
||||
logger.debug("Returning new TimerForm")
|
||||
form = TimerForm()
|
||||
|
||||
render_items = {'form': form}
|
||||
|
||||
return render(request, 'timerboard/add.html', context=render_items)
|
||||
|
||||
|
||||
@login_required
|
||||
@permission_required('auth.timer_management')
|
||||
def remove_timer(request, timer_id):
|
||||
logger.debug("remove_timer called by user %s for timer id %s" % (request.user, timer_id))
|
||||
timer = get_object_or_404(Timer, id=timer_id)
|
||||
timer.delete()
|
||||
logger.debug("Deleting timer id %s by user %s" % (timer_id, request.user))
|
||||
messages.success(request, _('Deleted timer in %(system)s at %(time)s.') % {'system': timer.system,
|
||||
'time': timer.eve_time})
|
||||
return redirect("timerboard:view")
|
||||
|
||||
|
||||
@login_required
|
||||
@permission_required('auth.timer_management')
|
||||
def edit_timer(request, timer_id):
|
||||
logger.debug("edit_timer called by user %s for timer id %s" % (request.user, timer_id))
|
||||
timer = get_object_or_404(Timer, id=timer_id)
|
||||
if request.method == 'POST':
|
||||
form = TimerForm(request.POST)
|
||||
logger.debug("Received POST request containing updated timer form, is valid: %s" % form.is_valid())
|
||||
if form.is_valid():
|
||||
character = request.user.profile.main_character
|
||||
corporation = EveCorporationInfo.get_corporation_info_by_id(character.corporation_id)
|
||||
logger.debug(
|
||||
"Determined timer edit request on behalf of character %s corporation %s" % (character, corporation))
|
||||
# calculate future time
|
||||
future_time = datetime.timedelta(days=form.cleaned_data['days_left'], hours=form.cleaned_data['hours_left'],
|
||||
minutes=form.cleaned_data['minutes_left'])
|
||||
current_time = datetime.datetime.utcnow()
|
||||
eve_time = current_time + future_time
|
||||
logger.debug(
|
||||
"Determined timer eve time is %s - current time %s, adding %s" % (eve_time, current_time, future_time))
|
||||
timer.details = form.cleaned_data['details']
|
||||
timer.system = form.cleaned_data['system']
|
||||
timer.planet_moon = form.cleaned_data['planet_moon']
|
||||
timer.structure = form.cleaned_data['structure']
|
||||
timer.objective = form.cleaned_data['objective']
|
||||
timer.eve_time = eve_time
|
||||
timer.important = form.cleaned_data['important']
|
||||
timer.corp_timer = form.cleaned_data['corp_timer']
|
||||
timer.eve_character = character
|
||||
timer.eve_corp = corporation
|
||||
logger.info("User %s updating timer id %s " % (request.user, timer_id))
|
||||
messages.success(request, _('Saved changes to the timer.'))
|
||||
timer.save()
|
||||
return redirect("timerboard:view")
|
||||
else:
|
||||
current_time = timezone.now()
|
||||
td = timer.eve_time - current_time
|
||||
tddays, tdhours, tdminutes = td.days, td.seconds // 3600, td.seconds // 60 % 60
|
||||
data = {
|
||||
'details': timer.details,
|
||||
'system': timer.system,
|
||||
'planet_moon': timer.planet_moon,
|
||||
'structure': timer.structure,
|
||||
'objective': timer.objective,
|
||||
'important': timer.important,
|
||||
'corp_timer': timer.corp_timer,
|
||||
'days_left': tddays,
|
||||
'hours_left': tdhours,
|
||||
'minutes_left': tdminutes,
|
||||
render_items = {
|
||||
'timers': Timer.objects.filter(corp_timer=False),
|
||||
'corp_timers': Timer.objects.filter(corp_timer=True, eve_corp=corp),
|
||||
'future_timers': Timer.objects.filter(corp_timer=False, eve_time__gte=timezone.now()),
|
||||
'past_timers': Timer.objects.filter(corp_timer=False, eve_time__lt=timezone.now()),
|
||||
}
|
||||
form = TimerForm(initial=data)
|
||||
return render(request, 'timerboard/update.html', context={'form': form})
|
||||
|
||||
return render(request, self.template_name, context=render_items)
|
||||
|
||||
|
||||
class TimerManagementView(BaseTimerView):
|
||||
permission_required = 'auth.timer_management'
|
||||
index_redirect = 'timerboard:view'
|
||||
success_url = reverse_lazy(index_redirect)
|
||||
model = Timer
|
||||
form_class = TimerForm
|
||||
|
||||
def get_timer(self, timer_id):
|
||||
return get_object_or_404(self.model, id=timer_id)
|
||||
|
||||
|
||||
class AddUpdateMixin:
|
||||
def get_form_kwargs(self):
|
||||
"""
|
||||
Inject the request user into the kwargs passed to the form
|
||||
"""
|
||||
kwargs = super(AddUpdateMixin, self).get_form_kwargs()
|
||||
kwargs.update({'user': self.request.user})
|
||||
return kwargs
|
||||
|
||||
|
||||
class AddTimerView(TimerManagementView, AddUpdateMixin, CreateView):
|
||||
template_name_suffix = '_create_form'
|
||||
|
||||
def form_valid(self, form):
|
||||
result = super(AddTimerView, self).form_valid(form)
|
||||
timer = self.object
|
||||
logger.info("Created new timer in {} at {} by user {}".format(timer.system, timer.eve_time, self.request.user))
|
||||
messages.success(self.request, _('Added new timer in %(system)s at %(time)s.') % {"system": timer.system,
|
||||
"time": timer.eve_time})
|
||||
return result
|
||||
|
||||
|
||||
class EditTimerView(TimerManagementView, AddUpdateMixin, UpdateView):
|
||||
template_name_suffix = '_update_form'
|
||||
|
||||
def form_valid(self, form):
|
||||
messages.success(self.request, _('Saved changes to the timer.'))
|
||||
return super(EditTimerView, self).form_valid(form)
|
||||
|
||||
|
||||
class RemoveTimerView(TimerManagementView, DeleteView):
|
||||
pass
|
||||
|
Loading…
x
Reference in New Issue
Block a user