From 0c16d9a73b3af5b50ee12ead9d532bb89a9b6aa4 Mon Sep 17 00:00:00 2001
From: Gergely Polonkai <gergely.polonkai@ericsson.com>
Date: Thu, 26 Nov 2015 14:51:04 +0100
Subject: [PATCH] Add word drawing support

---
 words/apps.py                              | 23 +++++++++++++++++++++
 words/migrations/0003_draw_word_related.py | 19 +++++++++++++++++
 words/models.py                            |  2 +-
 words/tests.py                             | 24 +++++++++++++++++++++-
 4 files changed, 66 insertions(+), 2 deletions(-)
 create mode 100644 words/migrations/0003_draw_word_related.py

diff --git a/words/apps.py b/words/apps.py
index e28e524..7a52dca 100644
--- a/words/apps.py
+++ b/words/apps.py
@@ -24,8 +24,31 @@ def _current_word(self):
 
     return None
 
+def _draw_word(self):
+    if self.current_word() is not None:
+        return self.current_word()
+
+    from .models import Word, Draw
+
+    # Find all words
+    # Exclude all words that has an accepted draw for this user
+    # Choose a random one
+    # If there are no more words, return None
+    word = Word.objects \
+               .exclude(draws__accepted=True, draws__user=self) \
+               .order_by('?') \
+               .first()
+
+    if word is None:
+        return None
+
+    Draw.objects.create(user=self, word=word, accepted=None)
+
+    return word
+
 class WordsConfig(AppConfig):
     name = 'words'
 
     def ready(self):
         User.current_word = _current_word
+        User.draw_word = _draw_word
diff --git a/words/migrations/0003_draw_word_related.py b/words/migrations/0003_draw_word_related.py
new file mode 100644
index 0000000..0458500
--- /dev/null
+++ b/words/migrations/0003_draw_word_related.py
@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('words', '0002_draw_work'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='draw',
+            name='word',
+            field=models.ForeignKey(related_name='draws', to='words.Word'),
+        ),
+    ]
diff --git a/words/models.py b/words/models.py
index f9af291..9139d53 100644
--- a/words/models.py
+++ b/words/models.py
@@ -50,7 +50,7 @@ class WordTranslation(models.Model):
 
 class Draw(models.Model):
     user = models.ForeignKey(User)
-    word = models.ForeignKey(Word)
+    word = models.ForeignKey(Word, related_name='draws')
     accepted = models.NullBooleanField()
     timestamp = models.DateTimeField(default=timezone.now)
 
diff --git a/words/tests.py b/words/tests.py
index 5f128d5..5f7da38 100644
--- a/words/tests.py
+++ b/words/tests.py
@@ -3,7 +3,7 @@ from django.core.exceptions import ValidationError
 from django.utils.translation import activate
 from django.test import TestCase
 
-from .models import Word, WordTranslation, Draw
+from .models import Word, WordTranslation, Draw, Work
 
 class WordTest(TestCase):
     def setUp(self):
@@ -71,3 +71,25 @@ class DrawTest(TestCase):
         draw.accepted = True
         draw.save()
         self.assertEquals(self.word, self.user.current_word())
+
+    def test_draw_word(self):
+        draw = Draw.objects.create(user=self.user,
+                            word=self.word,
+                            accepted=None)
+        self.assertEquals(self.word, self.user.draw_word())
+        draw.accepted = True
+        draw.save()
+        Work.objects.create(draw=draw)
+
+        word2 = Word.objects.create()
+        self.assertEquals(word2, self.user.draw_word())
+
+        draw = Draw.objects.get(user=self.user, word=word2)
+        self.assertIsNotNone(draw)
+        self.assertIsNone(draw.accepted)
+
+        draw.accepted = True
+        draw.save()
+        Work.objects.create(draw=draw)
+
+        self.assertIsNone(self.user.draw_word())