run.py

# -*- coding: latin1 -*-

import argparse
import logging

from legifrance_api.monitor import add_new_ids_to_database
from legifrance_api.content_retriever import download_juritext_info
from legifrance_api.constants import LOGGER_FILE_PATH, LOGGER_FORMAT

_logger = logging.getLogger('legifrance_api')



def _parse_args():
    parser = argparse.ArgumentParser(description = 'Script to orchestrate retrieval of jurisprudence articles from Légifrance.')
    parser.add_argument('--monitor', help = "Boolean flag activating or deactivating the search of new jurisprudence texts from the online database.",
                        action = argparse.BooleanOptionalAction, default = True)
    parser.add_argument('--content', help = "Boolean flag activating or deactivating the retrieval of identified jurisprudence texts contents from the online database.",
                        action = argparse.BooleanOptionalAction, default = True)
    parser.add_argument('--log'    , help = "Argument describing where to write logs. Value 0 disables loging. Value 1 shows logs only on console. Value 2 shows logs on console and saves them to a file too.",
                        choices = [0, 1, 2], default = 2)
    args = parser.parse_args()
    return args




if __name__ == '__main__':
    try:
        args = _parse_args()
        
        for val, handler in [(0, logging.NullHandler()), (1, logging.StreamHandler()), (2, logging.FileHandler(LOGGER_FILE_PATH, 'a'))]:
            if val <= args.log:
                handler.setFormatter(logging.Formatter(LOGGER_FORMAT))
                _logger.addHandler(handler)
        _logger.setLevel(logging.INFO)
        _logger.info(f"Starting run with the following line arguments: {args}")
    
        if args.monitor:
            add_new_ids_to_database()
        
        if args.content:
            download_juritext_info()
    
    except Exception:
        _logger.exception(f"Run stopped due to exception")
    else:
        _logger.info(f"Run completed successfully")