added week calculations, refactor mocks

This commit is contained in:
Mariano Ramon
2020-05-17 10:14:29 -03:00
parent 3e073a0e4b
commit 59964dcab3
6 changed files with 110 additions and 4 deletions

6
dmapp/config.py Normal file
View File

@@ -0,0 +1,6 @@
DESKTOPS = ("Work",
"Browse",
"Write",
"Learn",
"Idle",
"Self")

14
dmapp/dmcore/__init__.py Normal file
View File

@@ -0,0 +1,14 @@
class ActiveDesktop:
def __init__(self, desktops, unlabeled = "Other"):
self.desktops = desktops
def active_workspace():
workspaces = subprocess.check_output(["wmctrl", "-d"]) \
.decode("utf-8").strip("\n").split("\n")
for workspace in workspaces:
if workspace[3] == "*":
return int(workspace[0])

6
dmapp/dmdb/__init__.py Normal file
View File

@@ -0,0 +1,6 @@
from pymongo import MongoClient
client = MongoClient()
db = client.deskmeter
switches = db.switch

View File

@@ -146,7 +146,7 @@ def get_period_totals(start, end):
#TODO _1 remove
#day += total
rows.append( {"ws": ws,
"total" : datetime.timedelta(seconds=total)})
"total" : convert_timedelta(datetime.timedelta(seconds=total)) })
#TODO _1 remove
#rows.append({"ws":"sum","total": day})
@@ -172,3 +172,11 @@ def local_date(date, convert=False):
if convert:
return pytz.utc.localize(date, is_dst=None).astimezone(bsas)
return date.replace(tzinfo=bsas)
def convert_timedelta(duration):
days, seconds = duration.days, duration.seconds
hours = days * 24 + seconds // 3600
minutes = (seconds % 3600) // 60
seconds = (seconds % 60)
return "{}:{:02d}:{:02d}".format(hours, minutes, seconds)

View File

@@ -42,6 +42,77 @@ class DMHTMLCalendar(calendar.HTMLCalendar):
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.datetime(2020, month, start_day).replace(hour=0,
minute=0,
second=0)
end = datetime.datetime(2020, 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 formatmonthname(self, theyear, themonth, withyear=True):
# """
# Return a month name as a table row.
# """
# if withyear:
# s = '%s %s' % (month_name[themonth], theyear)
# else:
# s = '%s' % month_name[themonth]
# return '<tr><th colspan="8" class="%s">%s</th></tr>' % (
# self.cssclass_month_head, s)
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.