PEPify

Author: VorontsovIE

ennead/app.py

@@ -63,9 +63,10 @@ def create_app(config_path: Optional[str] = None) -> Flask:
     app.add_url_rule('/login', 'login_page', login_page)
     app.add_url_rule('/login', 'login', login, methods=['POST'])
     app.add_url_rule('/logout', 'logout', logout)
-    
+
     app.add_url_rule('/thread/<int:task_id>/<int:student_id>', 'thread', thread_page)
-    app.add_url_rule('/thread/<int:task_id>/<int:student_id>', 'post_to_thread', post_to_thread, methods=['POST'])
+    app.add_url_rule('/thread/<int:task_id>/<int:student_id>',
+                     'post_to_thread', post_to_thread, methods=['POST'])
 
     app.add_url_rule('/adm/tasks', 'adm_task_list_page', adm_task_list_page)
     app.add_url_rule('/adm/tasks/<int:task_id>', 'task_edit_page', task_edit_page)

ennead/models/thread.py

@@ -3,14 +3,15 @@
 import datetime
 from typing import List
 
-from peewee import DateTimeField, IntegerField, DecimalField, TextField, BooleanField, ForeignKeyField
+from peewee import DateTimeField, DecimalField, TextField, BooleanField, ForeignKeyField
 
 from ennead.models.user import User
 from ennead.models.task import Task
 from ennead.models.base import BaseModel
 
 from ennead.utils import render_markdown
 
+
 class Thread(BaseModel):
     """Student-with-teachers chat about `Task`
 
@@ -33,7 +34,6 @@ def ordered_posts(self, show_hidden=False):
             posts = filter(lambda post: not post.hide_from_student, posts)
         posts = sorted(posts, key=lambda post: post.date)
         return posts
-    
 
 
 class Post(BaseModel):

ennead/utils/__init__.py

@@ -9,7 +9,11 @@
 from ennead.utils.markdown import render_markdown
 
 
-__all__ = ['require_logged_in', 'require_not_logged_in', 'require_teacher', 'require_student', 'render_markdown']
+__all__ = [
+    'require_logged_in', 'require_not_logged_in',
+    'require_teacher', 'require_student',
+    'render_markdown',
+]
 
 
 def require_logged_in(func: Callable) -> Callable:
@@ -23,6 +27,7 @@ def wrapped(*args: Any, **kwargs: Any) -> Response:
 
     return wrapped
 
+
 def require_not_logged_in(func: Callable) -> Callable:
     """Make endpoint require NOT logged in user"""
 
@@ -47,6 +52,7 @@ def wrapped(*args: Any, **kwargs: Any) -> Response:
 
     return wrapped
 
+
 def require_student(func: Callable) -> Callable:
     """Make endpoint require logged in student"""
 

ennead/views/dialogue.py

@@ -3,27 +3,30 @@
 import datetime
 from flask import g
 
-from flask import session, current_app, render_template, request, redirect, url_for
+from flask import render_template, request, redirect, url_for
 from werkzeug.wrappers import Response
 
-from ennead.utils import require_logged_in, require_teacher, require_student
-from ennead.utils import render_markdown
-from ennead.models.user import User, UserGroup
+from ennead.utils import require_logged_in
+from ennead.models.user import User
 from ennead.models.task import Task
 from ennead.models.thread import Thread, Post
 
+
 def has_access_to_thread(student_id):
     return g.user.is_teacher or (g.user.is_student and (g.user.id == student_id))
 
+
 def get_thread(task_id, student_id):
     task = Task.get_by_id(task_id)
     student = User.get_by_id(student_id)
     thread, _ = Thread.get_or_create(task=task, student=student)
     return thread
 
+
 def correct_message(text):
     return (len(text) != 0)
 
+
 @require_logged_in
 def thread_page(task_id: int, student_id: int) -> Response:
     """GET /thread/{task}/{student}: show specified thread"""
@@ -32,7 +35,9 @@ def thread_page(task_id: int, student_id: int) -> Response:
         return redirect(url_for('index'))
     thread = get_thread(task_id, student_id)
     posts = thread.ordered_posts(show_hidden=g.user.is_teacher)
-    return render_template('dialogue.html', thread=thread, task=thread.task, student=thread.student, posts=posts)
+    return render_template('dialogue.html', thread=thread, task=thread.task,
+                           student=thread.student, posts=posts)
+
 
 @require_logged_in
 def post_to_thread(task_id: int, student_id: int) -> Response:
@@ -51,9 +56,9 @@ def post_to_thread(task_id: int, student_id: int) -> Response:
             if thread.score != score:
                 thread.update(score=score).execute()
 
-        post = Post.create(text=text, date=datetime.datetime.now(),
-                            author=g.user, thread=thread,
-                            hide_from_student=hide_from_student)
+        Post.create(text=text, date=datetime.datetime.now(),
+                    author=g.user, thread=thread,
+                    hide_from_student=hide_from_student)
     else:
         # TODO: redirect back to a dialogue, restoring a message draft not to lose it
         pass