#!/usr/bin/env python
# -*- coding: utf-8

import sys

import anvio
import anvio.dbops as dbops
import anvio.terminal as terminal

from anvio.errors import ConfigError, FilesNPathsError


__author__ = "A. Murat Eren"
__copyright__ = "Copyright 2015, The anvio Project"
__credits__ = []
__license__ = "GPL 3.0"
__version__ = anvio.__version__
__maintainer__ = "A. Murat Eren"
__email__ = "a.murat.eren@gmail.com"


run = terminal.Run()
progress = terminal.Progress()


if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser(description='Generate a new anvio contigs database.')

    parser.add_argument(*anvio.A('contigs-fasta'), **anvio.K('contigs-fasta'))
    parser.add_argument(*anvio.A('split-length'), **anvio.K('split-length'))
    parser.add_argument(*anvio.A('kmer-size'), **anvio.K('kmer-size'))
    parser.add_argument(*anvio.A('output-db-path'), **anvio.K('output-db-path', {'default': 'CONTIGS.db'}))

    args = parser.parse_args()

    try:
        a = dbops.ContigsDatabase(args.output_db_path, run, progress, quiet=False)
        a.create(args.contigs_fasta, args.split_length, args.kmer_size)
    except ConfigError, e:
        print e
        sys.exit(-1)
    except FilesNPathsError, e:
        print e
        sys.exit(-2)
