distribusi-verse/verse/distribusikan/theme_selector.py

import os
import shutil

from flask import render_template

from distribusikan.distribusis_info 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 theme_selector():
    themeform = ThemeForm()
    publicthemeform = PublicThemeForm()
    publicthemeform.publicthemes.choices = DistribusisInfo.public_themes()
    current_distribusi = UserHelper.current_distribusi()
    if themeform.validate_on_submit():
        copycssfile = os.path.join(
            "themes",
            f"{themeform.theme.data}.css",
        )
        move_css_to_user_folder(current_distribusi, copycssfile)
    if publicthemeform.validate_on_submit():
        copycssfile = os.path.join(
            "themes/publicthemes/",
            f"{publicthemeform.publicthemes.data}.css",
        )
        move_css_to_user_folder(current_distribusi, copycssfile)
    return render_distribusi_template(
        themeform, publicthemeform, current_distribusi
    )


def move_css_to_user_folder(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 render_distribusi_template(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