125 lines
3.7 KiB
Python
125 lines
3.7 KiB
Python
import os
|
|
import shutil
|
|
from flask import render_template
|
|
|
|
from sqlalchemy.exc import (
|
|
DataError,
|
|
DatabaseError,
|
|
InterfaceError,
|
|
InvalidRequestError,
|
|
)
|
|
from app import db
|
|
|
|
from usermodel import User
|
|
from distribusimodel import Distribusis
|
|
from distribusisinfo import DistribusisInfo
|
|
|
|
from forms.adminuserform import AdminUserForm
|
|
from forms.admindistribusiform import AdminDistribusiForm
|
|
|
|
|
|
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)
|
|
if adminuserform.tutors.data:
|
|
MakeUsersTutors(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 MakeUsersTutors(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()
|
|
MakeUserTutorinDb(user)
|
|
userform.errors.append(f"User {useremail} is now a tutor!")
|
|
|
|
|
|
def MakeUserTutorinDb(user):
|
|
try:
|
|
user.tutor = True
|
|
db.session.commit()
|
|
except (InvalidRequestError, DataError, InterfaceError, DatabaseError):
|
|
db.session.rollback()
|
|
|
|
|
|
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)
|
|
shutil.rmtree(userfolder)
|
|
cssfolder = os.path.join("themes/userthemes", distribusiname)
|
|
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
|