workshop attendee: models

proposals/migrations/0026_auto_20161106_2319.py

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.10 on 2016-11-06 23:19
+from __future__ import unicode_literals
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('proposals', '0025_event_label'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Attendee',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('created', models.DateTimeField(auto_now_add=True)),
+                ('updated', models.DateTimeField(auto_now=True)),
+                ('name', models.CharField(blank=True, default='', max_length=64)),
+                ('email', models.EmailField(blank=True, default='', max_length=254)),
+                ('user', models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'abstract': False,
+            },
+        ),
+        migrations.AddField(
+            model_name='talk',
+            name='attendees_limit',
+            field=models.PositiveIntegerField(default=0, verbose_name='Max. number of attendees'),
+        ),
+        migrations.AddField(
+            model_name='talk',
+            name='registration_required',
+            field=models.BooleanField(default=False),
+        ),
+        migrations.AddField(
+            model_name='talk',
+            name='attendees',
+            field=models.ManyToManyField(to='proposals.Attendee', verbose_name='Attendees'),
+        ),
+    ]

proposals/models.py

@@ -104,6 +104,25 @@ class Event(models.Model):
         return reverse('list-talks') + '?kind=%d' % self.pk
 
 
+class Attendee(PonyConfModel):
+
+    user = models.ForeignKey(User, null=True)
+    name = models.CharField(max_length=64, blank=True, default="")
+    email = models.EmailField(blank=True, default="")
+
+    def get_name(self):
+        if user:
+            return str(user.profile)
+        else:
+            return name
+
+    def get_email(self):
+        if user:
+            return user.email
+        else:
+            return self.email
+
+
 class Talk(PonyConfModel):
 
     site = models.ForeignKey(Site, on_delete=models.CASCADE)
@@ -123,6 +142,9 @@ class Talk(PonyConfModel):
     duration = models.PositiveIntegerField(default=0, verbose_name=_('Duration (min)'))
     room = models.ForeignKey(Room, blank=True, null=True, default=None)
     plenary = models.BooleanField(default=False)
+    registration_required = models.BooleanField(default=False)
+    attendees = models.ManyToManyField(Attendee, verbose_name=_('Attendees'))
+    attendees_limit = models.PositiveIntegerField(default=0, verbose_name=_('Max. number of attendees'))
 
     class Meta:
         ordering = ('title',)
@@ -177,6 +199,13 @@ class Talk(PonyConfModel):
         else:
             return None
 
+    @property
+    def remaining_attendee(self):
+        if self.registration_required and self.attendees_limit:
+            return self.attendees_limit - self.attendees.count()
+        else:
+            return None # = infinity \o/
+
     class Meta:
         ordering = ('event__id',)