book_library_server/fastapi_book_server/app/services/sequence.py

from app.models import Sequence

from app.services.common import TRGMSearchService


GET_OBJECT_IDS_QUERY = """
SELECT ARRAY (
    WITH filtered_sequences AS (
        SELECT 
            id,
            similarity(name, :query) as sml,
            (
                SELECT count(*) FROM book_sequences
                LEFT JOIN books ON books.id = book
                WHERE sequence = sequences.id AND books.is_deleted = 'f'
            ) as books_count
        FROM sequences
        WHERE name % :query AND
        EXISTS (
            SELECT * FROM book_sequences
            LEFT JOIN books ON books.id = book
            WHERE sequence = sequences.id AND books.is_deleted = 'f'
        )
    )
    SELECT fsequences.id FROM filtered_sequences as fsequences
    ORDER BY fsequences.sml DESC, fsequences.books_count DESC
    LIMIT 210
);
"""


class SequenceTGRMSearchService(TRGMSearchService):
    MODEL_CLASS = Sequence
    PREFETCH_RELATED = ["source"]
    GET_OBJECT_IDS_QUERY = GET_OBJECT_IDS_QUERY