mirror of
https://github.com/minetest/contentdb.git
synced 2024-11-10 09:33:44 +01:00
281 lines
8.8 KiB
Python
281 lines
8.8 KiB
Python
# ContentDB
|
|
# Copyright (C) rubenwardy
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
from flask import abort, current_app
|
|
from flask_babel import lazy_gettext
|
|
from sqlalchemy import or_
|
|
from sqlalchemy.orm import subqueryload
|
|
from sqlalchemy.sql.expression import func
|
|
from sqlalchemy_searchable import search
|
|
|
|
from .models import db, PackageType, Package, ForumTopic, License, MinetestRelease, PackageRelease, User, Tag, \
|
|
ContentWarning, PackageState, PackageDevState
|
|
from .utils import is_yes, get_int_or_abort
|
|
|
|
|
|
class QueryBuilder:
|
|
types = None
|
|
search = None
|
|
|
|
@property
|
|
def title(self):
|
|
if len(self.types) == 1:
|
|
package_type = str(self.types[0].plural)
|
|
else:
|
|
package_type = lazy_gettext("Packages")
|
|
|
|
if len(self.tags) == 0:
|
|
ret = package_type
|
|
elif len(self.tags) == 1:
|
|
ret = self.tags[0].title + " " + package_type
|
|
else:
|
|
tags = ", ".join([tag.title for tag in self.tags])
|
|
ret = f"{tags} - {package_type}"
|
|
|
|
if self.search:
|
|
ret = f"{self.search} - {ret}"
|
|
|
|
return ret
|
|
|
|
@property
|
|
def noindex(self):
|
|
return (self.search is not None or len(self.tags) > 1 or len(self.types) > 1 or len(self.hide_flags) > 0 or
|
|
self.random or self.lucky or self.author or self.version or self.game)
|
|
|
|
def __init__(self, args):
|
|
|
|
# Get request types
|
|
types = args.getlist("type")
|
|
types = [PackageType.get(tname) for tname in types]
|
|
types = [type for type in types if type is not None]
|
|
|
|
# Get tags types
|
|
tags = args.getlist("tag")
|
|
tags = [Tag.query.filter_by(name=tname).first() for tname in tags]
|
|
tags = [tag for tag in tags if tag is not None]
|
|
|
|
# Hide
|
|
self.hide_flags = set(args.getlist("hide"))
|
|
|
|
self.types = types
|
|
self.tags = tags
|
|
|
|
self.random = "random" in args
|
|
self.lucky = "lucky" in args
|
|
self.limit = 1 if self.lucky else get_int_or_abort(args.get("limit"), None)
|
|
self.order_by = args.get("sort")
|
|
self.order_dir = args.get("order") or "desc"
|
|
|
|
if "android_default" in self.hide_flags:
|
|
self.hide_flags.update(["*", "deprecated"])
|
|
self.hide_flags.discard("android_default")
|
|
|
|
if "desktop_default" in self.hide_flags:
|
|
self.hide_flags.update(["deprecated"])
|
|
self.hide_flags.discard("desktop_default")
|
|
|
|
self.hide_nonfree = "nonfree" in self.hide_flags
|
|
self.hide_wip = "wip" in self.hide_flags
|
|
self.hide_deprecated = "deprecated" in self.hide_flags
|
|
self.hide_flags.discard("nonfree")
|
|
self.hide_flags.discard("wip")
|
|
self.hide_flags.discard("deprecated")
|
|
|
|
# Filters
|
|
self.search = args.get("q")
|
|
self.author = args.get("author")
|
|
|
|
protocol_version = get_int_or_abort(args.get("protocol_version"))
|
|
minetest_version = args.get("engine_version")
|
|
if protocol_version or minetest_version:
|
|
self.version = MinetestRelease.get(minetest_version, protocol_version)
|
|
else:
|
|
self.version = None
|
|
|
|
self.show_discarded = is_yes(args.get("show_discarded"))
|
|
self.show_added = args.get("show_added")
|
|
if self.show_added is not None:
|
|
self.show_added = is_yes(self.show_added)
|
|
|
|
if self.search is not None and self.search.strip() == "":
|
|
self.search = None
|
|
|
|
self.game = args.get("game")
|
|
if self.game:
|
|
self.game = Package.get_by_key(self.game)
|
|
|
|
def set_sort_if_none(self, name, dir="desc"):
|
|
if self.order_by is None:
|
|
self.order_by = name
|
|
self.order_dir = dir
|
|
|
|
def get_releases(self):
|
|
releases_query = db.session.query(PackageRelease.package_id, func.max(PackageRelease.id)) \
|
|
.select_from(PackageRelease).filter(PackageRelease.approved) \
|
|
.group_by(PackageRelease.package_id)
|
|
|
|
if self.version:
|
|
releases_query = releases_query \
|
|
.filter(or_(PackageRelease.min_rel_id==None,
|
|
PackageRelease.min_rel_id <= self.version.id)) \
|
|
.filter(or_(PackageRelease.max_rel_id==None,
|
|
PackageRelease.max_rel_id >= self.version.id))
|
|
|
|
return releases_query.all()
|
|
|
|
def convert_to_dictionary(self, packages):
|
|
releases = {}
|
|
for [package_id, release_id] in self.get_releases():
|
|
releases[package_id] = release_id
|
|
|
|
def to_json(package: Package):
|
|
release_id = releases.get(package.id)
|
|
return package.as_short_dict(current_app.config["BASE_URL"], release_id=release_id, no_load=True)
|
|
|
|
return [to_json(pkg) for pkg in packages]
|
|
|
|
def build_package_query(self):
|
|
if self.order_by == "last_release":
|
|
query = db.session.query(Package).select_from(PackageRelease).join(Package) \
|
|
.filter_by(state=PackageState.APPROVED)
|
|
else:
|
|
query = Package.query.filter_by(state=PackageState.APPROVED)
|
|
|
|
query = query.options(subqueryload(Package.main_screenshot), subqueryload(Package.aliases))
|
|
|
|
query = self.order_package_query(self.filter_package_query(query))
|
|
|
|
if self.limit:
|
|
query = query.limit(self.limit)
|
|
|
|
return query
|
|
|
|
def filter_package_query(self, query):
|
|
if len(self.types) > 0:
|
|
query = query.filter(Package.type.in_(self.types))
|
|
|
|
if self.author:
|
|
author = User.query.filter_by(username=self.author).first()
|
|
if not author:
|
|
abort(404)
|
|
|
|
query = query.filter_by(author=author)
|
|
|
|
if self.game:
|
|
query = query.filter(Package.supported_games.any(game=self.game))
|
|
|
|
for tag in self.tags:
|
|
query = query.filter(Package.tags.any(Tag.id == tag.id))
|
|
|
|
if "*" in self.hide_flags:
|
|
query = query.filter(~ Package.content_warnings.any())
|
|
else:
|
|
for flag in self.hide_flags:
|
|
warning = ContentWarning.query.filter_by(name=flag).first()
|
|
if warning:
|
|
query = query.filter(~ Package.content_warnings.any(ContentWarning.id == warning.id))
|
|
|
|
if self.hide_nonfree:
|
|
query = query.filter(Package.license.has(License.is_foss == True))
|
|
query = query.filter(Package.media_license.has(License.is_foss == True))
|
|
|
|
if self.hide_wip:
|
|
query = query.filter(or_(Package.dev_state==None, Package.dev_state != PackageDevState.WIP))
|
|
if self.hide_deprecated:
|
|
query = query.filter(or_(Package.dev_state==None, Package.dev_state != PackageDevState.DEPRECATED))
|
|
|
|
if self.version:
|
|
query = query.join(Package.releases) \
|
|
.filter(PackageRelease.approved == True) \
|
|
.filter(or_(PackageRelease.min_rel_id==None,
|
|
PackageRelease.min_rel_id <= self.version.id)) \
|
|
.filter(or_(PackageRelease.max_rel_id==None,
|
|
PackageRelease.max_rel_id >= self.version.id))
|
|
|
|
return query
|
|
|
|
def order_package_query(self, query):
|
|
if self.search:
|
|
query = search(query, self.search, sort=self.order_by is None)
|
|
|
|
if self.random:
|
|
query = query.order_by(func.random())
|
|
return query
|
|
|
|
to_order = None
|
|
if self.order_by is None and self.search:
|
|
pass
|
|
elif self.order_by is None or self.order_by == "score":
|
|
to_order = Package.score
|
|
elif self.order_by == "reviews":
|
|
query = query.filter(Package.reviews.any())
|
|
to_order = (Package.score - Package.score_downloads)
|
|
elif self.order_by == "name":
|
|
to_order = Package.name
|
|
elif self.order_by == "title":
|
|
to_order = Package.title
|
|
elif self.order_by == "downloads":
|
|
to_order = Package.downloads
|
|
elif self.order_by == "created_at" or self.order_by == "date":
|
|
to_order = Package.created_at
|
|
elif self.order_by == "approved_at" or self.order_by == "date":
|
|
to_order = Package.approved_at
|
|
elif self.order_by == "last_release":
|
|
to_order = PackageRelease.releaseDate
|
|
else:
|
|
abort(400)
|
|
|
|
if to_order is not None:
|
|
if self.order_dir == "asc":
|
|
to_order = db.asc(to_order)
|
|
elif self.order_dir == "desc":
|
|
to_order = db.desc(to_order)
|
|
else:
|
|
abort(400)
|
|
|
|
query = query.order_by(to_order)
|
|
|
|
return query
|
|
|
|
def build_topic_query(self, show_added=False):
|
|
query = ForumTopic.query
|
|
|
|
if not self.show_discarded:
|
|
query = query.filter_by(discarded=False)
|
|
|
|
show_added = self.show_added == True or (self.show_added is None and show_added)
|
|
if not show_added:
|
|
query = query.filter(~ db.exists().where(Package.forums==ForumTopic.topic_id))
|
|
|
|
if self.order_by is None or self.order_by == "name":
|
|
query = query.order_by(db.asc(ForumTopic.wip), db.asc(ForumTopic.name), db.asc(ForumTopic.title))
|
|
elif self.order_by == "views":
|
|
query = query.order_by(db.desc(ForumTopic.views))
|
|
elif self.order_by == "created_at" or self.order_by == "date":
|
|
query = query.order_by(db.asc(ForumTopic.created_at))
|
|
|
|
if self.search:
|
|
query = query.filter(or_(ForumTopic.title.ilike('%' + self.search + '%'),
|
|
ForumTopic.name == self.search.lower()))
|
|
|
|
if len(self.types) > 0:
|
|
query = query.filter(ForumTopic.type.in_(self.types))
|
|
|
|
if self.limit:
|
|
query = query.limit(self.limit)
|
|
|
|
return query
|