forked from crunk/distribusi-verse
73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
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
|