|
@@ -3,8 +3,11 @@
|
|
|
All models for module
|
|
All models for module
|
|
|
"""
|
|
"""
|
|
|
|
|
|
|
|
|
|
+from hashlib import blake2b
|
|
|
|
|
+import hmac
|
|
|
|
|
+
|
|
|
from urllib.parse import quote
|
|
from urllib.parse import quote
|
|
|
-from datetime import datetime
|
|
|
|
|
|
|
+from datetime import datetime, timedelta
|
|
|
# from sqlalchemy.ext.hybrid import hybrid_method, hybrid_property
|
|
# from sqlalchemy.ext.hybrid import hybrid_method, hybrid_property
|
|
|
import markdown
|
|
import markdown
|
|
|
from flask import Markup
|
|
from flask import Markup
|
|
@@ -15,28 +18,16 @@ from app import db, argon2, login_manager
|
|
|
ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
|
|
ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
|
|
|
IMAGE_EXTENSIONS = set(['png', 'jpg', 'jpeg', 'gif'])
|
|
IMAGE_EXTENSIONS = set(['png', 'jpg', 'jpeg', 'gif'])
|
|
|
|
|
|
|
|
-page_file = db.Table(
|
|
|
|
|
- 'page_file',
|
|
|
|
|
- db.Column(
|
|
|
|
|
- 'page_id',
|
|
|
|
|
- db.Integer,
|
|
|
|
|
- db.ForeignKey('page.id'),
|
|
|
|
|
- primary_key=True
|
|
|
|
|
- ),
|
|
|
|
|
- db.Column(
|
|
|
|
|
- 'file_id',
|
|
|
|
|
- db.Integer,
|
|
|
|
|
- db.ForeignKey('file.id'),
|
|
|
|
|
- primary_key=True
|
|
|
|
|
- ),
|
|
|
|
|
-)
|
|
|
|
|
-
|
|
|
|
|
|
|
|
|
|
class User(db.Model, UserMixin):
|
|
class User(db.Model, UserMixin):
|
|
|
"""Model for User"""
|
|
"""Model for User"""
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
|
name = db.Column(db.String, unique=True, nullable=False)
|
|
name = db.Column(db.String, unique=True, nullable=False)
|
|
|
email = db.Column(db.String(255), unique=True)
|
|
email = db.Column(db.String(255), unique=True)
|
|
|
|
|
+ discord = db.Column(db.String(255), unique=True)
|
|
|
|
|
+ game_id = db.Column(db.BigInteger, unique=True)
|
|
|
|
|
+ alt = db.Column(db.Boolean, server_default='f', default=False)
|
|
|
|
|
+ party_member = db.Column(db.Boolean, server_default='f', default=False)
|
|
|
_password = db.Column("password", db.String(255))
|
|
_password = db.Column("password", db.String(255))
|
|
|
registration_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
registration_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
approved = db.Column(db.Boolean, server_default='f', default=False)
|
|
approved = db.Column(db.Boolean, server_default='f', default=False)
|
|
@@ -62,15 +53,21 @@ class User(db.Model, UserMixin):
|
|
|
"""Check if password is correct"""
|
|
"""Check if password is correct"""
|
|
|
return argon2.check_password_hash(self.password, password)
|
|
return argon2.check_password_hash(self.password, password)
|
|
|
|
|
|
|
|
|
|
+ def game_url(self):
|
|
|
|
|
+ """Give profile url"""
|
|
|
|
|
+ if self.game_id:
|
|
|
|
|
+ return "https://rivalregions.com/#slide/profile/%d" % self.game_id
|
|
|
|
|
+ return
|
|
|
|
|
+
|
|
|
|
|
|
|
|
-# function_id = db.Column(
|
|
|
|
|
-# db.Integer,
|
|
|
|
|
-# db.ForeignKey("function.id")
|
|
|
|
|
-# )
|
|
|
|
|
-# user = db.relationship(
|
|
|
|
|
-# "Function",
|
|
|
|
|
-# backref=db.backref("users", lazy="dynamic")
|
|
|
|
|
-# )
|
|
|
|
|
|
|
+ question_id = db.Column(
|
|
|
|
|
+ db.Integer,
|
|
|
|
|
+ db.ForeignKey("question.id")
|
|
|
|
|
+ )
|
|
|
|
|
+ user = db.relationship(
|
|
|
|
|
+ "Question",
|
|
|
|
|
+ backref=db.backref("users", lazy="dynamic")
|
|
|
|
|
+ )
|
|
|
|
|
|
|
|
|
|
|
|
|
class Page(db.Model):
|
|
class Page(db.Model):
|
|
@@ -152,9 +149,153 @@ class File(db.Model):
|
|
|
backref=db.backref("files", lazy="dynamic")
|
|
backref=db.backref("files", lazy="dynamic")
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
- files = db.relationship(
|
|
|
|
|
- 'Page',
|
|
|
|
|
- secondary=page_file,
|
|
|
|
|
- lazy='subquery',
|
|
|
|
|
- backref=db.backref('pages', lazy=True)
|
|
|
|
|
|
|
+
|
|
|
|
|
+class Ballot(db.Model):
|
|
|
|
|
+ """Model for Ballot"""
|
|
|
|
|
+ id = db.Column(db.Integer, primary_key=True)
|
|
|
|
|
+ name = db.Column(db.String, nullable=False)
|
|
|
|
|
+ description = db.Column(db.String)
|
|
|
|
|
+ start_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
|
+ end_at = db.Column(db.DateTime, nullable=False)
|
|
|
|
|
+
|
|
|
|
|
+ user_id = db.Column(
|
|
|
|
|
+ db.Integer,
|
|
|
|
|
+ db.ForeignKey("user.id")
|
|
|
|
|
+ )
|
|
|
|
|
+ user = db.relationship(
|
|
|
|
|
+ "User",
|
|
|
|
|
+ backref=db.backref("options", lazy="dynamic")
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+ priority_id = db.Column(
|
|
|
|
|
+ db.Integer,
|
|
|
|
|
+ db.ForeignKey("priority.id")
|
|
|
|
|
+ )
|
|
|
|
|
+ priority = db.relationship(
|
|
|
|
|
+ "Priority",
|
|
|
|
|
+ backref=db.backref("ballots", lazy="dynamic")
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+ def active(self):
|
|
|
|
|
+ """Check if vote is active"""
|
|
|
|
|
+ today = datetime.today()
|
|
|
|
|
+ if self.start_at <= today <= self.end_at:
|
|
|
|
|
+ return True
|
|
|
|
|
+ return False
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+class Priority(db.Model):
|
|
|
|
|
+ """Model for Priority"""
|
|
|
|
|
+ id = db.Column(db.Integer, primary_key=True)
|
|
|
|
|
+ name = db.Column(db.String, nullable=False)
|
|
|
|
|
+ description = db.Column(db.String, nullable=False)
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+class Question(db.Model):
|
|
|
|
|
+ """Model for Question"""
|
|
|
|
|
+ id = db.Column(db.Integer, primary_key=True)
|
|
|
|
|
+ name = db.Column(db.String, nullable=False)
|
|
|
|
|
+ description = db.Column(db.String)
|
|
|
|
|
+ combined_approval_voting = db.Column(db.Boolean, server_default='f', default=False)
|
|
|
|
|
+
|
|
|
|
|
+ ballot_id = db.Column(
|
|
|
|
|
+ db.Integer,
|
|
|
|
|
+ db.ForeignKey("ballot.id")
|
|
|
|
|
+ )
|
|
|
|
|
+ ballot = db.relationship(
|
|
|
|
|
+ "Ballot",
|
|
|
|
|
+ backref=db.backref("questions", lazy="dynamic")
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+ def has_voten(self, user_id):
|
|
|
|
|
+ """Check if user has voted"""
|
|
|
|
|
+ for option in self.options:
|
|
|
|
|
+ if option.votes.filter(Vote.user_id == user_id).first():
|
|
|
|
|
+ return True
|
|
|
|
|
+ return False
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+ def get_option(self, option):
|
|
|
|
|
+ """Get the pro option"""
|
|
|
|
|
+ return self.options.filter(Option.name == option).first()
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+ def score(self):
|
|
|
|
|
+ """Get winning option from question"""
|
|
|
|
|
+ if self.combined_approval_voting:
|
|
|
|
|
+ pro_option = self.get_option('Voor')
|
|
|
|
|
+ pro_votes = pro_option.votes.count()
|
|
|
|
|
+ against_option = self.get_option('Tegen')
|
|
|
|
|
+ against_votes = against_option.votes.count()
|
|
|
|
|
+ return pro_votes - against_votes
|
|
|
|
|
+
|
|
|
|
|
+ higest = Option()
|
|
|
|
|
+ for option in self.options:
|
|
|
|
|
+ if higest.votes.count() < option.votes.count():
|
|
|
|
|
+ higest = option
|
|
|
|
|
+ return higest.name
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+class Option(db.Model):
|
|
|
|
|
+ """Model for Option"""
|
|
|
|
|
+ id = db.Column(db.Integer, primary_key=True)
|
|
|
|
|
+ name = db.Column(db.String)
|
|
|
|
|
+ description = db.Column(db.String)
|
|
|
|
|
+
|
|
|
|
|
+ user_id = db.Column(
|
|
|
|
|
+ db.Integer,
|
|
|
|
|
+ db.ForeignKey("user.id")
|
|
|
|
|
+ )
|
|
|
|
|
+ user = db.relationship(
|
|
|
|
|
+ "User",
|
|
|
|
|
+ backref=db.backref("eligible", lazy="dynamic")
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+ question_id = db.Column(
|
|
|
|
|
+ db.Integer,
|
|
|
|
|
+ db.ForeignKey("question.id")
|
|
|
|
|
+ )
|
|
|
|
|
+ question = db.relationship(
|
|
|
|
|
+ "Question",
|
|
|
|
|
+ backref=db.backref("options", lazy="dynamic")
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+class Vote(db.Model):
|
|
|
|
|
+ """Model for Vote"""
|
|
|
|
|
+ id = db.Column(db.Integer, primary_key=True)
|
|
|
|
|
+ datetime = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
|
+
|
|
|
|
|
+ option_id = db.Column(
|
|
|
|
|
+ db.Integer,
|
|
|
|
|
+ db.ForeignKey("option.id")
|
|
|
|
|
+ )
|
|
|
|
|
+ option = db.relationship(
|
|
|
|
|
+ "Option",
|
|
|
|
|
+ backref=db.backref("votes", lazy="dynamic")
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+ user_id = db.Column(
|
|
|
|
|
+ db.Integer,
|
|
|
|
|
+ db.ForeignKey("user.id")
|
|
|
)
|
|
)
|
|
|
|
|
+ user = db.relationship(
|
|
|
|
|
+ "User",
|
|
|
|
|
+ backref=db.backref("votes", lazy="dynamic")
|
|
|
|
|
+ )
|
|
|
|
|
+
|
|
|
|
|
+class Code(db.Model):
|
|
|
|
|
+ """Model for code"""
|
|
|
|
|
+ id = db.Column(db.Integer, primary_key=True)
|
|
|
|
|
+ expire_date = db.Column(db.DateTime, default=datetime.utcnow() + timedelta(days=20))
|
|
|
|
|
+ secret = db.Column(db.String(255), unique=True)
|
|
|
|
|
+
|
|
|
|
|
+ def get_digest(self, string):
|
|
|
|
|
+ """Generate digest on string"""
|
|
|
|
|
+ if isinstance(string, int):
|
|
|
|
|
+ string = str(string)
|
|
|
|
|
+ string = string.encode('utf-8')
|
|
|
|
|
+ secret = self.secret.encode('utf-8')
|
|
|
|
|
+ blake = blake2b(digest_size=3)
|
|
|
|
|
+ blake.update(secret + string)
|
|
|
|
|
+ return blake.hexdigest()
|