From f9a60cacfe7d4f6dcca429ad0111f5dfdbc7e8f1 Mon Sep 17 00:00:00 2001
From: Simon Will <will@cl.uni-heidelberg.de>
Date: Thu, 8 Apr 2021 12:39:34 +0200
Subject: [PATCH] Make it possible to query for nl substrings
---
joeynmt_server/views/feedback.py | 15 +++++++++++----
1 file changed, 11 insertions(+), 4 deletions(-)
diff --git a/joeynmt_server/views/feedback.py b/joeynmt_server/views/feedback.py
index 06e7a40..93189e3 100644
--- a/joeynmt_server/views/feedback.py
+++ b/joeynmt_server/views/feedback.py
@@ -76,12 +76,19 @@ def list_feedback():
def query_feedback():
filters = request.json
fb_query = Feedback.query
+ count_query = Feedback.query
+
if 'user_id' in filters:
fb_query = fb_query.filter_by(user_id=filters['user_id'])
- total_count = Feedback.query.filter_by(
- user_id=filters['user_id']).count()
- else:
- total_count = Feedback.query.count()
+ count_query = count_query.filter_by(
+ user_id=filters['user_id'])
+
+ if 'nl_part' in filters:
+ like_arg = '%{}%'.format(filters['nl_part'])
+ fb_query = fb_query.filter(Feedback.nl.like(like_arg))
+ count_query = count_query.filter(Feedback.nl.like(like_arg))
+
+ total_count = count_query.count()
if 'limit' in filters or 'offset' in filters:
fb_query = fb_query.order_by(Feedback.id)
--
GitLab