From 026ee568e4cb165062d5d4fa58baca60ddc0ed82 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=89lie=20Bouttier?= <elie@bouttier.eu>
Date: Wed, 5 Oct 2016 21:46:59 +0200
Subject: [PATCH] track duration

---
 proposals/forms.py                            |  2 +-
 .../migrations/0015_auto_20161005_1932.py     | 25 +++++++++++++++++++
 proposals/models.py                           |  8 ++++++
 proposals/templates/proposals/track_list.html |  4 ++-
 proposals/templatetags/proposals_tags.py      |  7 ++++++
 proposals/views.py                            |  2 ++
 6 files changed, 46 insertions(+), 2 deletions(-)
 create mode 100644 proposals/migrations/0015_auto_20161005_1932.py

diff --git a/proposals/forms.py b/proposals/forms.py
index eeda9da..546e001 100644
--- a/proposals/forms.py
+++ b/proposals/forms.py
@@ -30,7 +30,7 @@ class TalkForm(forms.ModelForm):
 
     class Meta:
         model = Talk
-        fields = ['title', 'abstract', 'description', 'topics', 'track', 'notes', 'event', 'speakers']
+        fields = ['title', 'abstract', 'description', 'topics', 'track', 'notes', 'event', 'duration', 'speakers']
         widgets = {'topics': forms.CheckboxSelectMultiple(), 'speakers': Select2TagWidget()}
         help_texts = {
             'abstract': _('Should be less than 255 characters'),
diff --git a/proposals/migrations/0015_auto_20161005_1932.py b/proposals/migrations/0015_auto_20161005_1932.py
new file mode 100644
index 0000000..28b337c
--- /dev/null
+++ b/proposals/migrations/0015_auto_20161005_1932.py
@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.10 on 2016-10-05 19:32
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('proposals', '0014_track_managers'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='event',
+            name='duration',
+            field=models.IntegerField(default=0),
+        ),
+        migrations.AddField(
+            model_name='talk',
+            name='duration',
+            field=models.IntegerField(default=0),
+        ),
+    ]
diff --git a/proposals/models.py b/proposals/models.py
index 9ca97c6..6f5ec26 100644
--- a/proposals/models.py
+++ b/proposals/models.py
@@ -49,6 +49,9 @@ class Track(PonyConfModel):
     class Meta:
         unique_together = ('site', 'name')
 
+    def estimated_duration(self):
+        return sum([talk.estimated_duration() for talk in self.talk_set.all()])
+
     def __str__(self):
         return self.name
 
@@ -81,6 +84,7 @@ class Event(models.Model):
 
     site = models.ForeignKey(Site, on_delete=models.CASCADE)
     name = models.CharField(max_length=64)
+    duration = models.IntegerField(default=0)
 
     class Meta:
         unique_together = ('site', 'name')
@@ -108,6 +112,7 @@ class Talk(PonyConfModel):
     notes = models.TextField(blank=True, verbose_name=_('Notes'))
     event = models.ForeignKey(Event, verbose_name=_('Intervention kind'))
     accepted = models.NullBooleanField(default=None)
+    duration = models.IntegerField(default=0)
 
     class Meta:
         ordering = ('title',)
@@ -115,6 +120,9 @@ class Talk(PonyConfModel):
     def __str__(self):
         return self.title
 
+    def estimated_duration(self):
+        return self.duration or self.event.duration
+
     def get_absolute_url(self):
         return reverse('show-talk', kwargs={'slug': self.slug})
 
diff --git a/proposals/templates/proposals/track_list.html b/proposals/templates/proposals/track_list.html
index 1a5c7ca..0d526bd 100644
--- a/proposals/templates/proposals/track_list.html
+++ b/proposals/templates/proposals/track_list.html
@@ -1,6 +1,6 @@
 {% extends 'base.html' %}
 
-{% load bootstrap3 accounts_tags i18n %}
+{% load bootstrap3 accounts_tags proposals_tags i18n %}
 
 {% block tracktab %} class="active"{% endblock %}
 
@@ -23,6 +23,8 @@
         <a href="{{ track.get_absolute_url }}">{{ track.talk_set.count }} {% trans "talk" %}{{ track.talk_set.count|pluralize }}</a>
         |
         <a href="{% url 'edit-track' track.slug %}">{% bootstrap_icon "pencil" %}</a>
+        |
+        {{ track.estimated_duration|duration_format }}
     {% endif %}
   </div>
     {% cycle '' '<div class="clearfix visible-xs"></div>' %}
diff --git a/proposals/templatetags/proposals_tags.py b/proposals/templatetags/proposals_tags.py
index 226a924..2656151 100644
--- a/proposals/templatetags/proposals_tags.py
+++ b/proposals/templatetags/proposals_tags.py
@@ -9,3 +9,10 @@ register = template.Library()
 @register.simple_tag
 def markdown(value):
     return markdown_to_html(value)
+
+@register.filter('duration_format')
+def duration_format(value):
+    value = int(value)
+    hours = int(value/60)
+    minutes = value%60
+    return '%d h %02d' % (hours, minutes)
diff --git a/proposals/views.py b/proposals/views.py
index 03f4567..47bcb84 100644
--- a/proposals/views.py
+++ b/proposals/views.py
@@ -153,10 +153,12 @@ def talk_edit(request, talk=None):
         form.fields['topics'].disabled = True
         if not talk.is_editable_by(request.user):
             form.fields.pop('track')
+            form.fields.pop('duration')
     else:
         form.fields['speakers'].initial = [request.user]
         if not is_orga(request, request.user):
             form.fields.pop('track')
+            form.fields.pop('duration')
     if request.method == 'POST' and form.is_valid():
         if hasattr(talk, 'id'):
             talk = form.save()