app/querybuilder.py

   1 from .models import db, PackageType, Package, ForumTopic, License, MinetestRelease, PackageRelease, User, Tag
   2 from .models import tags as Tags
   3 from .utils import isNo, isYes, get_int_or_abort
   4 from sqlalchemy.sql.expression import func
   5 from flask import abort
   6 from sqlalchemy import or_
   7
   8 class QueryBuilder:
   9         title  = None
  10         types  = None
  11         search = None
  12
  13         def __init__(self, args):
  14                 title = "Packages"
  15
  16                 # Get request types
  17                 types = args.getlist("type")
  18                 types = [PackageType.get(tname) for tname in types]
  19                 types = [type for type in types if type is not None]
  20                 if len(types) > 0:
  21                         title = ", ".join([type.value + "s" for type in types])
  22
  23                 # Get tags types
  24                 tags = args.getlist("tag")
  25                 tags = [Tag.query.filter_by(name=tname).first() for tname in tags]
  26                 tags = [tag for tag in tags if tag is not None]
  27
  28                 # Hid
  29                 hide_flags = args.getlist("hide")
  30
  31                 self.title  = title
  32                 self.types  = types
  33                 self.tags   = tags
  34
  35                 self.random = "random" in args
  36                 self.lucky  = "lucky" in args
  37                 self.hide_nonfree = "nonfree" in hide_flags
  38                 self.limit  = 1 if self.lucky else None
  39                 self.order_by  = args.get("sort")
  40                 self.order_dir = args.get("order") or "desc"
  41
  42                 # Filters
  43
  44                 self.search = args.get("q")
  45                 self.minetest_version = args.get("engine_version")
  46                 self.protocol_version = get_int_or_abort(args.get("protocol_version"))
  47                 self.author = args.get("author")
  48
  49                 self.show_discarded = isYes(args.get("show_discarded"))
  50                 self.show_added = args.get("show_added")
  51                 if self.show_added is not None:
  52                         self.show_added = isYes(self.show_added)
  53
  54                 if self.search is not None and self.search.strip() == "":
  55                         self.search = None
  56
  57         def setSortIfNone(self, name):
  58                 if self.order_by is None:
  59                         self.order_by = name
  60
  61         def getMinetestVersion(self):
  62                 if not self.protocol_version and not self.minetest_version:
  63                         return None
  64
  65                 return MinetestRelease.get(self.minetest_version, self.protocol_version)
  66
  67         def buildPackageQuery(self):
  68                 query = Package.query.filter_by(soft_deleted=False, approved=True)
  69
  70                 if len(self.types) > 0:
  71                         query = query.filter(Package.type.in_(self.types))
  72
  73                 if self.search:
  74                         query = query.search(self.search, sort=True)
  75
  76                 if self.random:
  77                         query = query.order_by(func.random())
  78                 else:
  79                         to_order = None
  80                         if self.order_by is None or self.order_by == "score":
  81                                 to_order = Package.score
  82                         elif self.order_by == "name":
  83                                 to_order = Package.name
  84                         elif self.order_by == "title":
  85                                 to_order = Package.title
  86                         elif self.order_by == "downloads":
  87                                 to_order = Package.downloads
  88                         elif self.order_by == "created_at" or self.order_by == "date":
  89                                 to_order = Package.created_at
  90                         else:
  91                                 abort(400)
  92
  93                         if self.order_dir == "asc":
  94                                 to_order = db.asc(to_order)
  95                         elif self.order_dir == "desc":
  96                                 to_order = db.desc(to_order)
  97                         else:
  98                                 abort(400)
  99
 100                         query = query.order_by(to_order)
 101
 102                 if self.author:
 103                         author = User.query.filter_by(username=self.author).first()
 104                         if not author:
 105                                 abort(404)
 106
 107                         query = query.filter_by(author=author)
 108
 109                 for tag in self.tags:
 110                         query = query.filter(Package.tags.any(Tag.id == tag.id))
 111
 112                 if self.hide_nonfree:
 113                         query = query.filter(Package.license.has(License.is_foss == True))
 114                         query = query.filter(Package.media_license.has(License.is_foss == True))
 115
 116                 if self.protocol_version or self.minetest_version:
 117                         version = self.getMinetestVersion()
 118                         if version:
 119                                 query = query.join(Package.releases) \
 120                                         .filter(PackageRelease.approved==True) \
 121                                         .filter(or_(PackageRelease.min_rel_id==None, PackageRelease.min_rel_id<=version.id)) \
 122                                         .filter(or_(PackageRelease.max_rel_id==None, PackageRelease.max_rel_id>=version.id))
 123
 124                 if self.limit:
 125                         query = query.limit(self.limit)
 126
 127                 return query
 128
 129         def buildTopicQuery(self, show_added=False):
 130                 query = ForumTopic.query
 131
 132                 if not self.show_discarded:
 133                         query = query.filter_by(discarded=False)
 134
 135                 show_added = self.show_added == True or (self.show_added is None and show_added)
 136                 if not show_added:
 137                         query = query.filter(~ db.exists().where(Package.forums==ForumTopic.topic_id))
 138
 139                 if self.order_by is None or self.order_by == "name":
 140                         query = query.order_by(db.asc(ForumTopic.wip), db.asc(ForumTopic.name), db.asc(ForumTopic.title))
 141                 elif self.order_by == "views":
 142                         query = query.order_by(db.desc(ForumTopic.views))
 143                 elif self.order_by == "created_at" or self.order_by == "date":
 144                         query = query.order_by(db.asc(ForumTopic.created_at))
 145
 146                 if self.search:
 147                         query = query.filter(ForumTopic.title.ilike('%' + self.search + '%'))
 148
 149                 if len(self.types) > 0:
 150                         query = query.filter(ForumTopic.type.in_(self.types))
 151
 152                 if self.limit:
 153                         query = query.limit(self.limit)
 154
 155                 return query