distribusi-verse/verse/distribusikan/themeselector.py

import os
import shutil

from flask import render_template

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 statuspengguna.helper import UserHelper


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