from flask_login import current_user

from usermodel import User
from distribusimodel import Distribusis


class DistribusisInfo:
    def userdistribusis():
        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 publicthemes():
        publicthemes = []
        distribusis = Distribusis.query.filter(
            Distribusis.publictheme.isnot(None)
        ).all()
        for distribusi in distribusis:
            publicthemes.append(
                f"{distribusi.publictheme} used in {distribusi.distribusiname}"
            )
        return publicthemes