Using sessions to define a personalized counter

counter/templates/baseTemplate.html

@@ -18,7 +18,7 @@
 		}
 
 		.seum-counter {
-			height: 125px;
+			height: 75px;
 			overflow: auto;
 		}
 

counter/templates/counterTemplate.html

@@ -11,7 +11,7 @@
         <div class="panel-heading">
           <h2 class="panel-title">Dernier seum {% if not counter.lastReset.noSeum %}<span class="badge pull-right">{{ counter.seumCount }}</span>{% endif %}</h2>
         </div>
-        <div class="seum-counter panel-body">
+        <div class="seum-counter panel-body" style="height:125px" id="container{{counter.id}}">
           {% if counter.lastReset.noSeum %}
           <strong>N'a pas encore eu le seum.</strong>
           <br> {% else %}
@@ -20,7 +20,7 @@
 
           <p>{{ counter.lastReset.reason }}</p>
           <div class="text-center">
-            <button id="button{{counter.id}}" class="btn btn-default btn-danger" type="button" onclick="getElementById('counter{{counter.id}}').style.display='inline';getElementById('button{{counter.id}}').style.display='none';">Remettre à zéro</button>
+            <button id="button{{counter.id}}" class="btn btn-default btn-danger" type="button" onclick="getElementById('counter{{counter.id}}').style.display='inline';getElementById('button{{counter.id}}').style.display='none';getElementById('container{{counter.id}}').style.height='225px';">Remettre à zéro</button>
           </div>
           <form style="display:none" id="counter{{counter.id}}" action="{% url 'reset-counter' %}" method="post">
             {% csrf_token %}

counter/templates/homeTemplate.html

@@ -3,44 +3,101 @@
 	<h1>SeumBook™</h1>
 </div>
 <div class="container-fluid">
-	<div class="row">
-		{% for counter in counters %}
-		<div class="col-md-4 col-sm-6 col-lg-3">
-			<div class="panel panel-{{counter.CSSclass}}" style="opacity:{{counter.opacity}}">
+	{% if chooseCounter %}
+	<div class="row" id="my-counter">
+		<div class="col-sm-12">
+			<div class="panel panel-warning">
 				<div class="panel-heading">
-					<a class="counter-link" href="{% url 'counter' id_counter=counter.id %}">
-						<h2 class="panel-title">{{ counter.trigramme }} <small>{{ counter.name }}</small>
-							{% if not counter.lastReset.noSeum %}<span class="pull-right badge">{{ counter.seumCount }}</span>{% endif %}
+					<h2 class="panel-title">Mon compteur</h2>
+				</div>
+				<div class="panel-body">
+					<p>Choisis ton compteur pour qu'il apparaisse en haut de la liste quand tu reviendras sur le site !</p>
+					<form action="{% url 'set-my-counter'%}" method="post">
+						{% csrf_token %}
+						<div class="form-group">
+							<label for="myCounterSelect">Mon trigramme:</label>
+							<select class="form-control" name="myCounter">
+								{% for counter in counters %}
+								<option value="{{ counter.id }}">{{ counter.trigramme }} ({{ counter.name }})</option>
+								{% endfor %}
+							</select>
+						</div>
+						<input type="hidden" name="redirect" value="{% url 'home' %}"></input>
+						<div class="text-center">
+							<button type="submit" class="btn btn-default btn-success">Définir mon compteur</button>
+						</div>
+					</form>
+				</div>
+			</div>
+		</div>
+	</div>
+	{% else %}
+	<div class="row" id="my-counter">
+		<div class="col-sm-12">
+			<div class="panel panel-warning">
+				<div class="panel-heading">
+					<a class="counter-link" href="{% url 'counter' id_counter=myCounter.id %}">
+						<h2 class="panel-title">{{ myCounter.trigramme }} <small>{{ myCounter.name }}</small>
+							{% if not myCounter.lastReset.noSeum %}<span class="pull-right badge">{{ myCounter.seumCount }}</span>{% endif %}
 						</h2>
 					</a>
 				</div>
-				<div class="seum-counter panel-body">
-					{% if counter.lastReset.noSeum %}
+				<div class="seum-counter panel-body" id="container{{myCounter.id}}" style="height:125px">
+					{% if myCounter.lastReset.noSeum %}
 					<strong>N'a pas encore eu le seum.</strong>
 					<br> {% else %}
-					<strong>N'a pas eu le seum depuis {{ counter.lastReset.formatted_delta }}.</strong>
+					<strong>Je n'ai pas eu le seum depuis {{ myCounter.lastReset.formatted_delta }}.</strong>
 					<br> {% endif %}
 
-					<p>{{ counter.lastReset.reason }}</p>
-					<div class="text-center">
-						<button id="button{{counter.id}}" class="btn btn-default btn-danger" type="button" onclick="getElementById('counter{{counter.id}}').style.display='inline';getElementById('button{{counter.id}}').style.display='none';">Remettre à zéro</button>
+					<p>{{ myCounter.lastReset.reason }}</p>
+					<div class="text-center" id="button{{myCounter.id}}">
+						<form action="{% url 'reset-my-counter' %}" method="post">
+							{% csrf_token %}
+							<button class="btn btn-default btn-danger" type="button" onclick="getElementById('counter{{myCounter.id}}').style.display='inline';getElementById('button{{myCounter.id}}').style.display='none';getElementById('container{{myCounter.id}}').style.height='225px';">Remettre à zéro</button>
+							<input type="hidden" name="redirect" value="{% url 'home' %}"></input>
+							<button type="submit" class="btn btn-default">Changer mon compteur</button>
+						</form>
 					</div>
-					<form style="display:none" id="counter{{counter.id}}" action="{% url 'reset-counter' %}" method="post">
+					<form style="display:none" id="counter{{myCounter.id}}" action="{% url 'reset-counter' %}" method="post">
 						{% csrf_token %}
 						<div class="form-group">
 							<label for="reason">Motif du seum :</label>
 							<textarea class="form-control" name="reason"></textarea>
 						</div>
-						<input type="hidden" name="counter" value="{{counter.id}}"></input>
+						<input type="hidden" name="counter" value="{{myCounter.id}}"></input>
 						<input type="hidden" name="redirect" value="{% url 'home' %}"></input>
 						<div class="text-center">
-							<button type="submit" class="btn btn-default btn-success">Foutre le seum</button>
+							<button type="submit" class="btn btn-default btn-success">J'ai le seum</button>
 						</div>
 					</form>
 				</div>
 			</div>
 		</div>
-		{% endfor %}
+	</div>
+	{% endif %}
+	<div class="row">
+		{% for counter in counters %} {% if chooseCounter or not counter.id == myCounter.id %}
+		<div class="col-md-4 col-sm-6 col-lg-3">
+			<div class="panel panel-{{counter.CSSclass}}" style="opacity:{{counter.opacity}}">
+				<div class="panel-heading">
+					<a class="counter-link" href="{% url 'counter' id_counter=counter.id %}">
+						<h2 class="panel-title">{{ counter.trigramme }} <small>{{ counter.name }}</small>
+							{% if not counter.lastReset.noSeum %}<span class="pull-right badge">{{ counter.seumCount }}</span>{% endif %}
+						</h2>
+					</a>
+				</div>
+				<div class="seum-counter panel-body">
+					{% if counter.lastReset.noSeum %}
+					<strong>N'a pas encore eu le seum.</strong>
+					<br> {% else %}
+					<strong>N'a pas eu le seum depuis {{ counter.lastReset.formatted_delta }}.</strong>
+					<br> {% endif %}
+
+					<p>{{ counter.lastReset.reason }}</p>
+				</div>
+			</div>
+		</div>
+		{% endif %} {% endfor %}
 	</div>
 	<div class="row">
 		<div class="col-sm-12">

counter/urls.py

@@ -6,4 +6,6 @@ urlpatterns = [
     url(r'^$', views.home,name="home"),
     url(r'^reset-counter/',views.resetCounter,name="reset-counter"),
     url(r'^counter/(?P<id_counter>\d+)$', views.counter, name="counter"),
+    url(r'^set-my-counter/',views.setMyCounter,name="set-my-counter"),
+    url(r'^reset-my-counter/',views.resetMyCounter,name="reset-my-counter"),
 ]

counter/views.py

@@ -28,10 +28,38 @@ def home(request):
     maxJSS = 0
     bestSeumeursNumber = 15
     # Display counters
-    counters = Counter.objects.all()
     lastResets = []
     # Calculates infos for each counter
     timezero = timedelta(0)
+
+    # First we determine if the session
+    if 'my-counter' in request.session:
+        chooseCounter = False
+        try:
+            myCounter = Counter.objects.get(id=request.session['my-counter'])
+            lastReset = Reset.objects.filter(
+                counter=myCounter).order_by('-timestamp')
+            if (lastReset.count() == 0):
+                # This person never had the seum
+                myCounter.lastReset = Reset()
+                myCounter.lastReset.delta = timezero
+                myCounter.lastReset.noSeum = True
+            else:
+                myCounter.lastReset = lastReset[0]
+                myCounter.lastReset.noSeum = False
+                myCounter.lastReset.delta = datetime.now(
+                ) - myCounter.lastReset.timestamp.replace(tzinfo=None)
+                myCounter.seumCount = Reset.objects.filter(
+                    counter=myCounter).count()
+            myCounter.lastReset.formatted_delta = format_timedelta(
+                myCounter.lastReset.delta, locale='fr', threshold=1)
+        except ObjectDoesNotExist:
+            chooseCounter = True
+    else:
+        myCounter = None
+        chooseCounter = True
+
+    counters = Counter.objects.all()
     for counter in counters:
         lastReset = Reset.objects.filter(
             counter=counter).order_by('-timestamp')
@@ -189,6 +217,8 @@ def home(request):
         'noGraph': noGraph,
         'noBestSeum': noBestSeum,
         'noSeumActivity': noSeumActivity,
+        'chooseCounter': chooseCounter,
+        'myCounter': myCounter,
     })
 
 
@@ -287,3 +317,19 @@ def counter(request, id_counter):
         'resets': resets,
         'seumFrequency': seumFrequency
     })
+
+
+def setMyCounter(request):
+    if (request.method == 'POST'):
+        # create a form instance and populate it with data from the request:
+        data = dict(request.POST)
+        # We put the id of the counter in the session
+        request.session['my-counter'] = data['myCounter'][0]
+        return HttpResponseRedirect(data['redirect'][0])
+
+
+def resetMyCounter(request):
+    if (request.method == 'POST'):
+        data = dict(request.POST)
+        del request.session['my-counter']
+        return HttpResponseRedirect(data['redirect'][0])