First commit, working CLI

report.py

@@ -0,0 +1,395 @@
+#!/usr/bin/env python3
+import numpy as np
+from matplotlib import pyplot as plt
+import json
+import uuid
+import gettext
+from datetime import date, datetime, timedelta
+from babel.dates import format_date
+import locale
+from os.path import isfile
+from tempfile import mkdtemp
+from shutil import copyfile, rmtree
+from subprocess import run
+
+
+def i18n(language):
+    try:
+        lang = gettext.translation(
+            'report', localedir='locales', languages=[language])
+        lang.install()
+    except FileNotFoundError:
+        gettext.install('report')
+
+
+def bar(data, filename):
+    '''
+    Draw a bar chart from a documented JSON
+    '''
+    font = {'fontname': 'DejaVu Sans', 'size': 10, 'weight': 'normal'}
+    x_location = np.arange(len(data['data']))
+    fig = plt.figure(figsize=(data['width']/2.54, data['height']/2.54))
+    bars = plt.bar(x_location, data['data'],
+                   data['bar_width'], color=data['bar_color'])
+    plt.ylabel(data['Y_label'], **font, fontsize='x-small')
+    plt.title(data['title'], **font)
+    plt.xticks(x_location, data['data_labels'], **font, fontsize='x-small', rotation=45)
+    plt.yticks(**font, fontsize='x-small')
+    plt.ylim(0, max(data['data'])*1.15)
+    for bar in bars:
+        height = bar.get_height()
+        plt.text(bar.get_x() + bar.get_width()/2, height+(max(data['data'])/60), "{}\n({}%)".format(int(height), float(round(height/sum(data['data'])*100, 1))), ha='center', va='bottom', **font, fontsize='xx-small')
+    plt.tight_layout()
+    plt.savefig("{}".format(filename), dpi=300)
+    plt.close(fig)
+
+
+def pie(data, filename):
+    '''
+    Draw a pie chart
+    '''
+    font = {'fontname': 'DejaVu Sans', 'size': 10, 'weight': 'normal'}
+    fig = plt.figure(figsize=(data['width']/2.54, data['height']/2.54))
+    chart, texts, autotexts = plt.pie(data['parts'], labels=data['labels'],
+            colors=data['colors'], autopct='%1.1f%%', explode=data['explode'], startangle=180)
+    plt.title(data['title'], **font)
+    for text in texts:
+        text.set_size('x-small')
+    for text in autotexts:
+        text.set_size('xx-small')
+    plt.savefig("{}".format(filename), dpi=300)
+    plt.close(fig)
+
+
+def format_shutter_speed(speed):
+    speed = speed/1000
+    if 1/speed >= 4:
+        shutter_speed = "1/{}".format(int(round(1/speed, 0)))
+    else:
+        if speed < 60:
+            shutter_speed = "{}'".format(round(speed, 3))
+        else:
+            speed_minutes = speed//60
+            speed_seconds = speed%60
+            shutter_speed = "{}\"{}'".format(speed_minutes, speed_seconds)
+    return shutter_speed
+
+
+def title():
+    '''
+    Juste the title
+    '''
+    text = _("Statistical photo report")
+    text += "\n"
+    text += "#"*80
+    text += "\n\n"
+    return text
+
+
+def general(informations, language, colors, temp_dir):
+    '''
+    General informations. Take a dict() in argument.
+    '''
+    dimensions = []
+    for dimension in informations['dimensions'].keys():
+        dimensions.append(float(dimension))
+    text = _("Pictures informations")
+    text += "\n"
+    text += "="*80
+    text += "\n"
+    informations_total = informations['total']
+    start_date = datetime.strptime(sorted(informations['date'])[0], "%Y%m%d").date()
+    end_date = datetime.strptime(sorted(informations['date'])[-1], "%Y%m%d").date()
+    most_productive = datetime.strptime(sorted(informations['date'], key=informations['date'].get)[-1], "%Y%m%d").date()
+    if informations_total > 1:
+        text += _("The set contains {} pictures, ").format(informations['total'])
+    elif informations_total == 1:
+        text += _("The set contains only one picture, ").format(informations['total'])
+    else:
+        text += _("The set is empty. ")
+    if start_date != end_date:
+        text += _("taken from {} to {}. ").format(
+            format_date(start_date, locale=language, format="medium"),
+            format_date(end_date, locale=language, format="medium"))
+        text += _("The most productive day was {} ({} shots). ").format(
+            format_date(most_productive, locale=language, format="long"),
+            informations['date'][sorted(informations['date'], key=informations['date'].get)[-1]])
+    else:
+        text += _("taken on {}. ").format(start_date)
+    text += "\n\n"
+    if informations_total > 1:
+        text += _("Pictures sizes are from {1}{0}Mpx to {2}{0}Mpx, mostly {3}{0}Mpx. ").format(
+            b'\xc2\xa0'.decode(),
+            sorted(dimensions)[0],
+            sorted(dimensions)[-1],
+            sorted(informations['dimensions'], key=informations['dimensions'].get)[-1])
+    else:
+        text += _("Picture size is {}{}Mpx. ").format(sorted(dimensions)[0], b'\xc2\xa0'.decode())
+    text += "\n\n"
+    if start_date != end_date:
+        dates = {datetime.strptime(k, "%Y%m%d"):int(v) for k,v in informations['date'].items()}
+        top_date = sorted(dates, key=dates.get, reverse=True)[0:20]
+        new_date = {}
+        for date in top_date:
+            new_date.update({date: dates[date]})
+        bar(
+                {
+                    'width': 16,
+                    'height': 12,
+                    'data': [new_date[date] for date in sorted(new_date)],
+                    'data_labels': [format_date(date, locale=language, format="short") for date in sorted(new_date)],
+                    'title': _("Distribution of shots by date{}").format(_(" : top 20") if len(top_date) == 20 else ""),
+                    'Y_label': _("Number of shots"),
+                    'bar_color': colors[2],
+                    'bar_width': 0.8
+                },
+                temp_dir + "/picnb_datetime.pdf")
+        text += ".. figure:: picnb_datetime.pdf\n\t:width: 16cm\n"
+    return text
+
+
+def camera_and_lens(informations, language, colors, temp_dir):
+    '''
+    Informations about cameras and lenses used.
+    '''
+    with open("helpers/dictionary.json") as file:
+        helper = json.load(file)
+    text = _("Cameras & lenses")
+    text += "\n"
+    text += "="*80
+    text += "\n\n"
+    top_couple = sorted(informations['cameras+lenses'], key=informations['cameras+lenses'].get)[-1].split(" + ")
+    top_couple_camera, top_couple_lens = top_couple[0], top_couple[1]
+    top_lens = sorted(informations['lenses'], key=informations['lenses'].get)[-1]
+    top_camera = sorted(informations['cameras'], key=informations['cameras'].get)[-1]
+    text += _("The most commonly used couple is {} with {}. ").format(
+            (helper['camera'][top_couple_camera]['brand'] + " " + helper['camera'][top_couple_camera]['name']) if top_couple_camera in helper['camera'] else top_couple_camera,
+            (helper['lens'][top_couple_lens]['brand'] + " " + helper['lens'][top_couple_lens]['name']) if top_couple_lens in helper['lens'] else top_couple_lens,)
+    text += "\n\n"
+    if len(informations['lenses']) > 1:
+        text += _("There are {} lenses, the most used is the {}. ").format(
+                len(informations['lenses']),
+                (helper['lens'][top_lens]['brand'] + " " + helper['lens'][top_lens]['name']) if top_lens in helper['lens'] else top_lens)
+    else:
+        text += _("The lens is a {}. ").format(
+            (helper['lens'][top_lens]['brand'] + " " + helper['lens'][top_lens]['name']) if top_lens in helper['lens'] else top_lens)
+    text += _("The favorite aperture is {}{}, ").format(
+            _("F"),
+            sorted(informations['apertures'], key=informations['apertures'].get)[-1])
+    text += _("and the favorite focal lenght is {}mm. ").format(
+            sorted(informations['focals'], key=informations['focals'].get)[-1])
+    text += "\n\n"
+    if len(informations['cameras']) > 1:
+        text += _("There are {} cameras, the mose used is the {}. ").format(
+                len(informations['cameras']),
+                (helper['camera'][top_camera]['brand'] + " " + helper['camera'][top_camera]['name']) if top_camera in helper['camera'] else top_camera)
+    else:
+        text += _("The camera is a {}. ").format(
+            (helper['camera'][top_camera]['brand'] + " " + helper['camera'][top_camera]['name']) if top_camera in helper['camera'] else top_camera)
+    text += _(" The favorite ISO is {}. ").format(
+            sorted(informations['isos'], key=informations['isos'].get)[-1])
+    text += "\n\n"
+    if isfile("helpers/cameras/{}.jpg".format(top_camera.replace(" ", "_").replace("/", "_"))):
+        copyfile(
+                "helpers/cameras/{}.jpg".format(top_camera.replace(" ", "_").replace("/", "_")),
+                "{}/{}.jpg".format(temp_dir, top_camera.replace(" ", "_").replace("/", "_")))
+        text += ".. image:: {}.jpg\n\t:width: 8cm\n\t:align: left\n".format(top_camera.replace(" ", "_").replace("/", "_"))
+    if isfile("helpers/lenses/{}.jpg".format(top_lens.replace(" ", "_").replace("/", "_"))):
+        copyfile(
+                "helpers/lenses/{}.jpg".format(top_lens.replace(" ", "_").replace("/", "_")),
+                "{}/{}.jpg".format(temp_dir, top_lens.replace(" ", "_").replace("/", "_")))
+        text += ".. image:: {}.jpg\n\t:width: 8cm\n\t:align: right\n".format(top_lens.replace(" ", "_").replace("/", "_"))
+    if len(informations['cameras']) > 1:
+        pie(
+                {
+                    'width': 17,
+                    'height': 8,
+                    'parts': [informations['cameras'][camera] for camera in sorted(informations['cameras'], key=informations['cameras'].get)],
+                    'labels': [((helper['camera'][camera]['brand'] + " " + helper['camera'][camera]['name']) if camera in helper['camera'] else camera) for camera in sorted(informations['cameras'], key=informations['cameras'].get)],
+                    'colors': colors[0:len(list(informations['cameras'].keys()))],
+                    'explode': [0.02]*len(list(informations['cameras'].keys())),
+                    'title': _("Cameras")
+                },
+                temp_dir + "/cameras_parts.pdf")
+        text += "\n\n"
+        text += ".. figure:: cameras_parts.pdf\n\t:width: 17cm\n"
+    if len(informations['lenses']) > 1:
+        pie(
+                {
+                    'width': 17,
+                    'height': 8,
+                    'parts': [informations['lenses'][lens] for lens in sorted(informations['lenses'], key=informations['lenses'].get)],
+                    'labels': [((helper['lens'][lens]['brand'] + " " + helper['lens'][lens]['name']) if lens in helper['lens'] else lens) for lens in sorted(informations['lenses'], key=informations['lenses'].get)],
+                    'colors': colors[0:len(list(informations['lenses'].keys()))],
+                    'explode': [0.02]*len(list(informations['lenses'].keys())),
+                    'title': _("Lenses")
+                },
+                temp_dir + "/lenses_parts.pdf")
+        text += "\n\n"
+        text += ".. figure:: lenses_parts.pdf\n\t:width: 17cm\n"
+    return text
+
+
+def focal_aperture(informations, language, colors, temp_dir):
+    text = _("Focal lengths & apertures")
+    text += "\n" + "="*80 + "\n"
+    text += "\n\n"
+    text += _("Focal lenghts")
+    text += "\n" + "-"*80 + "\n"
+    if len(informations['focals']) > 1:
+        focals = {float(k): int(v) for k,v in informations['focals'].items()}
+        top_focals = sorted(focals, key=focals.get, reverse=True)[0:20]
+        new_focals = {}
+        for focal in top_focals:
+            new_focals.update({focal: focals[focal]})
+        bar(
+                {
+                    'width': 17,
+                    'height': 8,
+                    'data': [new_focals[focal] for focal in sorted(new_focals)],
+                    'bar_width': 0.8,
+                    'bar_color': colors[0],
+                    'data_labels': ["{} mm".format(int(focal) if int(focal) == float(focal) else float(focal)) for focal in sorted(new_focals)],
+                    'Y_label': _("Number of shots"),
+                    'title': _("Focals{}").format(_(" : top 20") if len(top_focals) == 20 else "")
+                },
+                temp_dir + "/focal_lenghts.pdf")
+        text += ".. figure:: focal_lenghts.pdf\n\t:width: 17cm\n"
+    else:
+        text += _("Only one focal lenght is used : {}mm. ").format(float(sorted(informations['focals'], key=informations['focals'].get)[0]))
+    text += "\n\n"
+    text += _("Apertures")
+    text += "\n" + "-"*80 + "\n"
+    if len(informations['apertures']) > 1:
+        apertures = {float(k):int(v) for k,v in informations['apertures'].items()}
+        top_apertures = sorted(apertures, key=apertures.get, reverse=True)[0:20]
+        new_apertures = {}
+        for aperture in top_apertures:
+            new_apertures.update({aperture: apertures[aperture]})
+        bar(
+                {
+                    'width': 17,
+                    'height': 8,
+                    'data': [new_apertures[aperture] for aperture in sorted(new_apertures)],
+                    'bar_width': 0.8,
+                    'bar_color': colors[5],
+                    'data_labels': ["{}{}".format(_("F"), int(aperture) if int(aperture) == float(aperture) else float(aperture)) for aperture in sorted(new_apertures)],
+                    'Y_label': _("Number of shots"),
+                    'title': _("Apertures{}").format(_(" : top 20") if len(top_apertures) == 20 else "")
+                },
+                temp_dir + "/apertures.pdf")
+        text += ".. figure:: apertures.pdf\n\t:width: 17cm\n"
+    else:
+        text += _("Only one aperture is used : {}{}. ").format(
+                    _("F"),
+                    float(sorted(informations['apertures'], key=informations['apertures'].get)[0]))
+    text += "\n\n"
+    return text
+
+def iso_shutter(informations, language, colors, temp_dir):
+    text = _("ISO sensitivities & shutter speeds")
+    text += "\n" + "="*80 + "\n"
+    text += _("ISO sensitivities")
+    text += "\n" + "-"*80 + "\n"
+    if len(informations['isos']) > 1:
+        isos = {float(k):int(v) for k,v in informations['isos'].items()}
+        top_isos = sorted(isos, key=isos.get, reverse=True)[0:20]
+        new_isos = {}
+        for iso in top_isos:
+            new_isos.update({iso: isos[iso]})
+        bar(
+                {
+                    'width': 17,
+                    'height': 8,
+                    'data': [new_isos[iso] for iso in sorted(new_isos)],
+                    'bar_width': 0.8,
+                    'bar_color': colors[3],
+                    'data_labels': ["ISO {}".format(int(iso) if int(iso) == float(iso) else float(iso)) for iso in sorted(new_isos)],
+                    'Y_label': _("Number of shots"),
+                    'title': _("ISO sensibilities{}").format(_(" : top 20") if len(top_isos) == 20 else "")
+                },
+                temp_dir + "/isos.pdf")
+        text += ".. figure:: isos.pdf\n\t:width: 17cm\n"
+    else:
+        text += _("All shots at the same sensitivity : ISO {} ").format(sorted(informations['isos'], key=informations['isos'].get)[0])
+    text += "\n\n"
+    text += _("Shutter speeds")
+    text += "\n" + "-"*80 + "\n"
+    if len(informations['shutter_speeds']) > 1:
+        shutter_speeds = {float(k):int(v) for k,v in informations['shutter_speeds'].items()}
+        top_shutter_speeds = sorted(shutter_speeds, key=shutter_speeds.get, reverse=True)[0:20]
+        new_shutter_speeds = {}
+        for shutter_speed in top_shutter_speeds:
+            new_shutter_speeds.update({shutter_speed: shutter_speeds[shutter_speed]})
+        bar(
+                {
+                    'width': 17,
+                    'height': 8,
+                    'data': [new_shutter_speeds[shutter_speed] for shutter_speed in sorted(new_shutter_speeds)],
+                    'bar_width': 0.8,
+                    'bar_color': colors[7],
+                    'data_labels': [format_shutter_speed(shutter_speed) for shutter_speed in sorted(new_shutter_speeds)],
+                    'Y_label': _("Number of shots"),
+                    'title': _("Shutter speeds{}").format(_(" : top 20") if len(top_shutter_speeds) == 20 else "")
+                },
+                temp_dir + "/speeds.pdf")
+        text += ".. figure:: speeds.pdf\n\t:width: 17cm\n"
+    else:
+        text += _("All shots at the same speed : {}. ").format(format_shutter_speed(sorted(informations['shutter_speeds'], key=informations['shutter_speeds'].get)[0]))
+    return text
+
+
+def footer():
+    return ".. footer::\n\n\t###Page###\n\n"
+
+
+def page_break():
+    return ".. raw:: pdf\n\n\tPageBreak\n\n"
+
+
+def create(informations, language, output_file):
+    with open("helpers/dictionary.json") as file:
+        helper = json.load(file)
+    colors = [
+            "#b58900", # yellow
+            "#cb4b16", # orange
+            "#dc322f", # red
+            "#d33682", # magenta
+            "#6c71c4", # violet
+            "#268bd2", # blue
+            "#2aa198", # cyan
+            "#859900" # green
+            ]
+    i18n(language)
+    temp_dir = mkdtemp()
+    document = footer()
+    document += title()
+    document += general(informations, language, colors, temp_dir)
+    document += page_break()
+    document += camera_and_lens(informations, language, colors, temp_dir)
+    document += page_break()
+    document += focal_aperture(informations, language, colors, temp_dir)
+    document += page_break()
+    document += iso_shutter(informations, language, colors, temp_dir)
+    with open(temp_dir + "/report.rst", "w") as rst_file:
+        rst_file.write(document)
+    try:
+        run(["rst2pdf", temp_dir + "/report.rst", "-s", "rst2pdf_stylsheet.style", "--use-floating-images", "-o", output_file], check=True)
+        rmtree(temp_dir)
+        return True
+    except CalledProcessError:
+        return False
+
+
+if __name__ == '__main__':
+    import argparse
+    import json
+    parser = argparse.ArgumentParser()
+    parser.add_argument("infile", help="JSON output from a reader")
+    parser.add_argument("--lang", help="i18n option", default="en")
+    parser.add_argument("--out", "-o", help="outfile", default="./report.pdf")
+    args = parser.parse_args()
+    with open(args.infile, "r") as json_data:
+        data = json.load(json_data)
+    create(data, args.lang, args.out)
+