distribusi-verse/verse/distribusikan/themeselector.py

import os
import shutil

from flask import render_template

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
from distribusikan.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