diff --git a/dmapp/dmweb/__init__.py b/dmapp/dmweb/__init__.py index 7180060..1d0206f 100644 --- a/dmapp/dmweb/__init__.py +++ b/dmapp/dmweb/__init__.py @@ -1,16 +1,9 @@ -import os - from flask import Flask from dmweb import dm -def create_app(test_config=None): +def create_app(): - app = Flask("deskmeter") #instance_relative_config=True - - try: - os.makedirs(app.instance_path) - except OSError: - pass + app = Flask("deskmeter") app.register_blueprint(dm.dmbp) diff --git a/dmapp/dmweb/dm.py b/dmapp/dmweb/dm.py index 59e75a4..b5b8a40 100644 --- a/dmapp/dmweb/dm.py +++ b/dmapp/dmweb/dm.py @@ -1,7 +1,9 @@ +from collections import Counter +import datetime +from pprint import pprint + from flask import Blueprint, render_template from pymongo import MongoClient -from pprint import pprint -import datetime import pytz client = MongoClient() @@ -24,11 +26,31 @@ def index(): minute=59, second=59) - rows = get_period_totals(start, end) + rows = get_period_totals( local_date(start), + local_date(end)) return render_template("pages.html", rows=rows) +@dmbp.route("/day//") +def oneday(month,day): + + start = datetime.datetime(2020, month, day).replace(hour=0, + minute=0, + second=0) + + end = datetime.datetime(2020, month, day).replace (hour=23, + minute=59, + second=59) + + rows = get_period_totals( local_date(start), + local_date(end)) + + return render_template("pages.html", rows=rows) + + + + @dmbp.route("/period//") def period(start,end): @@ -40,7 +62,8 @@ def period(start,end): minute=59, second=59) - rows = get_period_totals(start, end) + rows = get_period_totals( local_date(start), + local_date(end)) return render_template("pages.html", rows=rows) @@ -64,55 +87,82 @@ def totals(): 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" : start, "$lt" : end }}, + "previous_doc" : {"date" : {"$lt" : start }} + #"next_doc" : {"date":{"$lt":start}} + } - 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)]) + + #next_doc = switches.find_one(queries["next_doc"]) + last_doc = docs[length-1] - 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] + first_date = local_date(docs[0]["date"], True) + last_date = local_date(last_doc["date"], True) - local_first_date = docs[0]["date"].replace(tzinfo=pytz.utc).astimezone(bsas) - local_last_date = last_doc["date"].replace(tzinfo=pytz.utc).astimezone(bsas) + pipe = [ {'$match': queries["period"] }, + {'$group': { '_id':"$workspace",'totals': {'$sum': '$delta'}}}, + {'$sort': { "_id": 1}}] - delta_to_start = (local_first_date - local_start).total_seconds() - delta_to_end = (local_end - local_last_date).total_seconds() + pre_rows = Counter({}) + for total in switches.aggregate(pipeline=pipe): + pre_rows[total["_id"]] = total["totals"] - 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"] + time_corrections = [] + if first_date > start: + previous_doc = switches.find(queries["previous_doc"]).sort([("date", -1)]).limit(1)[0] + time_corrections.append(Counter({previous_doc["workspace"] : split_entry(previous_doc, start)})) + + now = local_date(datetime.datetime.now()) - if datetime.datetime.now().astimezone(bsas) < local_end: - delta_to_end = 0 + if end > now: + time_corrections.append( Counter({ last_doc["workspace"] : split_entry(last_doc, now, after=False)})) - if previous_doc["workspace"] not in pre_rows: - pre_rows[previous_doc["workspace"]] = 0 + if end > last_date and now > end: + time_corrections.append(Counter({ last_doc["workspace"] : split_entry(last_doc, end, after=False)})) - 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))}) + for correction in time_corrections: + pre_rows += correction - return rows + day = 0 + rows = [] + for ws, total in pre_rows.items(): + day += total + rows.append( {"ws": ws, + "total" : datetime.timedelta(seconds=total)}) + + + rows.append({"ws":"sum","total": day}) + return rows + + +def split_entry(entry, split_time, after=True): + """ + entry must have a date an number + split time must be timezone aware + after, bolean to return the time after the split time + """ + first_half = round((split_time - local_date(entry["date"], True)).total_seconds()) + last_half = entry["delta"] - first_half + if after: + return last_half + + return last_half * -1 + + +def local_date(date, convert=False): + bsas = pytz.timezone("Etc/GMT+3") + if convert: + return pytz.utc.localize(date, is_dst=None).astimezone(bsas) + return date.replace(tzinfo=bsas) diff --git a/dmapp/tests/test_dmapp.py b/dmapp/tests/test_dmapp.py new file mode 100644 index 0000000..fb46c5c --- /dev/null +++ b/dmapp/tests/test_dmapp.py @@ -0,0 +1,38 @@ +import os +import sys +import unittest + +sys.path.append(os.path.abspath('../')) +from dmweb import create_app + + +class TestDeskMeter(unittest.TestCase): + + @classmethod + def setUpClass(cls): + pass + + @classmethod + def tearDownClass(cls): + pass + + def setUp(self): + + dmapp = create_app() + self.client = dmapp.test_client() + self.client.testing = True + + def tearDown(self): + pass + + # test method + def test_home_status_code(self): + result = self.client.get('/') + + # assert the status code of the response + self.assertEqual(result.status_code, 200) + + +# runs the unit tests in the module +if __name__ == '__main__': + unittest.main() \ No newline at end of file