distribusi-verse/verse/themeselector.py

import os
import shutil
from flask import render_template

from statuspengguna.helper import UserHelper
from distribusisinfo import DistribusisInfo

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


def ThemeSelector():
    themeform = ThemeForm()
    publicthemeform = PublicThemeForm()
    publicthemeform.publicthemes.choices = DistribusisInfo.publicthemes()
    current_distribusi = UserHelper.current_distribusi()
    if themeform.validate_on_submit():
        copycssfile = os.path.join(
            "themes",
            f"{themeform.theme.data}.css",
        )
        MoveCssToUserFolder(current_distribusi, copycssfile)
    if publicthemeform.validate_on_submit():
        copycssfile = os.path.join(
            "themes/publicthemes/",
            f"{publicthemeform.publicthemes.data}.css",
        )
        MoveCssToUserFolder(current_distribusi, copycssfile)
    return RenderDistribusiTemplate(
        themeform,
        publicthemeform,
        current_distribusi
    )


def MoveCssToUserFolder(current_distribusi, copycssfile):
    newcssfolder = os.path.join("themes/userthemes", current_distribusi)
    if not os.path.exists(newcssfolder):
        os.mkdir(newcssfolder)
    shutil.copy(copycssfile, newcssfolder)


def RenderDistribusiTemplate(themeform, publicthemeform, current_distribusi):
    uploadform = UploadForm()
    distribusiform = DistribusiForm()
    selectorform = SelectorForm()

    files_uploaded = UserHelper.is_zip_uploaded(current_distribusi)
    distribusi_live = UserHelper.is_distribusi_live(current_distribusi)
    css_selected = UserHelper.is_css_selected(current_distribusi)
    selectorvisible = False
    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