"""This is the main flask distribusi page""" from datetime import timedelta from flask import ( render_template, redirect, url_for, session, send_from_directory, ) from flask_login import ( logout_user, login_required, current_user, ) from flask_wtf.csrf import CSRFError from app import create_app, login_manager from usermodel import User from distribusimodel import Distribusis # Forms! from forms.uploadform import UploadForm from forms.distribusiform import DistribusiForm from forms.themeform import ThemeForm from forms.publicthemeform import PublicThemeForm from forms.selectorform import SelectorForm # CRUD! from distribusiselector import ( SelectNewDistribusi, SelectUpdateDistribusi, DeleteDistribusi, SelectorVisible, ) # Interface! from editor import Editor from themeselector import ThemeSelector from upload import UploadNewDistribusi, UploadUpdatedFiles from adminpage import AdminPage from distribusiworkflow import DistribusiWorkflow # UserPengguna from statuspengguna.helper import ( IsZipUploaded, IsCssSelected, CurrentDistribusi, HasDistribusi, IsDistribusiLive, ResetUserState, DistribusiLimitReached, ) from statuspengguna.loginuser import LoginUser from statuspengguna.registeruser import RegisterUser from distribusisinfo import DistribusisInfo APP = create_app() @APP.before_request def session_handler(): session.permanent = True APP.permanent_session_lifetime = timedelta(minutes=30) @APP.route("/") def index(): ResetUserState() uploadform = UploadForm() distribusis = DistribusisInfo.visibledistribusis() distribusies = {} for distribusi in distribusis: user = User.query.filter_by(id=distribusi.userid).first() singledistribusi = { "username": user.username, "term": distribusi.term, "course": distribusi.course, "year": distribusi.year, "tags": distribusi.tags.split(","), } distribusies[distribusi.distribusiname] = singledistribusi years = uploadform.academicyear.choices terms = uploadform.term.choices courses = uploadform.course.choices adminuser = isadminuser() template = render_template( "index.html", distribusies=distribusies, years=years, terms=terms, courses=courses, adminuser=adminuser, ) return template @APP.route("/distribusi", methods=["GET", "POST"]) @login_required def distribusi(): return DistribusiWorkflow() @APP.route("/upload", methods=["POST"]) @login_required def upload(): distribusiform = DistribusiForm() themeform = ThemeForm() publicthemeform = PublicThemeForm() publicthemeform.publicthemes.choices = DistribusisInfo.publicthemes() selectorform = SelectorForm() selectorform.distribusis.choices = DistribusisInfo.userdistribusinames() selectorvisible = SelectorVisible() current_distribusi = CurrentDistribusi() if current_distribusi == "new" or HasDistribusi() is False: uploadform = UploadNewDistribusi(APP.config["UPLOAD_FOLDER"]) else: uploadform = UploadUpdatedFiles(APP.config["UPLOAD_FOLDER"]) files_uploaded = IsZipUploaded(uploadform.sitename.data) distribusi_live = IsDistribusiLive(current_distribusi) css_selected = IsCssSelected(current_distribusi) limit_reached = DistribusiLimitReached() 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 @APP.route("/theme", methods=["GET", "POST"]) @login_required def theme(): return ThemeSelector() @APP.route("/editor", methods=["GET", "POST"]) @login_required def editor(): return Editor() @APP.route("/selector", methods=["GET", "POST"]) @login_required def selector(): selectorform = SelectorForm() uploadform = UploadForm() distribusiform = DistribusiForm() themeform = ThemeForm() publicthemeform = PublicThemeForm() publicthemeform.publicthemes.choices = DistribusisInfo.publicthemes() selectorform = SelectorForm() selectorform.distribusis.choices = DistribusisInfo.userdistribusinames() current_distribusi = CurrentDistribusi() if selectorform.validate_on_submit(): if selectorform.new.data: selectorform = SelectNewDistribusi() if selectorform.delete.data: selectorform = DeleteDistribusi(selectorform.distribusis.data) selectorform.distribusis.choices = ( DistribusisInfo.userdistribusinames() ) if selectorform.update.data: selectorform = SelectUpdateDistribusi( selectorform.distribusis.data ) current_distribusi = CurrentDistribusi() distribusi = Distribusis.query.filter_by( distribusiname=current_distribusi ).first() uploadform.sitename.data = distribusi.distribusiname uploadform.sitename.render_kw = {"readonly": True} uploadform.term.data = distribusi.term uploadform.course.data = distribusi.course uploadform.academicyear.data = distribusi.year uploadform.tags.data = distribusi.tags files_uploaded = IsZipUploaded(current_distribusi) distribusi_live = IsDistribusiLive(current_distribusi) # because the user has chosen to update his distribusi, we assume # no selected css. css_selected = False selectorvisible = SelectorVisible() limit_reached = DistribusiLimitReached() 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 @APP.route("/stash/") def distribusistash(path): return send_from_directory("stash", path) @APP.route("/admin", methods=["GET", "POST"]) @login_required def admin(): return AdminPage() @APP.route("/logout") @login_required def logout(): logout_user() return redirect(url_for("index")) @APP.route("/login", methods=["GET", "POST"]) def login(): result = LoginUser() return result @APP.route("/register", methods=["GET", "POST"]) def register(): result = RegisterUser() return result @APP.errorhandler(CSRFError) def handle_csrf_error(e): return render_template("csrf_error.html", reason=e.description), 400 @login_manager.user_loader def load_user(user_id): return User.query.get(int(user_id)) def isadminuser(): if not current_user.is_authenticated: return False user = User.query.filter_by(email=current_user.email).first() return user.admin if __name__ == "__main__": APP.debug = True APP.run(port=5000)