699
|
1 #!/usr/bin/env python
|
|
2 # -*- coding: utf-8 -*-
|
|
3
|
|
4 """
|
|
5 slice a arbitrarily sized list
|
|
6 """
|
|
7
|
|
8 # imports
|
|
9 import argparse
|
|
10 import os
|
|
11 import subprocess
|
|
12 import sys
|
|
13
|
|
14 # module globals
|
700
|
15 __all__ = ['slice', 'main', 'Parser']
|
699
|
16
|
|
17 def slice(container, n_chunks):
|
700
|
18 size = int(len(container)/(n_chunks-1))
|
|
19 retval = []
|
|
20 start = 0
|
|
21 for i in range(n_chunks-1):
|
|
22 retval.append(container[start:start+size])
|
|
23 start += size
|
|
24 retval.append(container[start:])
|
|
25 return retval
|
699
|
26
|
|
27 class Parser(argparse.ArgumentParser):
|
|
28 """CLI option parser"""
|
|
29 def __init__(self, **kwargs):
|
|
30 kwargs.setdefault('description', __doc__)
|
|
31 argparse.ArgumentParser.__init__(self, **kwargs)
|
|
32 self.add_argument('N', type=int,
|
|
33 help="number of chunks")
|
700
|
34 self.add_argument('-M', '--len', dest='length', type=int, default=29,
|
|
35 help="length of list [DEFAULT: %(default)s]")
|
699
|
36 self.options = None
|
|
37
|
|
38 def parse_args(self, *args, **kw):
|
|
39 options = argparse.ArgumentParser.parse_args(self, *args, **kw)
|
|
40 self.validate(options)
|
|
41 self.options = options
|
|
42 return options
|
|
43
|
|
44 def validate(self, options):
|
|
45 """validate options"""
|
|
46
|
|
47 def main(args=sys.argv[1:]):
|
|
48 """CLI"""
|
|
49
|
|
50 # parse command line options
|
|
51 parser = Parser()
|
|
52 options = parser.parse_args(args)
|
|
53
|
700
|
54 # generate list
|
|
55 seq = range(options.length)
|
|
56
|
|
57 # chunk list
|
|
58 output = slice(seq, options.N)
|
|
59
|
|
60 # print output
|
|
61 for chunk in output:
|
|
62 print (",".join([str(i) for i in chunk]))
|
|
63
|
699
|
64 if __name__ == '__main__':
|
|
65 main()
|
|
66
|