119 lines
3.6 KiB
Python
119 lines
3.6 KiB
Python
from flask import Blueprint, render_template
|
|
from pymongo import MongoClient
|
|
from pprint import pprint
|
|
import datetime
|
|
import pytz
|
|
|
|
client = MongoClient()
|
|
db = client.deskmeter
|
|
switches = db.switch
|
|
|
|
dmbp = Blueprint("deskmeter", __name__, url_prefix="/", template_folder='templates')
|
|
|
|
@dmbp.route("/")
|
|
def index():
|
|
"""
|
|
Show total time used in each desktop for today
|
|
"""
|
|
|
|
start = datetime.datetime.today().replace(hour=0,
|
|
minute=0,
|
|
second=0)
|
|
|
|
end = datetime.datetime.today().replace (hour=23,
|
|
minute=59,
|
|
second=59)
|
|
|
|
rows = get_period_totals(start, end)
|
|
|
|
return render_template("pages.html", rows=rows)
|
|
|
|
|
|
@dmbp.route("/period/<start>/<end>")
|
|
def period(start,end):
|
|
|
|
start = datetime.datetime(*map(int, start.split("-"))).replace(hour=0,
|
|
minute=0,
|
|
second=0)
|
|
|
|
end = datetime.datetime(*map(int, end.split("-"))).replace (hour=23,
|
|
minute=59,
|
|
second=59)
|
|
|
|
rows = get_period_totals(start, end)
|
|
|
|
return render_template("pages.html", rows=rows)
|
|
|
|
|
|
@dmbp.route("/totals")
|
|
def totals():
|
|
"""
|
|
Show total time used in each desktop for all time
|
|
"""
|
|
|
|
pipe = [{'$group': {'_id':"$workspace",'totals': {'$sum': '$delta'}}},
|
|
{'$sort': { "_id": 1}}]
|
|
|
|
rows = []
|
|
for total in switches.aggregate(pipeline=pipe):
|
|
rows.append( {"ws" : total["_id"],
|
|
"total": str(datetime.timedelta(seconds=total["totals"]))})
|
|
|
|
|
|
return render_template("pages.html", rows=rows)
|
|
|
|
|
|
def get_period_totals(start, end):
|
|
"""
|
|
TODOs: refactor to have available the ws array and the active ws function in order to remove
|
|
the delta = 0 thing
|
|
refactor to pass in the timezone and make the variable names clearer
|
|
"""
|
|
|
|
bsas = pytz.timezone("Etc/GMT+3")
|
|
local_start = start.replace(tzinfo=bsas)
|
|
local_end = end.replace(tzinfo=bsas)
|
|
|
|
queries = {
|
|
"period" : {"date":{"$gt":local_start,"$lt":local_end}},
|
|
"previous_doc" : {"date":{"$lt":local_start}}
|
|
#"next_doc" : {"date":{"$lt":local_start}}
|
|
}
|
|
|
|
length = switches.count_documents(queries["period"])
|
|
docs = switches.find(queries["period"]).sort([("date", 1)])
|
|
previous_doc = switches.find(queries["previous_doc"]).sort([("date", -1)]).limit(1)[0]
|
|
#next_doc = switches.find_one(queries["next_doc"])
|
|
last_doc = docs[length-1]
|
|
|
|
local_first_date = docs[0]["date"].replace(tzinfo=pytz.utc).astimezone(bsas)
|
|
local_last_date = last_doc["date"].replace(tzinfo=pytz.utc).astimezone(bsas)
|
|
|
|
delta_to_start = (local_first_date - local_start).total_seconds()
|
|
delta_to_end = (local_end - local_last_date).total_seconds()
|
|
|
|
|
|
pipe = [ {'$match': queries["period"] },
|
|
{'$group': { '_id':"$workspace",'totals': {'$sum': '$delta'}}},
|
|
{'$sort': { "_id": 1}}]
|
|
|
|
pre_rows = {}
|
|
for total in switches.aggregate(pipeline=pipe):
|
|
pre_rows[total["_id"]] = total["totals"]
|
|
|
|
if datetime.datetime.now().astimezone(bsas) < local_end:
|
|
delta_to_end = 0
|
|
|
|
if previous_doc["workspace"] not in pre_rows:
|
|
pre_rows[previous_doc["workspace"]] = 0
|
|
|
|
pre_rows[previous_doc["workspace"]] += round(delta_to_start)
|
|
pre_rows[last_doc["workspace"]] += round(delta_to_end)
|
|
|
|
rows = []
|
|
for ws, delta in pre_rows.items():
|
|
rows.append( {"ws": ws,
|
|
"total" : str(datetime.timedelta(seconds=delta))})
|
|
|
|
return rows
|