imagelot_cli.py

#!/usr/bin/env python3
#coding: utf8
"""
Module principal permettant l'utilisation d'ImageLot en CLI

Dépendances externes : argparse, os, sys

"""
import argparse
import os
import sys
from imagelot_process import process_json, batch_processing

def main_cli():
    """
    Fonction permettant de lancer ImageLot en CLI.
    """
    # Parsage des arguments
    parser = argparse.ArgumentParser()
    parser.add_argument("f",
                        nargs="*",
                        help="Photos à traiter")
    parser.add_argument("-p",
                        "--parameters",
                        nargs="?",
                        default=None,
                        required=True,
                        help="Fichier de paramètres de traitement par lot au format JSON.")
    parser.add_argument("-d",
                        "--dest",
                        nargs="?",
                        default=None,
                        help="Répertoire de destination.")
    args = parser.parse_args()

    # Sanitize destination directory link (from the ending "/")
    args.dest.strip("/")

    # Vérifications des paramètres
    if args.parameters.split('.')[-1] != "json" and \
        args.parameters.split('.')[-1] != "JSON" and \
        not os.path.isfile(args.parameters):
        sys.stderr.write("Le fichier de paramètres n'est pas au format JSON.\n")
        sys.exit(1)

    if not os.path.isdir(args.dest):
        sys.stderr.write("Le répertoire de destination spécifié n'est pas valide.\n")
        sys.exit(1)

    # On ouvre le fichier des paramèrtes puis on lance le traitement par lot
    try:
        parameters_file = open(args.parameters, 'r')
    except IOError:
        sys.stderr.write("Impossible d'ouvrir le fichier {} en lecture.\n".format(args.parameters))
        sys.exit(1)

    parameters = process_json(parameters_file)
    parameters_file.close()
    response = batch_processing(args.f, parameters, args.dest)
    print(response)


# Si l'on exécute le fichier, on lance la fonction main_gui
if __name__ == '__main__':
    main_cli()