1 from .models import db, PackageType, Package, ForumTopic, License, MinetestRelease, PackageRelease, User, Tag, Tags, ContentWarning
2 from .utils import isNo, isYes, get_int_or_abort
3 from sqlalchemy.sql.expression import func
4 from flask import abort
5 from sqlalchemy import or_
12 def __init__(self, args):
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]
20 title = ", ".join([type.value + "s" for type in types])
23 tags = args.getlist("tag")
24 tags = [Tag.query.filter_by(name=tname).first() for tname in tags]
25 tags = [tag for tag in tags if tag is not None]
28 hide_flags = args.getlist("hide")
35 self.random = "random" in args
36 self.lucky = "lucky" in args
37 self.limit = 1 if self.lucky else None
38 self.order_by = args.get("sort")
39 self.order_dir = args.get("order") or "desc"
41 self.hide_nonfree = "nonfree" in hide_flags
42 self.hide_flags = set(hide_flags)
43 self.hide_flags.discard("nonfree")
47 self.search = args.get("q")
48 self.minetest_version = args.get("engine_version")
49 self.protocol_version = get_int_or_abort(args.get("protocol_version"))
50 self.author = args.get("author")
52 self.show_discarded = isYes(args.get("show_discarded"))
53 self.show_added = args.get("show_added")
54 if self.show_added is not None:
55 self.show_added = isYes(self.show_added)
57 if self.search is not None and self.search.strip() == "":
60 def setSortIfNone(self, name, dir="desc"):
61 if self.order_by is None:
65 def getMinetestVersion(self):
66 if not self.protocol_version and not self.minetest_version:
69 return MinetestRelease.get(self.minetest_version, self.protocol_version)
71 def buildPackageQuery(self):
72 query = Package.query.filter_by(soft_deleted=False, approved=True)
74 if len(self.types) > 0:
75 query = query.filter(Package.type.in_(self.types))
78 query = query.search(self.search, sort=self.order_by is None)
81 query = query.order_by(func.random())
84 if self.order_by is None and self.search:
86 elif self.order_by is None or self.order_by == "score":
87 to_order = Package.score
88 elif self.order_by == "name":
89 to_order = Package.name
90 elif self.order_by == "title":
91 to_order = Package.title
92 elif self.order_by == "downloads":
93 to_order = Package.downloads
94 elif self.order_by == "created_at" or self.order_by == "date":
95 to_order = Package.created_at
100 if self.order_dir == "asc":
101 to_order = db.asc(to_order)
102 elif self.order_dir == "desc":
103 to_order = db.desc(to_order)
107 query = query.order_by(to_order)
110 author = User.query.filter_by(username=self.author).first()
114 query = query.filter_by(author=author)
116 for tag in self.tags:
117 query = query.filter(Package.tags.any(Tag.id == tag.id))
119 if "android_default" in self.hide_flags:
120 query = query.filter(~ Package.content_warnings.any())
122 for flag in self.hide_flags:
123 warning = ContentWarning.query.filter_by(name=flag).first()
125 query = query.filter(~ Package.content_warnings.any(ContentWarning.id == warning.id))
127 if self.hide_nonfree:
128 query = query.filter(Package.license.has(License.is_foss == True))
129 query = query.filter(Package.media_license.has(License.is_foss == True))
131 if self.protocol_version or self.minetest_version:
132 version = self.getMinetestVersion()
134 query = query.join(Package.releases) \
135 .filter(PackageRelease.approved==True) \
136 .filter(or_(PackageRelease.min_rel_id==None, PackageRelease.min_rel_id<=version.id)) \
137 .filter(or_(PackageRelease.max_rel_id==None, PackageRelease.max_rel_id>=version.id))
140 query = query.limit(self.limit)
144 def buildTopicQuery(self, show_added=False):
145 query = ForumTopic.query
147 if not self.show_discarded:
148 query = query.filter_by(discarded=False)
150 show_added = self.show_added == True or (self.show_added is None and show_added)
152 query = query.filter(~ db.exists().where(Package.forums==ForumTopic.topic_id))
154 if self.order_by is None or self.order_by == "name":
155 query = query.order_by(db.asc(ForumTopic.wip), db.asc(ForumTopic.name), db.asc(ForumTopic.title))
156 elif self.order_by == "views":
157 query = query.order_by(db.desc(ForumTopic.views))
158 elif self.order_by == "created_at" or self.order_by == "date":
159 query = query.order_by(db.asc(ForumTopic.created_at))
162 query = query.filter(ForumTopic.title.ilike('%' + self.search + '%'))
164 if len(self.types) > 0:
165 query = query.filter(ForumTopic.type.in_(self.types))
168 query = query.limit(self.limit)