Improve package page styling

[cheatdb.git] / app / blueprints / threads / __init__.py

diff --git a/app/blueprints/threads/__init__.py b/app/blueprints/threads/__init__.py
index 73d64c2dee2372dfe47e7b59e94798e1990073a4..8d6c22d650a00f8fe46611c3110742d5ec8ffff0 100644 (file)
--- a/app/blueprints/threads/__init__.py
+++ b/app/blueprints/threads/__init__.py
@@ -28,12 +28,19 @@ import datetime
 from flask_wtf import FlaskForm
 from wtforms import *
 from wtforms.validators import *
+from app.utils import get_int_or_abort
 
 @bp.route("/threads/")
 def list_all():
        query = Thread.query
        if not Permission.SEE_THREAD.check(current_user):
                query = query.filter_by(private=False)
+
+       pid = request.args.get("pid")
+       if pid:
+               pid = get_int_or_abort(pid)
+               query = query.filter_by(package_id=pid)
+
        return render_template("threads/list.html", threads=query.all())