print-results.py

#! /usr/bin/python3

# _*_ coding: utf-8 _*_

"""
@file 		print-results.py
@author 	Simon Burkhardt - github.com/mnemocron
@date 		2018.10.10

This file prints the resulting numbers to the command line

Post about this code:
https://www.reddit.com/r/LongDistance/comments/9mgcol/oc_chat_statistics_from_telegram_using_python/

Inspiration:
https://www.reddit.com/r/LongDistance/comments/9jud8j/analysis_of_texts_from_a_long_distance/
"""

from __future__ import print_function

import sys
import os
import optparse
import json
from pprint import pprint

parser = optparse.OptionParser("telegram-stats")
parser.add_option(
    "-i", "--input-file", dest="indir", type="string", help="chat history file"
)
(opts, args) = parser.parse_args()


def load_file_to_raw(path):
    try:
        with open(path, encoding="utf-8-sig") as fh:
            data = json.load(fh)
        return data
    except IOError:
        print("Error: could not open the file")
        exit(-1)


### MAIN
def main():
    if opts.indir is None:
        parser.print_help()
        exit(0)
    metrics = load_file_to_raw(opts.indir)
    print("[ name: " + metrics["A"]["name"] + "]")
    print("total message count:     \t" + str(metrics["A"]["total_messages"]))
    print("total word count:        \t" + str(metrics["A"]["total_words"]))
    print("total character count:   \t" + str(metrics["A"]["total_chars"]))
    print("average word count:      \t" + str(metrics["A"]["avg_words"]))
    print("total unique words:      \t" + str(metrics["A"]["unique_words"]))
    print("average character count: \t" + str(metrics["A"]["avg_chars"]))
    print("used markdown in:        \t" + str(metrics["A"]["urls"]) + " messages")
    print("total urls in messages:  \t" + str(metrics["A"]["urls"]))
    print("total photos:            \t" + str(metrics["A"]["photo"]))
    for key in metrics["A"]["media"]:
        print("total " + str(key) + " count: \t\t" + str(metrics["A"]["media"][key]))

    print("")
    print("[ name: " + metrics["B"]["name"] + "]")
    print("total message count:     \t" + str(metrics["B"]["total_messages"]))
    print("total word count:        \t" + str(metrics["B"]["total_words"]))
    print("total character count:   \t" + str(metrics["B"]["total_chars"]))
    print("average word count:      \t" + str(metrics["B"]["avg_words"]))
    print("total unique words:      \t" + str(metrics["B"]["unique_words"]))
    print("average character count: \t" + str(metrics["B"]["avg_chars"]))
    print("used markdown in:        \t" + str(metrics["B"]["urls"]) + " messages")
    print("total urls in messages:  \t" + str(metrics["B"]["urls"]))
    print("total photos:            \t" + str(metrics["B"]["photo"]))
    for key in metrics["B"]["media"]:
        print("total " + str(key) + " count: \t\t" + str(metrics["B"]["media"][key]))

    print("")
    print("[ combined stats ]")
    print("total message count:     \t" + str(metrics["total"]))


if __name__ == "__main__":
    try:
        main()
    except KeyboardInterrupt as e:
        print("Aborted by KeyboardInterrupt")
        exit(0)