#!/usr/bin/env python
import immunedb.common.config as config
from immunedb.aggregation.collapse import run_collapse


if __name__ == '__main__':
    parser = config.get_base_arg_parser('Collapses sequences at the sample and'
                                        'subject level.', multiproc=True)
    parser.add_argument('--subject-ids', nargs='+', default=None, type=int,
                        help='Subject ID(s) to collapse.')
    args = parser.parse_args()

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