From f270bbb280fb84d632e0a6930d43ef099967daa5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=89lie=20Bouttier?= <elie@bouttier.eu>
Date: Mon, 10 Oct 2016 23:07:07 +0200
Subject: [PATCH] users disponibilities

---
 .../migrations/0013_availabilitytimeslot.py   | 25 +++++++++++++
 accounts/models.py                            | 23 ++++++++++++
 accounts/tests.py                             | 35 +++++++++++++++++++
 3 files changed, 83 insertions(+)
 create mode 100644 accounts/migrations/0013_availabilitytimeslot.py

diff --git a/accounts/migrations/0013_availabilitytimeslot.py b/accounts/migrations/0013_availabilitytimeslot.py
new file mode 100644
index 0000000..ab8eee7
--- /dev/null
+++ b/accounts/migrations/0013_availabilitytimeslot.py
@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.10 on 2016-10-10 21:08
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('accounts', '0012_auto_20161006_1853'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='AvailabilityTimeslot',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('start', models.DateTimeField(blank=True)),
+                ('end', models.DateTimeField(blank=True)),
+                ('participation', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='availabilities', to='accounts.Participation')),
+            ],
+        ),
+    ]
diff --git a/accounts/models.py b/accounts/models.py
index b97897f..4353604 100644
--- a/accounts/models.py
+++ b/accounts/models.py
@@ -105,3 +105,26 @@ class Participation(PonyConfModel):
     @property
     def talk_set(self):
         return self.user.talk_set.filter(site=self.site)
+
+    # return True, False or None if availabilities have not been filled
+    def is_available(self, start, end=None):
+        if not self.availabilities.exists():
+            return None
+        for timeslot in self.availabilities.all():
+            if start < timeslot.start:
+                continue
+            if start > timeslot.end:
+                continue
+            if end:
+                assert(start < end)
+                if end > timeslot.end:
+                    continue
+            return True
+        return False
+
+
+class AvailabilityTimeslot(models.Model):
+
+    participation = models.ForeignKey(Participation, related_name='availabilities')
+    start = models.DateTimeField(blank=True)
+    end = models.DateTimeField(blank=True)
diff --git a/accounts/tests.py b/accounts/tests.py
index 676be1e..77a7a61 100644
--- a/accounts/tests.py
+++ b/accounts/tests.py
@@ -71,3 +71,38 @@ class AccountTests(TestCase):
                                            }).status_code, 200)
         self.assertEqual(User.objects.get(username='a').profile.biography, 'foo')
         self.assertEqual(Participation.objects.get(user=User.objects.get(username='a')).video_licence, 2)
+
+
+from datetime import datetime
+from .models import AvailabilityTimeslot
+class DisponibilitiesTests(TestCase):
+
+    def setUp(self):
+        self.user = User.objects.create_user('a', email='a@example.org', password='a')
+        self.participation = Participation.objects.create(user=self.user, site=Site.objects.first())
+
+    def test_is_available(self):
+        from django.utils.timezone import is_naive, get_default_timezone
+        tz = get_default_timezone()
+        d = {}
+        for i in range(8, 18, 1):
+            d[i] = datetime(2016, 10, 10, i, 0, 0, tzinfo=tz)
+        self.assertEquals(self.participation.is_available(d[10]), None)
+        AvailabilityTimeslot.objects.create(participation=self.participation, start=d[10], end=d[12])
+        self.assertEquals(self.participation.is_available(d[9]), False)
+        self.assertEquals(self.participation.is_available(d[11]), True)
+        self.assertEquals(self.participation.is_available(d[13]), False)
+        self.assertEquals(self.participation.is_available(d[8], d[9]), False)
+        self.assertEquals(self.participation.is_available(d[9], d[11]), False)
+        self.assertEquals(self.participation.is_available(d[10], d[11]), True)
+        self.assertEquals(self.participation.is_available(d[11], d[12]), True)
+        self.assertEquals(self.participation.is_available(d[10], d[12]), True)
+        self.assertEquals(self.participation.is_available(d[11], d[13]), False)
+        self.assertEquals(self.participation.is_available(d[13], d[14]), False)
+        AvailabilityTimeslot.objects.create(participation=self.participation, start=d[14], end=d[16])
+        self.assertEquals(self.participation.is_available(d[10], d[12]), True)
+        self.assertEquals(self.participation.is_available(d[14], d[16]), True)
+        self.assertEquals(self.participation.is_available(d[11], d[15]), False)
+        self.assertEquals(self.participation.is_available(d[11], d[17]), False)
+        self.assertEquals(self.participation.is_available(d[13], d[17]), False)
+        self.assertEquals(self.participation.is_available(d[9], d[15]), False)