distribusi-verse/verse/adminpage.py

import os
import shutil

from flask import render_template
from sqlalchemy.exc import (
    DatabaseError,
    DataError,
    InterfaceError,
    InvalidRequestError,
)

from app import db
from distribusikan.distribusisinfo import DistribusisInfo
from forms.admindistribusiform import AdminDistribusiForm
from forms.adminuserform import AdminUserForm
from models.distribusi_model import Distribusis
from models.user_model import User


def AdminPage():
    adminuserform = AddUsersToForm(AdminUserForm())
    admindistribusiform = AddDistribusisToForm(AdminDistribusiForm())
    if admindistribusiform.validate_on_submit():
        DeleteDistribusis(admindistribusiform)

    if adminuserform.validate_on_submit():
        if adminuserform.delete.data:
            DeleteUsers(adminuserform)

    template = render_template(
        "admin.html",
        adminuserform=adminuserform,
        admindistribusiform=admindistribusiform,
    )
    return template


def DeleteUsers(adminuserform):
    for userform in adminuserform:
        if "user" in userform.id:
            if userform.data:
                useremail = userform.label.text
                user = User.query.filter_by(email=useremail).first()
                DeleteUserDistribusis(user)
                DeleteUserFromDb(user)
                userform.errors.append(f"User {useremail} deleted!")



def DeleteUserFromDb(user):
    try:
        db.session.delete(user)
        db.session.commit()
    except (InvalidRequestError, DataError, InterfaceError, DatabaseError):
        db.session.rollback()


def DeleteUserDistribusis(user):
    distribusis = DistribusisInfo.getuserdistribusis(user.email)
    for distribusi in distribusis:
        DeleteDistribusiFiles(distribusi.distribusiname)
        DeleteDistribusiFromDb(distribusi)


def DeleteDistribusis(admindistribusiform):
    for distribusiform in admindistribusiform:
        if "distribusi" in distribusiform.id:
            if distribusiform.data:
                distribusiname = distribusiform.label.text
                distribusi = Distribusis.query.filter_by(
                    distribusiname=distribusiname
                ).first()
                DeleteDistribusiFromDb(distribusi)
                DeleteDistribusiFiles(distribusiname)
                distribusiform.errors.append("Deleted distribusi")


def DeleteDistribusiFromDb(distribusi):
    try:
        db.session.delete(distribusi)
        db.session.commit()
    except (InvalidRequestError, DataError, InterfaceError, DatabaseError):
        db.session.rollback()


def DeleteDistribusiFiles(distribusiname):
    userfolder = os.path.join("stash", distribusiname)
    if os.path.exists(userfolder):
        shutil.rmtree(userfolder)
    cssfolder = os.path.join("themes/userthemes", distribusiname)
    if os.path.exists(cssfolder):
        shutil.rmtree(cssfolder)


def AddDistribusisToForm(admindistribusiform):
    distribusis = DistribusisInfo.visibledistribusis()
    admindistribusiform = AdminDistribusiForm.distribusi_list_form_builder(
        distribusis
    )
    return admindistribusiform


def AddUsersToForm(adminuserform):
    users = User.query.all()
    adminuserform = AdminUserForm.user_list_form_builder(users)
    return adminuserform