feat: searching notes by text

codeforjapan · Aug 18, 2024 · dc3a216 · dc3a216
1 parent ea859de
commit dc3a216
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/api/birdxplorer_api/routers/data.py b/api/birdxplorer_api/routers/data.py
@@ -76,6 +76,7 @@ def get_notes(
         topic_ids: Union[List[TopicId], None] = Query(default=None),
         post_ids: Union[List[PostId], None] = Query(default=None),
         language: Union[LanguageIdentifier, None] = Query(default=None),
+        search_text: Union[None, str] = Query(default=None),
     ) -> NoteListResponse:
         return NoteListResponse(
             data=list(
@@ -86,6 +87,7 @@ def get_notes(
                     topic_ids=topic_ids,
                     post_ids=post_ids,
                     language=language,
+                    search_text=search_text,
                 )
             )
         )

diff --git a/common/birdxplorer_common/storage.py b/common/birdxplorer_common/storage.py
@@ -225,6 +225,7 @@ def get_notes(
         topic_ids: Union[List[TopicId], None] = None,
         post_ids: Union[List[PostId], None] = None,
         language: Union[LanguageIdentifier, None] = None,
+        search_text: Union[None, str] = None,
     ) -> Generator[NoteModel, None, None]:
         with Session(self.engine) as sess:
             query = sess.query(NoteRecord)
@@ -248,6 +249,8 @@ def get_notes(
                 query = query.filter(NoteRecord.post_id.in_(post_ids))
             if language is not None:
                 query = query.filter(NoteRecord.language == language)
+            if search_text is not None and len(search_text) > 0:
+                query = query.filter(NoteRecord.text.like(f"%{search_text}%"))
             for note_record in query.all():
                 yield NoteModel(
                     note_id=note_record.note_id,