diff options
Diffstat (limited to 'uploader/publications/models.py')
-rw-r--r-- | uploader/publications/models.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/uploader/publications/models.py b/uploader/publications/models.py index 992ebfa..f4e8d33 100644 --- a/uploader/publications/models.py +++ b/uploader/publications/models.py @@ -71,16 +71,23 @@ def update_publications(conn: Connection , publications: tuple[dict, ...]) -> tu def fetch_publications( conn: Connection, - search: Optional[str] = None + search: Optional[str] = None, + offset: int = 0, + limit: int = -1 ) -> Iterable[dict]: """Fetch publications from the database.""" _query = "SELECT * FROM Publication" _params = None if search is not None and bool(search): _query = (f"{_query} " - "WHERE CONCAT(PubMed_ID, ' ', Authors, ' ', Title) " - "LIKE %s") - _params = (f"%{search}%",) + "WHERE PubMed_ID LIKE %s " + "OR Authors LIKE %s " + "OR Title LIKE %s") + _params = (f"%{search}%",) * 3 + + if limit > 0: + _query = (f"{_query} LIMIT {limit} OFFSET {offset}") + with conn.cursor(cursorclass=DictCursor) as cursor: cursor.execute(_query, _params) debug_query(_query, logger) |