121 lines
2.9 KiB
Python
121 lines
2.9 KiB
Python
import functools
|
|
from datetime import datetime, time, date, timedelta
|
|
|
|
from flask import g, session, redirect
|
|
|
|
from estusshots import config, app, db
|
|
|
|
TIME_FMT = "%H:%M"
|
|
DATE_FMT = "%Y-%m-%d"
|
|
|
|
|
|
def str_to_datetime(data: str) -> datetime:
|
|
"""
|
|
Convert %H:%M formatted string into a python datetime object
|
|
"""
|
|
data = ":".join(data.split(":")[:2])
|
|
return datetime.strptime(data, TIME_FMT)
|
|
|
|
|
|
def datetime_time_str(data: datetime) -> str:
|
|
"""
|
|
Convert a datetime object into a formatted string for display
|
|
:param data: datetime
|
|
:return: str
|
|
"""
|
|
return data.strftime(TIME_FMT)
|
|
|
|
|
|
def timedelta_to_str(data: timedelta) -> str:
|
|
"""
|
|
Remove second and microsecond portion from timedeltas for display
|
|
:param data: datetime.timedelta
|
|
:return: str
|
|
"""
|
|
return str(
|
|
data - timedelta(seconds=data.seconds, microseconds=data.microseconds)
|
|
)
|
|
|
|
|
|
def timedelta(start: time, end: time) -> float:
|
|
startDateTime = datetime.combine(date.today(), start)
|
|
# Check if the the end is still on the same day
|
|
if start.hour > end.hour:
|
|
base = date.today() + timedelta(days=1)
|
|
else:
|
|
base = date.today()
|
|
endDateTime = datetime.combine(base, end)
|
|
difference = startDateTime - endDateTime
|
|
difference_hours = difference.total_seconds() / 3600
|
|
return difference_hours
|
|
|
|
|
|
def combine_datetime(date: datetime.date, time: datetime.time):
|
|
"""
|
|
Combine a date and time object into a datetime object
|
|
"""
|
|
return datetime(
|
|
date.year,
|
|
date.month,
|
|
date.day,
|
|
time.hour,
|
|
time.minute,
|
|
time.second,
|
|
time.microsecond,
|
|
)
|
|
|
|
|
|
def get_user_type(password):
|
|
# TODO password hashing?
|
|
if password == config.WRITE_PW:
|
|
return "editor"
|
|
if password == config.READ_PW:
|
|
return "readonly"
|
|
return False
|
|
|
|
|
|
def set_user_role(data):
|
|
"""Set the users role in the flask g object for later usage"""
|
|
g.is_editor = data == "editor"
|
|
|
|
|
|
def authorize(func):
|
|
@functools.wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
try:
|
|
set_user_role(session["user_type"])
|
|
except KeyError:
|
|
return redirect("/login")
|
|
return func(*args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
@app.template_filter("format_time")
|
|
def format_time(value):
|
|
"""Make the datetime to time string formatting available to jinja2"""
|
|
if value is None:
|
|
return ""
|
|
return datetime_time_str(value)
|
|
|
|
|
|
@app.template_filter("format_timedelta")
|
|
def format_timedelta(value):
|
|
"""Make formatting for timedeltas available to jinja2"""
|
|
if value is None:
|
|
return ""
|
|
return timedelta_to_str(value)
|
|
|
|
|
|
@app.cli.command("initdb")
|
|
def init_db_command():
|
|
"""Initializes the database."""
|
|
db.init_db()
|
|
|
|
|
|
@app.teardown_appcontext
|
|
def close_connection(exception):
|
|
db = getattr(g, "_database", None)
|
|
if db is not None:
|
|
db.close()
|