models.py

from sqlalchemy import Column, Integer, String, ForeignKey, DateTime, Boolean
import datetime
from sqlalchemy.orm import relationship
from database import Base

class SolvedQuestion(Base):
        __tablename__ = 'solved_questions'
        username = Column(String(50), ForeignKey('hackers.username'), primary_key=True)
        question_id = Column(Integer, ForeignKey('questions.id', ondelete="CASCADE"), primary_key=True)
        date = Column(DateTime, default=datetime.datetime.utcnow)
        question = relationship("Question")

class User(Base):
    __tablename__ = 'hackers'
    username = Column(String(50), primary_key=True, unique=True)
    password = Column(String(50))
    admin = Column(Boolean, default=False)
    solved_questions = relationship("SolvedQuestion", cascade="save-update, merge, delete, delete-orphan")
    def __init__(self, username, password, admin=False):
        self.username = username
        self.password = password
        self.admin = admin
    def __repr__(self):
        return '<User {}>'.format(self.username)

    @property
    def is_authenticated(self):
        return True

    @property
    def is_active(self):
        return True

    @property
    def is_anonymous(self):
        return False

    @property
    def total_score(self):
        if len(self.solved_questions) > 0:
            total = 0
            # return reduce(lambda total,solvedquestion: total + solvedquestion.question.points, self.solved_questions)
            for solvedq in self.solved_questions:
                if solvedq.question.is_hidden:
                    continue
                total += solvedq.question.points
            return total
        else:
            return 0
    @property
    def is_admin(self):
        return self.admin
    
    def get_id(self):
        return str(self.username)
    
class Question(Base):
    __tablename__ = 'questions'
    id = Column(Integer, primary_key=True)
    name = Column(String(50))
    desc = Column(String(1000))
    flag = Column(String(200))
    points = Column(Integer)
    category = Column(String(50))
    hide = Column(Boolean)
    filename = Column(String(1000))

    def __init__(self, name = "", flag = "", desc="", category = "", points = 0, filename = "#", hide=False):
        self.name = name
        self.desc = desc
        self.flag = flag
        self.points = points
        self.category = category
        self.filename = filename
        self.hide = hide

    def __repr__(self):
        return '<Question ID: {} Name: {} Flag: {}>'.format(self.id, self.name, self.flag,)

    @property
    def is_hidden(self):
        return self.hide

    @property
    def solved_by(self):
        try:
            solvedByList = SolvedQuestion.query.filter(SolvedQuestion.question_id == self.id)
            solvedByListNoAdmin = filter(lambda x: not User.query.get(User.username == x.username).is_admin, solved_by_list)
            return solvedByListNoAdmin
        except:
            return 0