#!/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("{:{width}} {:>10}".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, " ")) 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)