Keywords in DB

counter/admin.py

@@ -1,8 +1,10 @@
 from django.contrib import admin
 
 # Register your models here.
-from .models import Counter, Reset, Like
+from .models import Counter, Reset, Like, Keyword, Hashtag
 
 admin.site.register(Counter)
 admin.site.register(Reset)
 admin.site.register(Like)
+admin.site.register(Keyword)
+admin.site.register(Hashtag)

counter/migrations/0007_auto_20170113_1603.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.10.4 on 2017-01-13 16:03
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('counter', '0006_counter_sort_by_score'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Hashtag',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+            ],
+            options={
+                'verbose_name': 'hashtags',
+            },
+        ),
+        migrations.CreateModel(
+            name='Keyword',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('text', models.CharField(max_length=128, unique=True, verbose_name='texte')),
+            ],
+            options={
+                'verbose_name': 'mot-clé',
+                'verbose_name_plural': 'mots-clés',
+            },
+        ),
+        migrations.AddField(
+            model_name='hashtag',
+            name='keyword',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='counter.Keyword', verbose_name='hashtag'),
+        ),
+        migrations.AddField(
+            model_name='hashtag',
+            name='reset',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='counter.Reset', verbose_name='remise à zéro'),
+        ),
+    ]

counter/models.py

@@ -67,3 +67,26 @@ class Like(models.Model):
 
     def __str__(self):
         return '%s aime %s' % (self.liker, self.reset)
+
+
+class Keyword(models.Model):
+    text = models.CharField('texte', max_length=128, unique=True)
+
+    class Meta:
+        verbose_name = 'mot-clé'
+        verbose_name_plural = 'mots-clés'
+
+    def __str__(self):
+        return '#%s' % (self.text)
+
+
+class Hashtag(models.Model):
+    keyword = models.ForeignKey('Keyword', verbose_name='hashtag')
+    reset = models.ForeignKey('Reset', verbose_name='remise à zéro')
+
+    class Meta:
+        verbose_name = 'hashtag'
+        verbose_name_plural = 'hashtags'
+
+    def __str__(self):
+        return '%s pour %s' % (self.keyword, self.reset)

counter/utils.py

@@ -0,0 +1,26 @@
+import re
+import unicodedata
+from counter.models import Keyword
+
+
+# Analyse the reason of the reset and retrieve all the keyword objects that it
+# contains, creating the objects in the DB on-the-fly if necessary
+def parseSeumReason(reason):
+    hashtags = re.findall(r'(?<=#)[^#\s]+', reason, re.I)
+    hashtags = [removeAccentsToLowercase(u) for u in hashtags]
+    keywords = []
+    for hashtag in hashtags:
+        try:
+            keyword = Keyword.objects.get(text=hashtag)
+            keywords.append(keyword)
+        except Keyword.DoesNotExist:
+            keyword = Keyword(text=hashtag)
+            keyword.save()
+            keywords.append(keyword)
+    return keywords
+
+
+def removeAccentsToLowercase(input_str):
+    nfkd_form = unicodedata.normalize('NFKD', input_str)
+    return u"".join([c.lower() for c in nfkd_form
+                     if not unicodedata.combining(c)])

counter/views.py

@@ -1,5 +1,5 @@
 from django.shortcuts import render
-from counter.models import Counter, Reset, Like
+from counter.models import Counter, Reset, Like, Keyword, Hashtag
 from django.contrib.auth.models import User
 from babel.dates import format_timedelta, format_datetime
 from datetime import datetime, timedelta
@@ -18,6 +18,7 @@ import math
 import copy
 import functools
 from django.utils import timezone
+from counter.utils import parseSeumReason
 
 # JSS above this limit will not be displayed on the home page col graph
 JSS_limit = 7
@@ -275,6 +276,7 @@ def resetCounter(request):
         data = dict(request.POST)
 
         who = Counter.objects.get(pk=int(data['who'][0]))
+        reason = data['reason'][0]
         if 'counter' in data.keys():
             counter = Counter.objects.get(pk=int(data['counter'][0]))
         else:
@@ -294,6 +296,13 @@ def resetCounter(request):
             return HttpResponseRedirect(data['redirect'][0])
 
         reset.save()
+
+        # Now we deal with the hashtags
+        keywords = parseSeumReason(reason)
+        for keyword in keywords:
+            hashtag = Hashtag(reset=reset, keyword=keyword)
+            hashtag.save()
+
         # We send the emails only to those who want
         emails = [u.email for u in Counter.objects.all()
                   if u.email_notifications]