distribusi-verse/verse/distribusikan/distribusis_info.py

from flask_login import current_user

from models.distribusi_model import Distribusis
from models.user_model import User


class DistribusisInfo:
    def user_distribusinames():
        distribusinames = []
        user = User.query.filter_by(email=current_user.email).first()
        for distribusi in Distribusis.query.filter_by(userid=user.id).all():
            distribusinames.append(distribusi.distribusiname)
        return distribusinames

    def public_themes():
        publicthemes = []
        distribusis = Distribusis.query.filter(
            Distribusis.publictheme.isnot(None)
        ).all()
        for distribusi in distribusis:
            user = User.query.filter_by(id=distribusi.userid).first()
            publictheme = (
                f"{distribusi.distribusiname}/{distribusi.publictheme}",
                f"""{distribusi.publictheme} used in {distribusi.distribusiname}
made by {user.username}""",
            )
            publicthemes.append(publictheme)
        return publicthemes

    def visible_distribusis():
        distribusis = Distribusis.query.filter(
            Distribusis.visible.isnot(False)
        ).all()
        return distribusis

    def get_user_distribusis(useremail):
        user = User.query.filter_by(email=useremail).first()
        return Distribusis.query.filter_by(userid=user.id).all()