Allow moderators & admin to edit & remove any post

db/sqlite.py

@@ -140,7 +140,7 @@ class DB:
 
     def get_user_private_info(self, user_id):
         return self._db().execute('''
-            select name, about
+            select about
             from users
             where user_id = ?
             ''',
@@ -158,6 +158,15 @@ class DB:
         )
         db.commit()
 
+    def get_user_name_role(self, user_id):
+        return self._db().execute('''
+            select name, role
+            from users
+            where user_id = ?
+            ''',
+            (user_id,)
+        ).fetchone()
+
     def get_user_name(self, user_id):
         return self._db().execute('''
             select name
@@ -193,9 +202,13 @@ class DB:
         c.execute('''
             delete
             from threads
-            where thread_id = ? and author_id = ?
+            -- 1 = moderator, 2 = admin
+            where thread_id = ? and (
+              author_id = ?
+              or (select 1 from users where user_id = ? and (role = 1 or role = 2))
+            )
             ''',
-            (thread_id, user_id)
+            (thread_id, user_id, user_id)
         )
         db.commit()
         return c.rowcount > 0
@@ -206,9 +219,16 @@ class DB:
         c.execute('''
             delete
             from comments
-            where comment_id = ? and author_id = ?
+            where comment_id = ?
+              and (
+                author_id = ?
+                -- 1 = moderator, 2 = admin
+                or (select 1 from users where user_id = ? and (role = 1 or role = 2))
+              )
+              -- Don't allow deleting comments with children
+              and (select 1 from comments where parent_id = ?) is null
             ''',
-            (comment_id, user_id)
+            (comment_id, user_id, user_id, comment_id)
         )
         db.commit()
         return c.rowcount > 0
@@ -270,9 +290,13 @@ class DB:
         c.execute('''
             update threads
             set title = ?, text = ?, modify_time = ?
-            where thread_id = ? and author_id = ?
+            where thread_id = ? and (
+              author_id = ?
+              -- 1 = moderator, 2 = admin
+              or (select 1 from users where user_id = ? and (role = 1 or role = 2))
+            )
             ''',
-            (title, text, time, thread_id, user_id)
+            (title, text, time, thread_id, user_id, user_id)
         )
         if c.rowcount > 0:
             db.commit()
@@ -285,9 +309,13 @@ class DB:
         c.execute('''
             update comments
             set text = ?, modify_time = ?
-            where comment_id = ? and author_id = ?
+            where comment_id = ? and (
+              author_id = ?
+              -- 1 = moderator, 2 = admin
+              or (select 1 from users where user_id = ? and (role = 1 or role = 2))
+            )
             ''',
-            (text, time, comment_id, user_id)
+            (text, time, comment_id, user_id, user_id)
         )
         if c.rowcount > 0:
             db.commit()

main.py

@@ -16,9 +16,19 @@ captcha_key = 'piss off bots'
 app.jinja_env.trim_blocks = True
 app.jinja_env.lstrip_blocks = True
 
+class Role:
+    USER = 0
+    MODERATOR = 1
+    ADMIN = 2
+
 @app.route('/')
 def index():
-    return render_template('index.html', title = NAME, forums = db.get_forums())
+    return render_template(
+        'index.html',
+        title = NAME,
+        user = get_user(),
+        forums = db.get_forums()
+    )
 
 @app.route('/forum/<int:forum_id>/')
 def forum(forum_id):
@@ -27,6 +37,7 @@ def forum(forum_id):
     return render_template(
         'forum.html',
         title = title,
+        user = get_user(),
         forum_id = forum_id,
         description = description,
         threads = threads,
@@ -40,6 +51,7 @@ def thread(thread_id):
     return render_template(
         'thread.html',
         title = title,
+        user = get_user(),
         text = text,
         author = author,
         author_id = author_id,
@@ -51,7 +63,6 @@ def thread(thread_id):
 
 @app.route('/comment/<int:comment_id>/')
 def comment(comment_id):
-    user_id = session.get('user_id')
     thread_id, parent_id, title, comments = db.get_subcomments(comment_id)
     comments = create_comment_tree(comments)
     reply_comment, = comments
@@ -60,6 +71,7 @@ def comment(comment_id):
     return render_template(
         'comments.html',
         title = title,
+        user = get_user(),
         reply_comment = reply_comment,
         comments = comments,
         parent_id = parent_id,
@@ -80,8 +92,11 @@ def login():
             # Sleep to reduce effectiveness of bruteforce
             time.sleep(0.1)
         flash('Username or password is invalid', 'error')
-        return render_template('login.html', title = "Login")
+    return render_template(
-    return render_template('login.html', title = "Login")
+        'login.html',
+        title = 'Login',
+        user = get_user()
+    )
 
 @app.route('/logout/')
 def logout():
@@ -90,22 +105,21 @@ def logout():
 
 @app.route('/user/', methods = ['GET', 'POST'])
 def user_edit():
-    user_id = session.get('user_id')
+    user = get_user()
-    if user_id is None:
+    if user is None:
         return redirect(url_for('login'))
 
     if request.method == 'POST':
         about = request.form['about'].replace('\r', '')
-        db.set_user_private_info(user_id, about)
+        db.set_user_private_info(user.id, about)
-        name, = db.get_user_name(user_id)
         flash('Updated profile', 'success')
     else:
-        name, about = db.get_user_private_info(user_id)
+        about, = db.get_user_private_info(user.id)
 
     return render_template(
         'user_edit.html',
-        name = name,
         title = 'Edit profile',
+        user = user,
         about = about
     )
 
@@ -115,6 +129,7 @@ def user_info(user_id):
     return render_template(
         'user_info.html',
         title = 'Profile',
+        user = get_user(),
         name = name,
         about = about
     )
@@ -133,6 +148,7 @@ def new_thread(forum_id):
     return render_template(
         'new_thread.html',
         title = 'Create new thread',
+        user = get_user(),
     )
 
 @app.route('/thread/<int:thread_id>/confirm_delete/')
@@ -141,6 +157,7 @@ def confirm_delete_thread(thread_id):
     return render_template(
         'confirm_delete_thread.html',
         title = 'Delete thread',
+        user = get_user(),
         thread_title = title,
     )
 
@@ -187,6 +204,7 @@ def confirm_delete_comment(comment_id):
     return render_template(
         'confirm_delete_comment.html',
         title = 'Delete comment',
+        user = get_user(),
         thread_title = title,
         text = text,
     )
@@ -228,6 +246,7 @@ def edit_thread(thread_id):
     return render_template(
         'edit_thread.html',
         title = 'Edit thread',
+        user = get_user(),
         thread_title = title,
         text = text,
     )
@@ -255,6 +274,7 @@ def edit_comment(comment_id):
     return render_template(
         'edit_comment.html',
         title = 'Edit comment',
+        user = get_user(),
         thread_title = title,
         text = text,
     )
@@ -283,6 +303,7 @@ def register():
     return render_template(
         'register.html',
         title = 'Register',
+        user = get_user(),
         captcha = capt,
         answer = answer,
     )
@@ -300,6 +321,8 @@ class Comment:
         self.parent_id = parent_id
 
 def create_comment_tree(comments):
+    comments = [*comments]
+    print(comments)
     start = time.time();
     # Collect comments first, then build the tree in case we encounter a child before a parent
     comment_map = {
@@ -326,6 +349,26 @@ def create_comment_tree(comments):
     return root
 
 
+class User:
+    def __init__(self, id, name, role):
+        self.id = id
+        self.name = name
+        self.role = role
+
+    def is_moderator(self):
+        return self.role in (Role.ADMIN, Role.MODERATOR)
+
+    def is_admin(self):
+        return self.role == Role.ADMIN
+
+def get_user():
+    id = session.get('user_id')
+    if id is not None:
+        name, role = db.get_user_name_role(id)
+        return User(id, name, role)
+    return None
+
+
 @app.context_processor
 def utility_processor():
     def format_since(t):

templates/base.html

@@ -9,8 +9,8 @@
 	<nav>
 		<a class=logo href="{{ url_for('index') }}">A</a>
 		<div style="margin:auto"></div>
-		{% if 'user_id' in session %}
+		{% if user is not none %}
-		<a href="{{ url_for('user_edit') }}">User panel</a>
+		<a href="{{ url_for('user_edit') }}">{{ user.name }}</a>
 		<span>|</span>
 		<a href="{{ url_for('logout') }}">Logout</a>
 		{% else %}

templates/comment.html

@@ -2,7 +2,7 @@
 <i><a href="{{ url_for('user_info', user_id = id) }}">{{ name }}</a> - {{ format_since(ctime) }}{% if ctime != mtime %} (last modified {{ format_since(mtime) }}){% endif %}</i>
 {%- endmacro -%}
 
-{%- macro comment_author(comment, thread_id) -%}
+{%- macro comment_author(comment, thread_id, can_delete) -%}
 <p><sub>
 	{{- author(comment.author_id, comment.author, comment.create_time, comment.modify_time) }} |
 	{# Suffixing a # prevents unnecessary reloads #}
@@ -10,30 +10,28 @@
 	{%- if comment.parent_id is not none -%}
 	<a href="{{ url_for('comment', comment_id = comment.parent_id) }}#"> parent</a>
 	{%- endif -%}
-	{%- if comment.author_id == session.get('user_id') -%}
+	{%- if user is not none and (comment.author_id == user.id or user.is_moderator()) -%}
 	<a href="{{ url_for('edit_comment', comment_id = comment.id) }}"> edit</a>
-	{%- endif -%}
+	{%- if can_delete -%}
-	{%- if comment.author_id == session.get('user_id') -%}
 	<a href="{{ url_for('confirm_delete_comment', comment_id = comment.id) }}"> delete</a>
 	{%- endif -%}
+	{%- endif -%}
 </sub></p>
 {%- endmacro -%}
 
 {%- macro thread_author(author_id, name, ctime, mtime) -%}
 <p><sub>
 	{{- author(author_id, name, ctime, mtime) -}}
-	{%- if author_id == session.get('user_id') -%}
+	{%- if user is not none and (author_id == user.id or user.is_moderator()) -%}
 	<a href="{{ url_for('edit_thread', thread_id = thread_id) }}"> edit</a>
-	{%- endif -%}
-	{%- if author_id == session.get('user_id') -%}
 	<a href="{{ url_for('confirm_delete_thread', thread_id = thread_id) }}"> delete</a>
 	{%- endif -%}
 </sub></p>
 {%- endmacro -%}
 
-{%- macro render_comment_pre(comment, thread_id) -%}
+{%- macro render_comment_pre(comment, thread_id, can_delete) -%}
 <div class=comment>
-	{{- comment_author(comment, thread_id) -}}
+	{{- comment_author(comment, thread_id, can_delete) -}}
 	<p>{{- minimd(comment.text) | safe -}}</p>
 {%- endmacro -%}
 
@@ -45,13 +43,13 @@
 {%- endmacro -%}
 
 {%- macro render_comment(comment, thread_id) -%}
-{{- render_comment_pre(comment, thread_id) -}}
+{{- render_comment_pre(comment, thread_id, comment.children | length == 0) -}}
 <sup><a href="{{ url_for("comment", comment_id = comment.id) }}">reply</a></sup>
 {{- render_comment_post(comment, thread_id) -}}
 {%- endmacro -%}
 
 {%- macro reply() -%}
-{%- if 'user_id' in session -%}
+{%- if user is not none -%}
 <form method="post" action="comment/">
 	<p><textarea name="text"></textarea></p>
 	<p><input type="submit" value="Post comment"></p>

templates/comments.html

@@ -3,7 +3,7 @@
 
 {% block content %}
 
-{{ render_comment_pre(reply_comment, thread_id) }}
+{{ render_comment_pre(reply_comment, thread_id, comments | length == 0) }}
 
 {{ reply() }}
 

templates/user_edit.html

@@ -1,11 +1,11 @@
 {% extends 'base.html' %}
 
 {% block content %}
-<p><a href="{{ url_for('user_info', user_id = session['user_id']) }}">View public profile</a></p>
+<p><a href="{{ url_for('user_info', user_id = user.id) }}">View public profile</a></p>
 <form method="post">
 <table>
-<tr><td>Username</td><td>{{ name }}</td></tr>
+<tr><td>Username</td><td>{{ user.name }}</td></tr>
-<tr><td>ID</td><td>{{ session['user_id'] }}</td></tr>
+<tr><td>ID</td><td>{{ user.id }}</td></tr>
 <tr><td>About</td><td><textarea name="about">{{ about }}</textarea></td></tr>
 </form>
 </table>