1
0
Fork 0
mirror of https://codeberg.org/Mo8it/AdvLabDB.git synced 2024-11-08 21:21:06 +00:00
AdvLabDB/advlabdb/utils.py

117 lines
3.1 KiB
Python

from random import choice
from string import ascii_letters, digits
from flask import flash
from flask_security import current_user
from advlabdb import db
from advlabdb.models import Semester
def makeTable(headerAndDataList, rows, tableId="table"):
def cellString(cell):
cell = str(cell)
if cell == "[]":
return "<em>None</em>"
excludeChars = """<>"'[]"""
for c in excludeChars:
if c in cell:
cell = cell.replace(c, "")
return cell
def td(cell):
return "<td>" + cellString(cell) + "</td>"
def th(cell):
return '<th data-sortable="true" data-field="' + titleToTemplate(cell) + '">' + cellString(cell) + "</th>"
table = (
'''<table
id="'''
+ tableId
+ """"
data-classes="table table-bordered table-striped"
data-toggle="table"
data-thead-classes="table-dark"
data-search="true"
data-show-toggle="true"
data-show-columns="true"
data-click-to-select="true"
data-show-export="true"
data-export-types="['json', 'xml', 'csv', 'txt', 'sql', 'pdf']">
<thead>
<tr>
<th data-field="state" data-checkbox="true"></th>"""
)
for i in headerAndDataList:
table += th(i[0])
table += """</tr>
</thead>
<tbody>\n"""
for row in rows:
table += "<tr><td></td>"
for i in headerAndDataList:
table += td(eval(i[1]))
table += "</tr>\n"
table += """</tbody>
</table>
<br>
<hr>
<br>"""
return table
def appointmentDate(date):
return date.strftime("%a %d.%m.%Y")
def randomPassword():
return "".join(choice(ascii_letters + digits) for i in range(12))
def titleToTemplate(page):
return page.lower().replace(" ", "_")
def userActiveSemester(flashWarning=False):
lastSemesterId = Semester.query.all()[-1].id
if current_user.active_semester_id is None:
current_user.active_semester_id = lastSemesterId
db.session.commit()
elif current_user.active_semester_id != lastSemesterId:
activeSemester = Semester.query.get(current_user.active_semester_id)
if activeSemester:
if flashWarning:
flash(f"You are in the old semester {activeSemester.label}!", "warning")
else:
current_user.active_semester_id = lastSemesterId
db.session.commit()
flash("Semester changed!", "warning")
return Semester.query.get(current_user.active_semester_id)
def partFromLabelInUserActiveSemester(partLabel):
for part in userActiveSemester().parts:
if part.label == partLabel:
return part
def setUserActiveSemester(semesterId):
if current_user.active_semester_id != semesterId:
if Semester.query.get(semesterId):
try:
current_user.active_semester_id = semesterId
db.session.commit()
except Exception as ex:
flash(str(ex), "error")
db.session.rollback()
else:
flash(f"Active semester changed to {Semester.query.get(semesterId).label}!", "warning")
else:
flash(f"No semester with ID {semesterId}!", "error")