Add support for multiple types in packages list

1 files changed, 11 insertions, 9 deletions

diff --git a/app/views/packages/__init__.py b/app/views/packages/__init__.py
index 8e0410f..f227b19 100644
--- a/app/views/packages/__init__.py
+++ b/app/views/packages/__init__.py
@@ -29,20 +29,22 @@ from flask_wtf import FlaskForm
 from wtforms import *
 from wtforms.validators import *
 from wtforms.ext.sqlalchemy.fields import QuerySelectField, QuerySelectMultipleField
-from sqlalchemy import or_
+from sqlalchemy import or_, any_
 
 def build_packages_query():
-	type_name = request.args.get("type")
-	type = None
-	if type_name is not None:
-		type = PackageType[type_name.upper()]
-
 	title = "Packages"
+
 	query = Package.query.filter_by(soft_deleted=False, approved=True)
 
-	if type is not None:
-		title = type.value + "s"
-		query = query.filter_by(type=type)
+	# Filter by requested type(s)
+	types = request.args.getlist("type")
+	types = [PackageType.get(tname) for tname in types]
+	types = [type for type in types if type is not None]
+	if len(types) > 0:
+		title = ", ".join([type.value + "s" for type in types])
+
+		query = query.filter(Package.type.in_(types))
+
 
 	search = request.args.get("q")
 	if search is not None and search.strip() != "":