import os import shutil import zipfile # Tada! from distribusi.cli import build_argparser from distribusi.distribusi import distribusify from flask import flash, redirect, render_template, url_for from flask_login import current_user from sqlalchemy.exc import ( DatabaseError, DataError, InterfaceError, InvalidRequestError, ) from app import db from distribusikan.distribusiselector import SelectorVisible from distribusikan.distribusisinfo import DistribusisInfo from distribusikan.forms.distribusiform import DistribusiForm from distribusikan.forms.publicthemeform import PublicThemeForm from distribusikan.forms.selectorform import SelectorForm from distribusikan.forms.themeform import ThemeForm # Forms! from distribusikan.forms.uploadform import UploadForm from models.distribusi_model import Distribusis from models.user_model import User # UserPengguna from statuspengguna.helper import UserHelper def DistribusiWorkflow(): distribusiform = DistribusiForm() current_distribusi = UserHelper.current_distribusi() user = User.query.filter_by(email=current_user.email).first() distribusi = Distribusis.query.filter_by( distribusiname=current_distribusi ).first() if distribusiform.validate_on_submit(): userfolder = os.path.join("stash", distribusi.distribusiname) cssfile = GetCssFile(distribusi) UnzipDistribusiFiles(distribusi, userfolder) CleanUpDistribusiFiles(userfolder) RunDistribusi(userfolder, cssfile) SetDistribusiToVisible(distribusi, user) DeleteCssFile(cssfile) return redirect(url_for("index")) return RenderDistribusiTemplate(distribusiform, current_distribusi) def UnzipDistribusiFiles(distribusi, userfolder): zipfilename = "{}.zip".format(distribusi.distribusiname) unzipfile = os.path.join(userfolder, zipfilename) if os.path.exists(unzipfile): with zipfile.ZipFile(unzipfile, "r") as zip_ref: zip_ref.extractall(userfolder) # after extracting all files remove zip file. if os.path.exists(unzipfile): os.remove(os.path.join(userfolder, zipfilename)) def CleanUpDistribusiFiles(userfolder): if os.path.exists(userfolder): RemoveMacFolders(userfolder) def RemoveMacFolders(path): for filename in os.listdir(path): fullpath = os.path.join(path, filename) if filename.startswith("."): if os.path.isdir(fullpath): shutil.rmtree(fullpath) else: os.remove(fullpath) if filename == "__MACOSX": shutil.rmtree(fullpath) if os.path.isdir(fullpath): RemoveMacFolders(fullpath) def GetCssFile(distribusi): cssfile = "" cssfolder = os.path.join("themes/userthemes", distribusi.distribusiname) if os.path.exists(cssfolder): for filename in os.listdir(cssfolder): if filename.endswith(".css"): cssfile = os.path.join(cssfolder, filename) return cssfile def RunDistribusi(userfolder, cssfile): parser = build_argparser() args = parser.parse_args(["-t", "--menu-with-index", "-s", cssfile]) distribusify(args, userfolder) def SetDistribusiToVisible(distribusi, user): try: distribusi.visible = True user.currentdistribusi = None db.session.commit() except (InvalidRequestError, DataError, InterfaceError, DatabaseError): db.session.rollback() flash("Unknown error occured!") def DeleteCssFile(cssfile): if os.path.exists(cssfile): os.remove(cssfile) def RenderDistribusiTemplate(distribusiform, current_distribusi): uploadform = UploadForm() themeform = ThemeForm() publicthemeform = PublicThemeForm() publicthemeform.publicthemes.choices = DistribusisInfo.publicthemes() selectorform = SelectorForm() selectorform.distribusis.choices = DistribusisInfo.userdistribusinames() selectorvisible = SelectorVisible() files_uploaded = UserHelper.is_zip_uploaded(current_distribusi) distribusi_live = UserHelper.is_distribusi_live(current_distribusi) css_selected = UserHelper.is_css_selected(current_distribusi) 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