AdvLabDB/advlabdb/utils.py

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")