nibg_analysis.py

import datetime
import json
import logging
import os
from argparse import ArgumentParser

from ruamel.yaml import CommentedMap

from analysis.config import load_config
from analysis.shared_parsers import to_pruned_sorted_quarterly, plot_counts


def main(config: CommentedMap) -> int:
    start = datetime.datetime.now()
    nibg_cfg = config['data']['nibg']

    aggregate_stats_path = os.path.join(nibg_cfg['json_output_dir'], 'nibg_aggregate_stats.json')
    with open(aggregate_stats_path, 'rt') as f:
        aggregate_stats = json.loads(f.read())

    quarterly_counts = to_pruned_sorted_quarterly(aggregate_stats)
    plot_counts(quarterly_counts, nibg_cfg)

    end = datetime.datetime.now()
    logging.info(f'Script took {end - start}')

    return 0


if __name__ == '__main__':
    parser = ArgumentParser('Performs the Nederlands Instituut voor Beeld en Geluid metadata analysis')
    parser.add_argument('-c', '--config', default='config.yaml')

    args = parser.parse_args()
    config = load_config(args.config)
    raise SystemExit(main(config))