import os import shutil from flask import flash, render_template, redirect, url_for from flask_login import current_user from sqlalchemy.exc import ( DatabaseError, DataError, InterfaceError, InvalidRequestError, ) from app import db from distribusikan.distribusisinfo import DistribusisInfo from forms.distribusiform import DistribusiForm from forms.publicthemeform import PublicThemeForm from forms.selectorform import SelectorForm from forms.themeform import ThemeForm from forms.uploadform import UploadForm from models.distribusi_model import Distribusis from models.user_model import User # UserPengguna from statuspengguna.helper import UserHelper def DistribusiSelector(): uploadform = UploadForm() selectorform = SelectorForm() selectorform.distribusis.choices = DistribusisInfo.userdistribusinames() current_distribusi = UserHelper.current_distribusi() if selectorform.validate_on_submit(): if selectorform.new.data: SelectNewDistribusi() if selectorform.describe.data: return SelectDescribeDistribusi(selectorform.distribusis.data) if selectorform.delete.data: selectorform = DeleteDistribusi(selectorform.distribusis.data) selectorform.distribusis.choices = ( DistribusisInfo.userdistribusinames() ) if selectorform.update.data: SelectUpdateDistribusi(selectorform.distribusis.data) current_distribusi = UserHelper.current_distribusi() uploadform = AutoFillInUploadForm(uploadform, current_distribusi) return RenderDistribusiTemplate( selectorform, uploadform, current_distribusi ) def AutoFillInUploadForm(uploadform, current_distribusi): distribusi = Distribusis.query.filter_by( distribusiname=current_distribusi ).first() uploadform.sitename.data = distribusi.distribusiname uploadform.sitename.render_kw = {"readonly": True} uploadform.category.data = distribusi.category uploadform.year.data = distribusi.year uploadform.tags.data = distribusi.tags return uploadform def SelectNewDistribusi(): print("make a new distribusi") SelectCurrentDistribusi("new") def SelectDescribeDistribusi(distribusiname): return redirect( url_for( "describer.show_distribusi_files", distribusiname=distribusiname, ) ) def SelectUpdateDistribusi(distribusiname): print(f"Update this distribusi {distribusiname}") SelectCurrentDistribusi(distribusiname) 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) if os.path.exists(userfolder): shutil.rmtree(userfolder) cssfolder = os.path.join( "themes/userthemes", distribusi.distribusiname ) if os.path.exists(cssfolder): shutil.rmtree(cssfolder) if distribusi.publictheme is not None: publicthemefolder = os.path.join( "themes/publicthemes", distribusi.distribusiname ) if os.path.exists(publicthemefolder): shutil.rmtree(publicthemefolder) # SelectField error is list is a tuple?? why?? # selectorform.distribusis.errors.append("Distribusi deleted!") except (InvalidRequestError, DataError, InterfaceError, DatabaseError): db.session.rollback() # selectorform.distribusis.errors.append("Unknown error occured!") flash("An error occured !", "danger") return selectorform def SelectCurrentDistribusi(distribusiname): if not current_user.is_authenticated: return user = User.query.filter_by(email=current_user.email).first() try: user.currentdistribusi = distribusiname db.session.commit() except (InvalidRequestError, DataError, InterfaceError, DatabaseError): db.session.rollback() flash("An error occured !", "danger") def DistribusiSelected(): user = User.query.filter_by(email=current_user.email).first() if user.currentdistribusi is None: return False return True def SelectorVisible(): has_distribusi = UserHelper.has_distribusi() distribusi_selected = DistribusiSelected() if distribusi_selected: return False if not has_distribusi: return False return True def RenderDistribusiTemplate(selectorform, uploadform, current_distribusi): distribusiform = DistribusiForm() themeform = ThemeForm() publicthemeform = PublicThemeForm() publicthemeform.publicthemes.choices = DistribusisInfo.publicthemes() files_uploaded = UserHelper.is_zip_uploaded(current_distribusi) distribusi_live = UserHelper.is_distribusi_live(current_distribusi) # because the user has chosen to update his distribusi, we assume # no selected css. css_selected = False selectorvisible = SelectorVisible() limit_reached = UserHelper.distribusi_limit_reached() template = render_template( "distribusi.html", uploadform=uploadform, distribusiform=distribusiform, themeform=themeform, publicthemeform=publicthemeform, selectorform=selectorform, files_uploaded=files_uploaded, distribusi_live=distribusi_live, css_selected=css_selected, selectorvisible=selectorvisible, limit_reached=limit_reached, ) return template