1 files changed, 40 insertions, 0 deletions
diff --git a/s3p/generate_perf.py b/s3p/generate_perf.py
new file mode 100644
index 00000000..dfd62fc8
--- /dev/null
+++ b/s3p/generate_perf.py
@@ -0,0 +1,40 @@
+import argparse
+import csv
+import os
+import shutil
+import sys
+import subprocess
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(description='Prepare CSAR for policy distrition performance test')
+    parser.add_argument('--total', required=True, type=int, help='total number of CSAR to be generated')
+    parser.add_argument('--dest', required=True, help='dest directory where the CSAR files will be stored')
+    parser.add_argument('--out',  default='perf_data.csv', help='list of generated CSAR identifier')
+    return parser.parse_args(sys.argv[1:])
+
+
+def main():
+    args = parse_args()
+    
+    # create dest dir
+    shutil.rmtree(args.dest, ignore_errors=True)
+    os.makedirs(args.dest)
+
+    # prepartion
+    count = 0
+    maxwidth = len(str(args.total))
+    scripts = os.path.dirname(os.path.abspath(__file__))
+    scripts = os.path.join(scripts, 'generate.sh')
+    
+    with open(args.out, 'w') as out_file:
+        out_writer = csv.writer(out_file, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL)
+        while (count < args.total):
+            seed = str(count).zfill(maxwidth)
+            subprocess.check_call([scripts, args.dest, seed])
+            out_writer.writerow(["s3p_" + seed])
+            count += 1
+    return 0
+
+if __name__ == '__main__':
+    main()