added student profile model, student profile page, student profile views

ennead/app.py

@@ -8,11 +8,13 @@
 
 from ennead.views.auth import register, register_page, login, login_page, logout
 from ennead.views.tasks import index
+from ennead.views.student_profile import create_update_student_profile, read_student_profile
 
 from ennead.models.base import database
 from ennead.models.user import User
 from ennead.models.task import TaskSet, Task
 from ennead.models.thread import Thread, Post
+from ennead.models.student_profile import StudentProfile
 
 
 def inject_user() -> None:
@@ -38,7 +40,7 @@ def create_app(config_path: Optional[str] = None) -> Flask:
     app.config.from_object(config)
 
     database.initialize(config.DB_CLASS(config.DB_NAME, **config.DB_PARAMS))
-    database.create_tables([User, Task, TaskSet, Thread, Post])
+    database.create_tables([User, Task, TaskSet, Thread, Post, StudentProfile])
 
     app.before_request(inject_user)
 
@@ -50,8 +52,12 @@ def create_app(config_path: Optional[str] = None) -> Flask:
     app.add_url_rule('/login', 'login', login, methods=['POST'])
     app.add_url_rule('/logout', 'logout', logout)
 
+    app.add_url_rule('/student_profile', 'read_student_profile', read_student_profile)
+    app.add_url_rule('/student_profile', 'create_student_profile', create_update_student_profile, methods=['POST'])
+
     return app
 
 
 if __name__ == '__main__':
     create_app('ennead.json').run()
+

ennead/models/student_profile.py

@@ -0,0 +1,42 @@
+"""Module for Ennead task model and corresponding helper classes"""
+
+from typing import TYPE_CHECKING, List
+
+from peewee import BooleanField, CharField, TextField, IntegerField, ForeignKeyField, DateField
+
+from datetime import date
+from ennead.models.user import User
+from flask import Request
+
+
+from ennead.models.base import BaseModel
+if TYPE_CHECKING:
+    # pylint: disable=R0401
+    from ennead.models.thread import Thread  # noqa: F401
+
+
+class StudentProfile(BaseModel):
+    """Student questionnaire
+
+    Attributes:
+        grade: year of education
+        city:  city of living
+        birth_date: date of birth
+        sex: sex (True if male)
+        allergy: human-readable list of allergies 
+        communication: preferred communication type (vk/telegram profile)
+        telephone: student's telephone number
+        parent_information: human-readable name and contacts of one 
+    """
+
+    grade: int = IntegerField()
+    city: str = CharField(32)
+    birth_date: date = DateField()
+    sex: bool = BooleanField()
+    allergy: str = TextField()
+    communication: str = TextField()
+    telephone: str = CharField(20)
+    parent_information: str = TextField()
+
+    user: User = ForeignKeyField(User, backref='student_profiles')
+

ennead/static/style.css

@@ -14,3 +14,7 @@
 	position: relative;
 	top: 0.05em;
 }
+
+textarea.student-profile-area {
+    height: 6em;
+}

ennead/templates/student_profile.html

@@ -0,0 +1,48 @@
+{% extends 'base.html' %}
+
+{% block title %}Ennead / Анкета ученика{% endblock %}
+
+{% block body %}
+	<div class="row">
+		<div class="col-4 offset-4">
+			<form method="POST">
+				<div class="form-group row">
+					<label class="col-12" for="grade">Класс, который вы закончили:</label>
+					<input class="col-12 form-control" type="number" name="grade" id="grade" required value="{{ student_profile.grade }}">
+				</div>
+				<div class="form-group row">
+					<label class="col-12" for="city">Город, в котором вы живёте:</label>
+					<input class="col-12 form-control" name="city" id="city" required value="{{ student_profile.city }}">
+				</div>
+				<div class="form-group row">
+					<label class="col-12" for="birth_date">Дата вашего рождения:</label>
+					<input class="col-12 form-control" type="date" name="birth_date" id="birth_date" required value="{{ student_profile.birth_date }}">
+				</div>
+				<div class="form-group row">
+					<label class="col-12" for="sex">Ваш пол:</label>
+                    <select class="col-12 form-control" name="sex" id="sex" required value="{{ student_profile.sex }}">
+                        <option value="1">Мужской</option>
+                        <option value="0">Женский</option>
+                    </select>
+				</div>
+				<div class="form-group row">
+					<label class="col-12" for="allergy">Перечислите всё, на что у вас аллергия:</label>
+                    <textarea class="col-12 form-control" type="textarea" name="allergy" id="allergy" required>{{ student_profile.allergy }}</textarea>
+				</div>
+				<div class="form-group row">
+					<label class="col-12" for="communication">Ссылка на вас в социальных сетях (vk, telegram):</label>
+					<input class="col-12 form-control" name="communication" id="communication" required value="{{ student_profile.communication }}">
+				</div>
+				<div class="form-group row">
+					<label class="col-12" for="telephone">Ваш мобильный телефон:</label>
+					<input class="col-12 form-control" name="telephone" id="telephone" required value="{{ student_profile.telephone }}">
+				</div>
+				<div class="form-group row">
+					<label class="col-12" for="parent_information">Контактный телефон и имя вашего родителя:</label>
+					<textarea class="col-12 student-profile-area form-control" type="textarea" name="parent_information" id="parent_information" required >{{ student_profile.parent_information }}</textarea>
+				</div>
+				<input type="submit" class="btn btn-primary" value="Сохранить">
+			</form>
+		</div>
+	</div>
+{% endblock %}

ennead/views/student_profile.py

@@ -0,0 +1,48 @@
+"""Views, used for student profiles"""
+
+from datetime import date
+from ennead.models.user import User
+
+from flask import session, current_app, render_template, request, redirect, url_for, g
+from werkzeug.wrappers import Response
+
+from ennead.utils import require_logged_in
+from ennead.models.user import User, UserGroup
+from ennead.models.student_profile import StudentProfile
+
+
+@require_logged_in
+def read_student_profile() -> Response:
+    """GET /student_profile: read profile page"""
+
+    student_profile = None
+    try:
+        student_profile = g.user.student_profiles[0]
+    except:
+        student_profile = StudentProfile()
+
+    return render_template('student_profile.html', student_profile=student_profile)
+
+
+@require_logged_in
+def create_update_student_profile() -> Response:
+    """POST /student_profile: update student profile"""
+
+    student_profile = None
+    try:
+        student_profile = g.user.student_profiles[0]
+    except:
+        student_profile = StudentProfile()
+
+    for field in ('grade', 'city', 'birth_date', 'allergy', 'sex',
+                  'communication', 'parent_information'):
+        setattr(student_profile, field, request.form[field])
+
+    # TODO: add telephone validation
+    student_profile.telephone = request.form['telephone']
+
+    student_profile.user = g.user
+
+    student_profile.save()
+    return redirect(url_for('read_student_profile'))
+