gengen

#!/usr/bin/env python
from __future__ import print_function
import random
import docopt

DOC = """
USAGE:
    gengen <length> ...

Give a length for each chromosome you want
"""

def printwrapped(string, width=80):
    for start in range(0, len(string), width):
        print(string[start:start+width])


def gen_nts(bp, alpha='ACGT'):
    nts = []
    for _ in range(bp):
        nts.append(random.choice(alpha))
    return ''.join(nts)


def main():
    opts = docopt.docopt(DOC)
    for idx, chr_len in enumerate(opts['<length>']):
        print('>{}'.format(idx))
        printwrapped(gen_nts(int(chr_len)))

if __name__ == '__main__':
    main()