distribusi-verse/verse/selector.py

import os
import shutil
from flask import flash
from flask_login import current_user

from usermodel import User
from distribusimodel import Distribusis

from forms.selectorform import SelectorForm

from statuspengguna.helper import HasDistribusi
from statuspengguna.helper import DistribusiSelected

from app import db


def SelectNewDistribusi():
    print("make a new distribusi")
    selectorform = SelectorForm()
    user = User.query.filter_by(email=current_user.email).first()
    try:
        user.currentdistribusi = "new"
        db.session.commit()
    except:
        db.session.rollback()
        selectorform.distribusis.errors.append("Unknown error occured!")
        flash("An error occured !", "danger")
    return selectorform


def SelectUpdateDistribusi(distribusiname):
    print(f"Update this distribusi {distribusiname}")
    selectorform = SelectorForm()
    user = User.query.filter_by(email=current_user.email).first()
    try:
        user.currentdistribusi = distribusiname
        db.session.commit()
    except:
        db.session.rollback()
        selectorform.distribusis.errors.append("Unknown error occured!")
        flash("An error occured !", "danger")
    return selectorform


def DeleteDistribusi(distribusiname):
    print(f"delete this distribusi {distribusiname}")
    selectorform = SelectorForm()
    try:
        user = User.query.filter_by(email=current_user.email).first()
        distribusi = Distribusis.query.filter_by(
            distribusiname=distribusiname
        ).first()
        if distribusi.userid is user.id:
            db.session.delete(distribusi)
            db.session.commit()
            userfolder = os.path.join("stash", distribusi.distribusiname)
            shutil.rmtree(userfolder)
    except:
        db.session.rollback()
        selectorform.distribusis.errors.append("Unknown error occured!")
        flash("An error occured !", "danger")
    return selectorform


def SelectorVisible():
    has_distribusi = HasDistribusi()
    distribusi_selected = DistribusiSelected()
    if distribusi_selected:
        return False
    if not has_distribusi:
        return False
    return True