figure out permutations command line input

util/generate_permutations.py

@@ -8,27 +8,29 @@ from pprint import pprint
 from itertools import product
 import sys
 
+
 # # def parse_args():
 # parser = ArgumentParser(description='Create course permutations')
 # parser.add_argument('--fields', action='', nargs=3)
 #
-# def generate_permutations():
-#
-#     class Permutations(argparse.Action):
-#         def __call__(self, parser, args, values, option_string=None):
-#             setattr(args, self.dest, values)
-#             return Permutations
-
-
-parser = argparse.ArgumentParser()
-#parser.add_argument('--fields', action='generate_permutations', nargs=3)
-parser.add_argument('filename')
-args = parser.parse_args()
-
-# parser.register('action', 'generate_permutations', GenerationPermutations)
-with open(args.filename) as file:
-    permutation_data = json.load(file)
-    start_dates = permutation_data["permutation_data"]["start"]
-    availability = permutation_data["permutation_data"]['availability']
-    permutation_generation = [start_dates, availability]
-    print list(product(*permutation_generation))
+def generate_permutations():
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--fields', nargs=3, action="append",
+                        help="Specify which fields to generate permutations on")
+    parser.add_argument('filename')
+    args = parser.parse_args()
+
+    file = open(args.filename)
+
+
+    for permutation_choices in args.fields:
+        permutation_data = json.load(file)
+        first_field = permutation_data["permutation_data"][permutation_choices[0]]
+        second_field = permutation_data["permutation_data"][permutation_choices[1]]
+        permutation_generation = [first_field, second_field]
+        print list(product(*permutation_generation))
+
+
+if __name__ == "__main__":
+    generate_permutations()