from flask_migrate import Migrate
from urllib.parse import urlparse
from app import app, gravatar
-from datetime import datetime
from sqlalchemy.orm import validates
from flask_user import login_required, UserManager, UserMixin, SQLAlchemyAdapter
-import enum
+import enum, datetime
# Initialise database
db = SQLAlchemy(app)
replies = db.relationship("ThreadReply", backref="author", lazy="dynamic")
def __init__(self, username, active=False, email=None, password=None):
- import datetime
-
self.username = username
self.confirmed_at = datetime.datetime.now() - datetime.timedelta(days=6000)
self.display_name = username
else:
raise Exception("Permission {} is not related to users".format(perm.name))
+ def canCommentRL(self):
+ hour_ago = datetime.datetime.utcnow() - datetime.timedelta(hours=1)
+ return ThreadReply.query.filter_by(author=self) \
+ .filter(ThreadReply.created_at > hour_ago).count() < 4
+
+ def canOpenThreadRL(self):
+ hour_ago = datetime.datetime.utcnow() - datetime.timedelta(hours=1)
+ return Thread.query.filter_by(author=self) \
+ .filter(Thread.created_at > hour_ago).count() < 2
+
class UserEmailVerification(db.Model):
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey("user.id"))
shortDesc = db.Column(db.String(200), nullable=False)
desc = db.Column(db.Text, nullable=True)
type = db.Column(db.Enum(PackageType))
- created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
+ created_at = db.Column(db.DateTime, nullable=False, default=datetime.datetime.utcnow)
license_id = db.Column(db.Integer, db.ForeignKey("license.id"), nullable=False, default=1)
license = db.relationship("License", foreign_keys=[license_id])
isOwner = user == self.author
+ if perm == Permission.CREATE_THREAD:
+ return user.rank.atLeast(UserRank.MEMBER)
+
# Members can edit their own packages, and editors can edit any packages
- if perm == Permission.MAKE_RELEASE or perm == Permission.ADD_SCREENSHOTS or perm == Permission.CREATE_THREAD:
+ if perm == Permission.MAKE_RELEASE or perm == Permission.ADD_SCREENSHOTS:
return isOwner or user.rank.atLeast(UserRank.EDITOR)
if perm == Permission.EDIT_PACKAGE or perm == Permission.APPROVE_CHANGES:
raise Exception("Permission {} is not related to packages".format(perm.name))
def recalcScore(self):
- import datetime
-
self.score = 10
if self.forums is not None:
def __init__(self):
- self.releaseDate = datetime.now()
+ self.releaseDate = datetime.datetime.now()
class PackageReview(db.Model):
title = db.Column(db.String(100), nullable=False)
private = db.Column(db.Boolean, server_default="0")
- created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
+ created_at = db.Column(db.DateTime, nullable=False, default=datetime.datetime.utcnow)
replies = db.relationship("ThreadReply", backref="thread", lazy="dynamic")
thread_id = db.Column(db.Integer, db.ForeignKey("thread.id"), nullable=False)
comment = db.Column(db.String(500), nullable=False)
author_id = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
- created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
+ created_at = db.Column(db.DateTime, nullable=False, default=datetime.datetime.utcnow)
REPO_BLACKLIST = [".zip", "mediafire.com", "dropbox.com", "weebly.com", \
posts = db.Column(db.Integer, nullable=False)
views = db.Column(db.Integer, nullable=False)
- created_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
+ created_at = db.Column(db.DateTime, nullable=False, default=datetime.datetime.utcnow)
def getRepoURL(self):
if self.link is None:
from app.models import *
from app.utils import triggerNotif, clearNotifications
+import datetime
+
from flask_wtf import FlaskForm
from wtforms import *
from wtforms.validators import *
if current_user.is_authenticated and request.method == "POST":
comment = request.form["comment"]
+ if not current_user.canCommentRL():
+ flash("Please wait before commenting again", "danger")
+ if package:
+ return redirect(package.getDetailsURL())
+ else:
+ return redirect(url_for("home_page"))
+
if len(comment) <= 500 and len(comment) > 3:
reply = ThreadReply()
reply.author = current_user
if package is None:
flash("Unable to find that package!", "error")
- # Don't allow making threads on approved packages for now
+ # Don't allow making orphan threads on approved packages for now
if package is None:
abort(403)
def_is_private = request.args.get("private") or False
- if not package.approved:
+ if package is None or not package.approved:
def_is_private = True
- allow_change = package.approved
- is_review_thread = package is not None and not package.approved
+ allow_change = package and package.approved
+ is_review_thread = package and not package.approved
# Check that user can make the thread
if not package.checkPerm(current_user, Permission.CREATE_THREAD):
# Only allow creating one thread when not approved
elif is_review_thread and package.review_thread is not None:
flash("A review thread already exists!", "error")
- if request.method == "GET":
- return redirect(url_for("thread_page", id=package.review_thread.id))
+ return redirect(url_for("thread_page", id=package.review_thread.id))
+
+ elif not current_user.canOpenThreadRL():
+ flash("Please wait before opening another thread", "danger")
+
+ if package:
+ return redirect(package.getDetailsURL())
+ else:
+ return redirect(url_for("home_page"))
# Set default values
elif request.method == "GET":
if is_review_thread:
package.review_thread = thread
+ notif_msg = None
if package is not None:
- triggerNotif(package.author, current_user,
- "New thread '{}' on package {}".format(thread.title, package.title), url_for("thread_page", id=thread.id))
+ notif_msg = "New thread '{}' on package {}".format(thread.title, package.title)
+ triggerNotif(package.author, current_user, notif_msg, url_for("thread_page", id=thread.id))
+ else:
+ notif_msg = "New thread '{}'".format(thread.title)
+
+ for user in User.query.filter(User.rank >= UserRank.EDITOR).all():
+ triggerNotif(user, current_user, notif_msg, url_for("thread_page", id=thread.id))
db.session.commit()
projects / cheatdb.git / commitdiff