From b0910daa4bcfaadd750eb42c90408f3237708ec4 Mon Sep 17 00:00:00 2001 From: Franck Cuny Date: Fri, 16 Jun 2017 08:46:42 -0700 Subject: [bin] Add a script to split MH mirror set --- bin/tw-mh-split-mirrorset.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100755 bin/tw-mh-split-mirrorset.py diff --git a/bin/tw-mh-split-mirrorset.py b/bin/tw-mh-split-mirrorset.py new file mode 100755 index 0000000..4a3d228 --- /dev/null +++ b/bin/tw-mh-split-mirrorset.py @@ -0,0 +1,34 @@ +#!/usr/bin/env python + +import sys +from collections import defaultdict + + +def main(): + input = sys.argv[1] + batch_number = sys.argv[2] + mms = {} + + max_mmset_size = 0 + + with open(input, 'r') as f: + for l in f: + mm, hostname = l.rstrip().split(' ') + if mm not in mms: + mms[mm] = [] + mms[mm].append(hostname) + + max_mmset_size = max([len(mms[x]) for x in mms]) + + batches = {x: [] for x in range(0, max_mmset_size)} + for i in range(0, max_mmset_size): + for mm in mms: + if len(mms[mm]) -1 >= i: + batches[i].append(mms[mm][i]) + + for h in batches[int(batch_number)]: + print(h) + + +if __name__ == '__main__': + main() -- cgit 1.4.1