ulysia/szurubooru
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Impl post_search_config.py's create_around_query()

Browse source
This commit is contained in:
Rebecca Nelson 2024-04-13 23:50:56 -05:00
parent 6a0d5741c3
commit 719f6dc741
1 changed files with 14 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
server/szurubooru/search/configs/post_search_config.py
View file

@ -220,8 +220,20 @@ class PostSearchConfig(BaseSearchConfig):
if token.name == "pool" and isinstance(token.criterion, criteria.PlainCriterion):
self.pool_id = token.criterion.value
def create_around_query(self) -> SaQuery:
return db.session.query(model.Post).options(sa.orm.lazyload("*"))
def create_around_query(self, filter_query: SaQuery, entity_id: int) -> SaQuery:
prev_filter_query = (
filter_query.filter(self.config.id_column > entity_id)
.order_by(None)
.order_by(sa.func.abs(self.config.id_column - entity_id).asc())
.limit(1)
)
next_filter_query = (
filter_query.filter(self.config.id_column < entity_id)
.order_by(None)
.order_by(sa.func.abs(self.config.id_column - entity_id).asc())
.limit(1)
)
return prev_filter_query, next_filter_query
def create_around_filter_queries(self, filter_query: SaQuery, entity_id: int) -> Tuple[SaQuery, SaQuery]:
if self.pool_id is not None: