app/querybuilder.py

   1 from .models import db, PackageType, Package, ForumTopic, License, MinetestRelease, PackageRelease
   2 from .utils import isNo
   3 from sqlalchemy.sql.expression import func
   4 from flask import abort
   5 from sqlalchemy import or_
   6
   7 class QueryBuilder:
   8         title  = None
   9         types  = None
  10         search = None
  11
  12         def __init__(self, args):
  13                 title = "Packages"
  14
  15                 # Get request types
  16                 types = args.getlist("type")
  17                 types = [PackageType.get(tname) for tname in types]
  18                 types = [type for type in types if type is not None]
  19                 if len(types) > 0:
  20                         title = ", ".join([type.value + "s" for type in types])
  21
  22                 self.title  = title
  23                 self.types  = types
  24                 self.search = args.get("q")
  25                 self.random = "random" in args
  26                 self.lucky  = self.random or "lucky" in args
  27                 self.hide_nonfree = isNo(args.get("nonfree"))
  28                 self.limit  = 1 if self.lucky else None
  29                 self.order_by  = args.get("sort") or "score"
  30                 self.order_dir = args.get("order") or "desc"
  31                 self.protocol_version = args.get("protocol_version")
  32
  33                 if self.search is not None and self.search.strip() == "":
  34                         self.search = None
  35
  36         def buildPackageQuery(self):
  37                 query = Package.query.filter_by(soft_deleted=False, approved=True)
  38
  39                 if len(self.types) > 0:
  40                         query = query.filter(Package.type.in_(self.types))
  41
  42                 if self.search:
  43                         query = query.filter(Package.title.ilike('%' + self.search + '%'))
  44
  45                 if self.random:
  46                         query = query.order_by(func.random())
  47                 else:
  48                         to_order = None
  49                         if self.order_by == "score":
  50                                 to_order = Package.score
  51                         elif self.order_by == "created_at":
  52                                 to_order = Package.created_at
  53                         else:
  54                                 abort(400)
  55
  56                         if self.order_dir == "asc":
  57                                 to_order = db.asc(to_order)
  58                         elif self.order_dir == "desc":
  59                                 to_order = db.desc(to_order)
  60                         else:
  61                                 abort(400)
  62
  63                         query = query.order_by(to_order)
  64
  65                 if self.hide_nonfree:
  66                         query = query.filter(Package.license.has(License.is_foss == True))
  67                         query = query.filter(Package.media_license.has(License.is_foss == True))
  68
  69                 if self.protocol_version:
  70                         self.protocol_version = int(self.protocol_version)
  71                         version = MinetestRelease.query.filter(MinetestRelease.protocol>=self.protocol_version).first()
  72                         if version is not None:
  73                                 version = version.id
  74                         else:
  75                                 version = 10000000
  76
  77                         query = query.join(Package.releases) \
  78                                 .filter(or_(PackageRelease.min_rel_id==None, PackageRelease.min_rel_id<=version)) \
  79                                 .filter(or_(PackageRelease.max_rel_id==None, PackageRelease.max_rel_id>=version))
  80
  81                 if self.limit:
  82                         query = query.limit(self.limit)
  83
  84                 return query
  85
  86         def buildTopicQuery(self):
  87                 topics = ForumTopic.query \
  88                                 .filter(~ db.exists().where(Package.forums==ForumTopic.topic_id)) \
  89                                 .order_by(db.asc(ForumTopic.wip), db.asc(ForumTopic.name), db.asc(ForumTopic.title))
  90
  91                 if self.search:
  92                         topics = topics.filter(ForumTopic.title.ilike('%' + self.search + '%'))
  93
  94                 if len(self.types) > 0:
  95                         topics = topics.filter(ForumTopic.type.in_(self.types))
  96
  97                 if self.limit:
  98                         topics = topics.limit(self.limit)
  99
 100                 return topics