# along with this program. If not, see <https://www.gnu.org/licenses/>.
+import enum, datetime
+
+from app import app, gravatar
+from urllib.parse import urlparse
+
from flask import Flask, url_for
-from flask_sqlalchemy import SQLAlchemy
+from flask_sqlalchemy import SQLAlchemy, BaseQuery
from flask_migrate import Migrate
-from urllib.parse import urlparse
-from app import app, gravatar
-from sqlalchemy.orm import validates
from flask_user import login_required, UserManager, UserMixin, SQLAlchemyAdapter
-import enum, datetime
+from sqlalchemy.orm import validates
+from sqlalchemy_searchable import SearchQueryMixin
+from sqlalchemy_utils.types import TSVectorType
+from sqlalchemy_searchable import make_searchable
+
# Initialise database
db = SQLAlchemy(app)
migrate = Migrate(app, db)
+make_searchable(db.metadata)
+
+
+class ArticleQuery(BaseQuery, SearchQueryMixin):
+ pass
class UserRank(enum.Enum):
class PackagePropertyKey(enum.Enum):
name = "Name"
title = "Title"
- shortDesc = "Short Description"
+ short_desc = "Short Description"
desc = "Description"
type = "Type"
license = "License"
return retval
-
class Package(db.Model):
+ query_class = ArticleQuery
+
id = db.Column(db.Integer, primary_key=True)
# Basic details
author_id = db.Column(db.Integer, db.ForeignKey("user.id"))
name = db.Column(db.String(100), nullable=False)
- title = db.Column(db.String(100), nullable=False)
- shortDesc = db.Column(db.String(200), nullable=False)
- desc = db.Column(db.Text, nullable=True)
+ title = db.Column(db.Unicode(100), nullable=False)
+ short_desc = db.Column(db.Unicode(200), nullable=False)
+ desc = db.Column(db.UnicodeText, nullable=True)
type = db.Column(db.Enum(PackageType))
created_at = db.Column(db.DateTime, nullable=False, default=datetime.datetime.utcnow)
+ search_vector = db.Column(TSVectorType("title", "short_desc", "desc"))
+
license_id = db.Column(db.Integer, db.ForeignKey("license.id"), nullable=False, default=1)
license = db.relationship("License", foreign_keys=[license_id])
media_license_id = db.Column(db.Integer, db.ForeignKey("license.id"), nullable=False, default=1)
for e in PackagePropertyKey:
setattr(self, e.name, getattr(package, e.name))
- def getAsDictionaryShort(self, base_url, protonum=None):
+ def getState(self):
+ if self.approved:
+ return "approved"
+ elif self.review_thread_id:
+ return "thread"
+ elif (self.type == PackageType.GAME or \
+ self.type == PackageType.TXP) and \
+ self.screenshots.count() == 0:
+ return "wip"
+ elif not self.getDownloadRelease():
+ return "wip"
+ elif "Other" in self.license.name or "Other" in self.media_license.name:
+ return "license"
+ else:
+ return "ready"
+
+ def getAsDictionaryShort(self, base_url, version=None, protonum=None):
tnurl = self.getThumbnailURL(1)
+ release = self.getDownloadRelease(version=version, protonum=protonum)
return {
"name": self.name,
"title": self.title,
"author": self.author.display_name,
- "short_description": self.shortDesc,
+ "short_description": self.short_desc,
"type": self.type.toName(),
- "release": self.getDownloadRelease(protonum).id if self.getDownloadRelease(protonum) is not None else None,
+ "release": release and release.id,
"thumbnail": (base_url + tnurl) if tnurl is not None else None,
"score": round(self.score * 10) / 10
}
- def getAsDictionary(self, base_url, protonum=None):
+ def getAsDictionary(self, base_url, version=None, protonum=None):
tnurl = self.getThumbnailURL(1)
+ release = self.getDownloadRelease(version=version, protonum=protonum)
return {
"author": self.author.display_name,
"name": self.name,
"title": self.title,
- "short_description": self.shortDesc,
+ "short_description": self.short_desc,
"desc": self.desc,
"type": self.type.toName(),
"created_at": self.created_at,
"screenshots": [base_url + ss.url for ss in self.screenshots],
"url": base_url + self.getDownloadURL(),
- "release": self.getDownloadRelease(protonum).id if self.getDownloadRelease(protonum) is not None else None,
+ "release": release and release.id,
"score": round(self.score * 10) / 10
}
return url_for("package_download_page",
author=self.author.username, name=self.name)
- def getDownloadRelease(self, protonum=None):
- version = None
- if protonum is not None:
+ def getDownloadRelease(self, version=None, protonum=None):
+ if version is None and protonum is not None:
version = MinetestRelease.query.filter(MinetestRelease.protocol >= int(protonum)).first()
if version is not None:
version = version.id
for rel in self.releases:
- if rel.approved and (protonum is None or
+ if rel.approved and (version is None or
((rel.min_rel is None or rel.min_rel_id <= version) and \
(rel.max_rel is None or rel.max_rel_id >= version))):
return rel
def __init__(self):
self.releaseDate = datetime.datetime.now()
+ def approve(self, user):
+ if not self.package.checkPerm(user, Permission.APPROVE_RELEASE):
+ return False
+
+ assert(self.task_id is None and self.url is not None and self.url != "")
+
+ self.approved = True
+ return True
+
class PackageReview(db.Model):
id = db.Column(db.Integer, primary_key=True)
projects / cheatdb.git / blobdiff