135 lines
3.6 KiB
Python
135 lines
3.6 KiB
Python
from datetime import datetime
|
|
from pprint import pprint
|
|
from flask import Blueprint, render_template
|
|
import pytz
|
|
|
|
import calendar
|
|
|
|
|
|
from dmweb.dm import dmbp, get_period_totals, local_date
|
|
|
|
|
|
|
|
class DMHTMLCalendar(calendar.HTMLCalendar):
|
|
|
|
|
|
# def formatmonth(self, theyear, themonth, withyear=True):
|
|
# self.dmmonth = themonth
|
|
# super().formatmonth(self, theyear, themonth)
|
|
|
|
|
|
def setcalmonth(self, month):
|
|
self.dmmonth = month
|
|
|
|
def setcalyear(self, year):
|
|
self.dmyear = year
|
|
|
|
def oneday(self,month,day):
|
|
|
|
current_year = datetime.today().year
|
|
start = datetime(self.dmyear, month, day).replace(hour=0,
|
|
minute=0,
|
|
second=0)
|
|
|
|
end = datetime(self.dmyear, month, day).replace (hour=23,
|
|
minute=59,
|
|
second=59)
|
|
|
|
rows = get_period_totals( local_date(start),
|
|
local_date(end))
|
|
|
|
returnstr = "<table class='totaltable'>"
|
|
for row in rows:
|
|
returnstr += "<tr><td>{}</td><td>{}</td></tr>".format(row["ws"],row["total"])
|
|
|
|
returnstr += "</table>"
|
|
return returnstr
|
|
|
|
|
|
def oneweek(self,month,week):
|
|
|
|
start_day = None
|
|
end_day = None
|
|
|
|
for (d, wd) in week:
|
|
if d == 0:
|
|
continue
|
|
else:
|
|
start_day = d
|
|
break
|
|
|
|
|
|
for (d, wd) in reversed(week):
|
|
if d == 0:
|
|
continue
|
|
else:
|
|
end_day = d
|
|
break
|
|
|
|
|
|
start = datetime(self.dmyear, month, start_day).replace(hour=0,
|
|
minute=0,
|
|
second=0)
|
|
|
|
end = datetime(self.dmyear, month, end_day).replace (hour=23,
|
|
minute=59,
|
|
second=59)
|
|
|
|
rows = get_period_totals( local_date(start),
|
|
local_date(end))
|
|
|
|
returnstr = "<table class='totaltable'>"
|
|
for row in rows:
|
|
returnstr += "<tr><td>{}</td><td>{}</td></tr>".format(row["ws"],row["total"])
|
|
|
|
returnstr += "</table>"
|
|
return returnstr
|
|
|
|
|
|
def formatweekheader(self):
|
|
"""
|
|
Return a header for a week as a table row.
|
|
"""
|
|
s = ''.join(self.formatweekday(i) for i in self.iterweekdays())
|
|
s += "<td>Week Totals</td>"
|
|
return '<tr>%s</tr>' % s
|
|
|
|
|
|
|
|
def formatweek(self, theweek):
|
|
"""
|
|
Return a complete week as a table row.
|
|
"""
|
|
s = ''.join(self.formatday(d, wd) for (d, wd) in theweek)
|
|
s += "<td>{}</td>".format(self.oneweek(self.dmmonth, theweek))
|
|
return '<tr>%s</tr>' % s
|
|
|
|
|
|
def formatday(self, day, weekday):
|
|
"""
|
|
Return a day as a table cell.
|
|
"""
|
|
if day == 0:
|
|
return '<td class="noday"> </td>' # day outside month
|
|
else:
|
|
return '<td class="%s">%s</td>' % (self.cssclasses[weekday], self.oneday(self.dmmonth, day))
|
|
|
|
|
|
|
|
@dmbp.route("/calendar")
|
|
@dmbp.route("/calendar/<int:month>")
|
|
def calmain(month=None):
|
|
|
|
usemonth = datetime.today().month
|
|
useyear = datetime.today().year
|
|
if month:
|
|
usemonth = month
|
|
|
|
cal = DMHTMLCalendar(calendar.SATURDAY)
|
|
|
|
cal.setcalmonth(usemonth)
|
|
cal.setcalyear(useyear)
|
|
|
|
return render_template("calendar.html", content=cal.formatmonth(useyear,usemonth))
|
|
|