mirror of
https://github.com/sahinakkaya/dotfiles.git
synced 2024-11-14 21:19:37 +01:00
193 lines
7.0 KiB
Python
Executable File
193 lines
7.0 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
|
||
###############################################################################
|
||
#
|
||
# Copyright 2016 - 2023, Thomas Lauf, Paul Beckingham, Federico Hernandez.
|
||
#
|
||
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
# of this software and associated documentation files (the "Software"), to deal
|
||
# in the Software without restriction, including without limitation the rights
|
||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
# copies of the Software, and to permit persons to whom the Software is
|
||
# furnished to do so, subject to the following conditions:
|
||
#
|
||
# The above copyright notice and this permission notice shall be included
|
||
# in all copies or substantial portions of the Software.
|
||
#
|
||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
||
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
||
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||
# SOFTWARE.
|
||
#
|
||
# https://www.opensource.org/licenses/mit-license.php
|
||
#
|
||
###############################################################################
|
||
|
||
import datetime
|
||
import json
|
||
import sys
|
||
|
||
from dateutil import tz
|
||
|
||
DATEFORMAT = "%Y%m%dT%H%M%SZ"
|
||
|
||
|
||
def format_seconds(seconds):
|
||
"""Convert seconds to a formatted string
|
||
|
||
Convert seconds: 3661
|
||
To formatted: " 1:01:01"
|
||
"""
|
||
hours = seconds // 3600
|
||
minutes = seconds % 3600 // 60
|
||
seconds = seconds % 60
|
||
return "{:4d}:{:02d}:{:02d}".format(hours, minutes, seconds)
|
||
|
||
|
||
def calculate_totals(input_stream):
|
||
from_zone = tz.tzutc()
|
||
to_zone = tz.tzlocal()
|
||
|
||
# Extract the configuration settings.
|
||
header = 1
|
||
configuration = dict()
|
||
body = ""
|
||
|
||
for line in input_stream:
|
||
if header:
|
||
if line == "\n":
|
||
header = 0
|
||
else:
|
||
fields = line.strip().split(": ", 2)
|
||
if len(fields) == 2:
|
||
configuration[fields[0]] = fields[1]
|
||
else:
|
||
configuration[fields[0]] = ""
|
||
else:
|
||
body += line
|
||
|
||
j = json.loads(body)
|
||
|
||
if "temp.report.start" in configuration:
|
||
report_start_utc = datetime.datetime.strptime(configuration["temp.report.start"], DATEFORMAT)
|
||
report_start_utc = report_start_utc.replace(tzinfo=from_zone)
|
||
report_start = report_start_utc.astimezone(tz=to_zone)
|
||
else:
|
||
report_start_utc = None
|
||
report_start = None
|
||
|
||
if "temp.report.end" in configuration:
|
||
report_end_utc = datetime.datetime.strptime(configuration["temp.report.end"], DATEFORMAT)
|
||
report_end_utc = report_end_utc.replace(tzinfo=from_zone)
|
||
report_end = report_end_utc.astimezone(tz=to_zone)
|
||
else:
|
||
report_end_utc = None
|
||
report_end = None
|
||
|
||
if len(j) == 0:
|
||
if report_start is not None and report_end is not None:
|
||
return ["No data in the range {:%Y-%m-%d %H:%M:%S} - {:%Y-%m-%d %H:%M:%S}".format(report_start, report_end)]
|
||
elif report_start is None and report_end is not None:
|
||
return ["No data in the range until {:%Y-%m-%d %H:%M:%S}".format(report_end)]
|
||
elif report_start is not None and report_end is None:
|
||
return ["No data in the range since {:%Y-%m-%d %H:%M:%S}".format(report_start)]
|
||
else:
|
||
return ["No data to display"]
|
||
|
||
if "start" in j[0]:
|
||
if report_start_utc is not None:
|
||
j[0]["start"] = max(report_start_utc, datetime.datetime.strptime(j[0]["start"], DATEFORMAT).replace(tzinfo=from_zone)).strftime(DATEFORMAT)
|
||
else:
|
||
report_start_utc = datetime.datetime.strptime(j[0]["start"], DATEFORMAT).replace(tzinfo=from_zone)
|
||
report_start = report_start_utc.astimezone(tz=to_zone)
|
||
else:
|
||
return ["Cannot display an past open range"]
|
||
|
||
if "end" in j[-1]:
|
||
if report_end_utc is not None:
|
||
j[-1]["end"] = min(report_end_utc, datetime.datetime.strptime(j[-1]["end"], DATEFORMAT).replace(tzinfo=from_zone)).strftime(DATEFORMAT)
|
||
else:
|
||
report_end_utc = datetime.datetime.strptime(j[-1]["end"], DATEFORMAT).replace(tzinfo=from_zone)
|
||
report_end = report_end_utc.astimezone(tz=to_zone)
|
||
else:
|
||
if report_end_utc is not None:
|
||
j[-1]["end"] = min(report_end_utc, datetime.datetime.now(tz=from_zone)).strftime(DATEFORMAT)
|
||
else:
|
||
j[-1]["end"] = datetime.datetime.now(tz=from_zone).strftime(DATEFORMAT)
|
||
report_end = datetime.datetime.now(tz=to_zone)
|
||
|
||
# Sum the seconds tracked by tag.
|
||
totals = dict()
|
||
untagged = None
|
||
|
||
for object in j:
|
||
start = datetime.datetime.strptime(object["start"], DATEFORMAT).replace(tzinfo=from_zone)
|
||
end = datetime.datetime.strptime(object["end"], DATEFORMAT).replace(tzinfo=from_zone)
|
||
|
||
tracked = end - start
|
||
|
||
if "tags" not in object or object["tags"] == []:
|
||
if untagged is None:
|
||
untagged = tracked
|
||
else:
|
||
untagged += tracked
|
||
else:
|
||
for tag in object["tags"]:
|
||
if tag in totals:
|
||
totals[tag] += tracked
|
||
else:
|
||
totals[tag] = tracked
|
||
|
||
# Determine largest tag width.
|
||
max_width = len("Total")
|
||
for tag in totals:
|
||
if len(tag) > max_width:
|
||
max_width = len(tag)
|
||
|
||
# Compose report header.
|
||
output = [
|
||
"",
|
||
"Total by Tag, for {:%Y-%m-%d %H:%M:%S} - {:%Y-%m-%d %H:%M:%S}".format(report_start, report_end),
|
||
""
|
||
]
|
||
|
||
# Compose table header.
|
||
if configuration["color"] == "on":
|
||
output.append("[4m{:{width}}[0m [4m{:>10}[0m".format("Tag", "Total", width=max_width))
|
||
else:
|
||
output.append("{:{width}} {:>10}".format("Tag", "Total", width=max_width))
|
||
output.append("{} {}".format("-" * max_width, "----------"))
|
||
|
||
# Compose table rows.
|
||
grand_total = 0
|
||
for tag in sorted(totals):
|
||
seconds = int(totals[tag].total_seconds())
|
||
formatted = format_seconds(seconds)
|
||
grand_total += seconds
|
||
output.append("{:{width}} {:10}".format(tag, formatted, width=max_width))
|
||
|
||
if untagged is not None:
|
||
seconds = int(untagged.total_seconds())
|
||
formatted = format_seconds(seconds)
|
||
grand_total += seconds
|
||
output.append("{:{width}} {:10}".format("", formatted, width=max_width))
|
||
|
||
# Compose total.
|
||
if configuration["color"] == "on":
|
||
output.append("{} {}".format(" " * max_width, "[4m [0m"))
|
||
else:
|
||
output.append("{} {}".format(" " * max_width, "----------"))
|
||
|
||
output.append("{:{width}} {:10}".format("Total", format_seconds(grand_total), width=max_width))
|
||
output.append("")
|
||
|
||
return output
|
||
|
||
|
||
if __name__ == "__main__":
|
||
for line in calculate_totals(sys.stdin):
|
||
print(line)
|