import argparse

from . import parse

from . import analysis


LOC="uniprot_receptor.xml.gz"

def dump(args):
    for record in parse.uniprot_seqrecords(LOC):
        print(record)



def names(args):
    for record in parse.uniprot_seqrecords(LOC):
        print(record.name)



def average(args):
    print("Average Length is {}".format(
        analysis.average_len(parse.uniprot_seqrecords(LOC))))






def cli():

    # create a new parser

    parser  = argparse.ArgumentParser(prog="uniplot")

    subparsers = parser.add_subparsers(help="Sub Command Help")

    ## add subparsers

    subparsers.add_parser("dump").set_defaults(func=dump)
    subparsers.add_parser("list").set_defaults(func=names)
    subparsers.add_parser("average").set_defaults(func=average)