blob: 0cdb1ea27e72a431bb36ce7186bf9262964f1ce2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
from typing import Dict
import sqlalchemy as sa
from szurubooru import db, model
from szurubooru.func import metrics, util
from szurubooru.search.configs import util as search_util
from szurubooru.search.configs.base_search_config import (
BaseSearchConfig, Filter)
from szurubooru.search.typing import SaQuery
class PostMetricSearchConfig(BaseSearchConfig):
def __init__(self) -> None:
self.all_metric_names = []
def refresh_metrics(self) -> None:
self.all_metric_names = metrics.get_all_metric_tag_names()
def create_filter_query(self, _disable_eager_loads: bool) -> SaQuery:
self.refresh_metrics()
return db.session.query(model.PostMetric).options(sa.orm.lazyload('*'))
def create_count_query(self, disable_eager_loads: bool) -> SaQuery:
return self.create_filter_query(disable_eager_loads)
def create_around_query(self) -> SaQuery:
return self.create_filter_query()
def finalize_query(self, query: SaQuery) -> SaQuery:
return query.order_by(model.PostMetric.value.asc())
@property
def anonymous_filter(self) -> Filter:
return search_util.create_subquery_filter(
model.PostMetric.tag_id,
model.TagName.tag_id,
model.TagName.name,
search_util.create_str_filter)
@property
def named_filters(self) -> Dict[str, Filter]:
num_filter = search_util.create_float_filter(model.PostMetric.value)
return {tag_name: num_filter for tag_name in self.all_metric_names}
|
