distribusi-verse/verse/statuspengguna/helper.py

import os
from flask_login import current_user
from flask import flash
from sqlalchemy.exc import (
    DataError,
    DatabaseError,
    InterfaceError,
    InvalidRequestError,
)

from usermodel import User
from distribusimodel import Distribusis
from app import db


def IsZipUploaded(distribusiname):
    userfolder = os.path.join("stash", distribusiname)
    if os.path.exists(userfolder):
        zipfilename = "{}.zip".format(distribusiname)
        if os.path.exists(os.path.join(userfolder, zipfilename)):
            print("folder with zipfile found, file uploaded")
            return True
    print("distribusi folder has no zipfile")
    return False


def IsDistribusiLive(distribusiname):
    userfolder = os.path.join("stash", distribusiname)
    if os.path.exists(userfolder):
        zipfilename = "{}.zip".format(distribusiname)
        if os.path.exists(os.path.join(userfolder, zipfilename)):
            print("folder with zipfile found, file uploaded")
            return False
        if len(os.listdir(userfolder)) > 0:
            return True
    print("distribusi folder is empty")
    return False


def IsCssSelected(distribusiname):
    userfolder = os.path.join("themes/userthemes", distribusiname)
    if os.path.exists(userfolder):
        for file in os.listdir(userfolder):
            if file.endswith(".css"):
                return True
    print("distribusi folder has no css file")
    return False


def HasDistribusi():
    user = User.query.filter_by(email=current_user.email).first()
    distribusi = Distribusis.query.filter_by(userid=user.id).first()
    if distribusi is None:
        print("no distribusi found")
        return False
    return True


def CurrentDistribusi():
    user = User.query.filter_by(email=current_user.email).first()
    if user.currentdistribusi is None:
        return "None"
    return user.currentdistribusi


def ResetUserState():
    """reset user state upon visiting index, so that distribusi workflow can
    be done correctly"""
    if not current_user.is_authenticated:
        return
    try:
        user = User.query.filter_by(email=current_user.email).first()
        user.currentdistribusi = None
        db.session.commit()
    except (InvalidRequestError, DataError, InterfaceError, DatabaseError):
        db.session.rollback()
        flash("An error occured !", "danger")





# def IsThemeSelected
# def IsCustomThemePresent
# def IsDistribusiLive