#!/usr/bin/env python
import immunedb.common.config as config
from immunedb.aggregation.clone_stats import run_clone_stats

if __name__ == '__main__':
    parser = config.get_base_arg_parser('Generates statistics for clones',
                                        multiproc=True)
    parser.add_argument('--clone-ids', nargs='+', type=int, help='Limit '
                        'to certain clone IDS')
    parser.add_argument('--subject-ids', nargs='+', type=int, help='Limit '
                        'to certain subjects')
    parser.add_argument('--regen', action='store_true', help='Regenerates '
                        'stats even if they already exist')
    args = parser.parse_args()
    if args.subject_ids is not None and args.clone_ids is not None:
        parser.error('May only specify subject or clone IDs')

    session = config.init_db(args.db_config)
    run_clone_stats(session, args)
