39 | 0
40 | 0
41 |
42 |
14 |
35 |
36 | {% endblock %}
37 |
38 | {% block scripts %}
39 | {{ block.super }}
40 |
41 |
42 |
43 |
44 |
45 |
46 |
47 |
48 |
49 | {% endblock %}
--------------------------------------------------------------------------------
/biproductive/habits/migrations/0001_initial.py:
--------------------------------------------------------------------------------
1 | # Generated by Django 3.2.5 on 2021-09-22 16:31
2 |
3 | import django.db.models.deletion
4 | from django.conf import settings
5 | from django.db import migrations, models
6 |
7 |
8 | class Migration(migrations.Migration):
9 | initial = True
10 |
11 | dependencies = [
12 | migrations.swappable_dependency(settings.AUTH_USER_MODEL),
13 | ]
14 |
15 | operations = [
16 | migrations.CreateModel(
17 | name="Habit",
18 | fields=[
19 | (
20 | "id",
21 | models.BigAutoField(
22 | auto_created=True,
23 | primary_key=True,
24 | serialize=False,
25 | verbose_name="ID",
26 | ),
27 | ),
28 | ("habit_name", models.CharField(max_length=30)),
29 | ("creation_date", models.DateTimeField()),
30 | (
31 | "user",
32 | models.ForeignKey(
33 | on_delete=django.db.models.deletion.CASCADE,
34 | to=settings.AUTH_USER_MODEL,
35 | ),
36 | ),
37 | ],
38 | ),
39 | migrations.CreateModel(
40 | name="HabitUsage",
41 | fields=[
42 | (
43 | "id",
44 | models.BigAutoField(
45 | auto_created=True,
46 | primary_key=True,
47 | serialize=False,
48 | verbose_name="ID",
49 | ),
50 | ),
51 | ("usage_time", models.DateTimeField()),
52 | (
53 | "habit",
54 | models.ForeignKey(
55 | on_delete=django.db.models.deletion.CASCADE, to="habits.habit"
56 | ),
57 | ),
58 | ],
59 | ),
60 | ]
61 |
--------------------------------------------------------------------------------
/biproductive/report_generator/views.py:
--------------------------------------------------------------------------------
1 | import os
2 | from datetime import datetime
3 | from pathlib import Path
4 |
5 | import seaborn as sns
6 | import temppathlib
7 | from django.contrib.auth.decorators import login_required
8 | from django.http import FileResponse, Http404
9 | from matplotlib import pyplot as plt
10 |
11 | from home.scripts import correlation
12 |
13 |
14 | # Create your views here.
15 |
16 |
17 | def download_file(request, file_path):
18 | try:
19 | file_name = f'habit-report-{datetime.now().date()}.pdf'
20 | response = FileResponse(open(file_path, 'rb'))
21 | response['content_type'] = "application/octet-stream"
22 | response['Content-Disposition'] = 'attachment; filename=' + file_name
23 | return response
24 | except Exception:
25 | raise Http404
26 |
27 |
28 | def generate_md_report(path_dir: Path, user):
29 | habit_table, productivity_table, corr = correlation(user)
30 | plt.figure(figsize=(6, 3))
31 | plt.xticks(rotation=20)
32 | fig = sns.lineplot(data=productivity_table, x='date', y='brain-activity')
33 | plot_path = str(path_dir / 'plot.png')
34 | plt.tight_layout()
35 | fig.figure.savefig(plot_path)
36 |
37 | with open(path_dir / "output.md", 'w') as file:
38 | file.write(f'# Weekly report, generated in {datetime.now().date()}\n')
39 | file.write("### Weekly habit statistics\n")
40 | file.write(habit_table.to_markdown() + "\n")
41 | file.write("### Weekly brain activity plot\n")
42 | file.write(f"\n")
43 | file.write("### Weekly correlation analysis of habits\n")
44 | file.write(corr.to_markdown() + "\n")
45 |
46 |
47 | @login_required(redirect_field_name='login')
48 | def download_view(request):
49 | with temppathlib.TemporaryDirectory() as folder:
50 | generate_md_report(path_dir=folder.path, user=request.user)
51 |
52 | md_path = str(folder.path / 'output.md')
53 | pdf_path = str(folder.path / 'output.pdf')
54 | os.system(f'md2pdf {md_path} {pdf_path}')
55 |
56 | return download_file(request, pdf_path)
57 |
--------------------------------------------------------------------------------
/biproductive/habits/views.py:
--------------------------------------------------------------------------------
1 | from datetime import datetime
2 |
3 | from django.contrib.auth.decorators import login_required
4 | from django.shortcuts import redirect, render
5 | from django.views.decorators.csrf import csrf_exempt
6 |
7 | from .forms import AddHabitForm, HabitTrackingForm
8 | from .models import Habit, HabitUsage
9 |
10 |
11 | @csrf_exempt
12 | @login_required(login_url="login")
13 | def add_habit(request):
14 | if request.method == "POST":
15 | form = AddHabitForm(request.POST)
16 | if form.is_valid():
17 | habit_name = form.cleaned_data["habit_name"]
18 | habit = Habit(
19 | habit_name=habit_name, user=request.user, creation_date=datetime.now()
20 | )
21 | habit.save()
22 | return redirect("home")
23 | else:
24 | return render(request, "add_habit.html", context={"form": form}, status=401)
25 | else:
26 | form = AddHabitForm()
27 | return render(request, "add_habit.html", {"form": form}, status=200)
28 |
29 |
30 | @csrf_exempt
31 | @login_required(login_url="login")
32 | def track_habits(request):
33 | if request.method == 'POST':
34 | form = HabitTrackingForm(request.user, request.POST)
35 | if form.is_valid():
36 | for item in form.cleaned_data.items():
37 | if item[1] == 'True':
38 | habit = Habit.objects.filter(user=request.user, habit_name=item[0])[0]
39 | usage = HabitUsage(habit=habit, usage_time=datetime.now(), status=True)
40 | usage.save()
41 | elif item[1] == 'False':
42 | habit = Habit.objects.filter(user=request.user, habit_name=item[0])[0]
43 | usage = HabitUsage(habit=habit, usage_time=datetime.now(), status=False)
44 | usage.save()
45 | return redirect('home')
46 | else:
47 | return render(request, "track_habits.html", context={"form": form}, status=401)
48 | else:
49 | form = HabitTrackingForm(request.user)
50 | return render(request, "track_habits.html", {'form': form}, status=200)
51 |
--------------------------------------------------------------------------------
/biproductive/productivity/tests.py:
--------------------------------------------------------------------------------
1 | import json
2 | from datetime import date
3 | from http import HTTPStatus
4 |
5 | from django.contrib.auth.models import User
6 | from django.test import TestCase
7 | from django.urls import reverse
8 | from productivity.models import ProductivityCheck
9 |
10 |
11 | class TestProductivityGame(TestCase):
12 | def setUp(self) -> None:
13 | self.username = "testUser"
14 | self.email = "testUser"
15 | self.password = "+AV/*~K\4u3k^5HC"
16 |
17 | self.user = User.objects.create_user(
18 | username=self.username, email=self.email, password=self.password
19 | )
20 | self.user.save()
21 |
22 | def test_can_play(self):
23 | self.assertTrue(
24 | self.client.login(username=self.username, password=self.password),
25 | "login failed",
26 | )
27 | response = self.client.get(path=reverse("game"), data={})
28 | self.assertEqual(response.status_code, 200)
29 |
30 | def test_already_played(self):
31 | productivity = ProductivityCheck.objects.create(
32 | user=self.user, productivity_value=50, date=date.today()
33 | )
34 | productivity.save()
35 |
36 | self.assertTrue(
37 | self.client.login(username=self.username, password=self.password),
38 | "login failed",
39 | )
40 | response = self.client.get(path=reverse("game"), data={})
41 | self.assertEqual(response.status_code, 403)
42 |
43 | def test_not_logged_in(self):
44 | response = self.client.get(path=reverse("game"), data={})
45 | self.assertEqual(response.status_code, 302) # Redirect to login page
46 |
47 | def test_can_send_fetch_query(self):
48 | self.assertTrue(
49 | self.client.login(username=self.username, password=self.password),
50 | "login failed",
51 | )
52 | json_data = json.dumps({"prod_score": 50})
53 | response = self.client.generic(
54 | method="POST", path=reverse("game"), data=json_data
55 | )
56 | self.assertEqual(response.status_code, HTTPStatus.OK)
57 |
58 | response = self.client.get(path=reverse("game"), data={})
59 | self.assertEqual(response.status_code, HTTPStatus.FORBIDDEN)
60 |
--------------------------------------------------------------------------------
/biproductive/habits/templates/add_habit.html:
--------------------------------------------------------------------------------
1 | {% extends "base_generic.html" %}
2 |
3 | {% block title %} Add habit {% endblock %}
4 |
5 | {% block content %}
6 |
7 | {% if message %}
15 |
16 |
18 |
19 |
26 |
27 | Habit statistics of last days
17 || {{ name }} | 22 | {% endfor %} 23 |
|---|
28 |
34 | Change of brain activity during last days
29 | 30 |
31 |
32 |
33 | {{ message }}
{% endif %} 8 | 9 |
11 |
48 |
12 |
47 |
13 |
46 |
14 |
45 |
15 |
44 |
16 |
43 | Add habit
17 | 18 | 41 | 42 |
46 | Chose your theme !
47 |
54 | Pokemon
48 |Star Wars
49 |Lord of the Ring
50 |Disney
51 |Pixar
52 |Harry Potter
53 |{{ message }}
{% endif %} 8 | 9 |
11 |
65 |
12 |
64 |
13 |
63 |
14 |
62 |
15 |
61 |
16 |
60 | Login
17 | 18 | 58 | 59 |{{ message }}
{% endif %} 8 | 9 |
11 |
59 |
12 |
58 |
13 |
57 |
14 |
56 |
15 |
55 |
16 |
54 | Fill habits
17 | 18 | 52 | 53 |{{ message }}
{% endif %} 8 | 9 |
11 |
83 |
12 |
82 |
13 |
81 |
14 |
80 |
15 |
79 |
16 |
78 | Create an account
17 | 18 | 76 | 77 |in " + time + " seconds"; 270 | 271 | postScore("/productivity/", {"prod_score": score}).then((data) => {}) 272 | postElt.classList.remove("hidden"); 273 | } 274 | } 275 | click = 0; 276 | } 277 | } 278 | } 279 | 280 | againElt.addEventListener("click", resetGame); 281 | 282 | function resetGame() { 283 | // reset game 284 | tempElt1 = ""; 285 | tempElt2 = ""; 286 | click = -1; 287 | win = 0; 288 | score = 0; 289 | time = 0; 290 | postElt.classList.add("hidden"); 291 | preElt.classList.remove("hidden"); 292 | for (let i = 0; i < 20; i++) { 293 | boxElts[i].classList.add("play"); 294 | boxElts[i].firstChild.classList.add("hidden"); 295 | } 296 | timeElt.textContent = time; 297 | scoreElt.textContent = score; 298 | } --------------------------------------------------------------------------------