gave a name to main script

main.py → swf2sessions.py

 #!/usr/bin/env python3
 
-import argparse, json, re
+import argparse, json, re, os.path, sys
 from user_session_builder import User
 from workload import SwfField, Job
 
 
-def main(input_swf, output_folder, delim_approach, delim_threshold,
+def swf2sessions(input_swf, output_folder, delim_approach, delim_threshold,
          dynamic_reduction, build_graph_rep, quiet):
     users = {}
-        
+
+    # if not os.path.exists(input_swf):
+    #     raise FileExistsError(f"Input file '{input_swf}' does not exist")
+
+    if not os.path.exists(output_folder):
+        raise FileExistsError(f"Output folder '{output_folder}' does not exist")
+
     # Read SWF
     element = '([-+]?\d+(?:\.\d+)?)'
     r = re.compile('\s*' + (element + '\s+') * 17 + element + '\s*')
@@ -56,64 +62,64 @@ def main(input_swf, output_folder, delim_approach, delim_threshold,
         print("Number of sessions: ", sum([len(u.sessions) for u in users.values()]))
         print(f"The output files have been stored in the folder {output_folder}")
 
-
-parser = argparse.ArgumentParser(description='TODO')
-parser.add_argument('input_swf',
-                    type=argparse.FileType('r'),
-                    help='The input SWF file')
-parser.add_argument('output_folder',
-                    type=str,
-                    help='The folder that will store the output files')
-
-group = parser.add_mutually_exclusive_group(required=True)
-group.add_argument('-a',
-                   '--arrival',
-                   help='Arrival delimitation approach',
-                   action="store_true")
-group.add_argument('-l',
-                   '--last',
-                   help='Last delimitation approach',
-                   action="store_true")
-group.add_argument('-m',
-                   '--max',
-                   help='Max delimitation approach',
-                   action="store_true")
-parser.add_argument(
-    '-t',
-    '--threshold',
-    type=int,
-    help=
-    'Unit: minutes. The threshold (on think-time or inter-arrival time, depending on the delimiation approach) above which a job will be considered to be in a new session.'
-)
-
-parser.add_argument(
-    '--no_dynamic_reduction',
-    action="store_true",
-    help=
-    'Unless this option is specified, during the construction of the graph the algorithm dynamically avoids to add an edge between two nodes if a path already exists.'
-)
-parser.add_argument(
-    '--graph',
-    action="store_true",
-    help=
-    "Build a graphical representation of each session graph and save it in the output folder as a gml file"
-)
-
-parser.add_argument("-q", "--quiet", action="store_true")
-
-args = parser.parse_args()
-
-if args.last:
-    delim = 'last'
-elif args.max:
-    delim = 'max'
-else:
-    delim = 'arrival'
-
-main(input_swf=args.input_swf,
-     output_folder=args.output_folder,
-     delim_approach=delim,
-     delim_threshold=args.threshold,
-     dynamic_reduction=not (args.no_dynamic_reduction),
-     build_graph_rep=args.graph,
-     quiet=args.quiet)
+if __name__== "__main__":
+    parser = argparse.ArgumentParser(description='TODO')
+    parser.add_argument('input_swf',
+                        type=argparse.FileType('r'),
+                        help='The input SWF file')
+    parser.add_argument('output_folder',
+                        type=str,
+                        help='The folder that will store the output files')
+
+    group = parser.add_mutually_exclusive_group(required=True)
+    group.add_argument('-a',
+                    '--arrival',
+                    help='Arrival delimitation approach',
+                    action="store_true")
+    group.add_argument('-l',
+                    '--last',
+                    help='Last delimitation approach',
+                    action="store_true")
+    group.add_argument('-m',
+                    '--max',
+                    help='Max delimitation approach',
+                    action="store_true")
+    parser.add_argument(
+        '-t',
+        '--threshold',
+        type=int,
+        help=
+        'Unit: minutes. The threshold (on think-time or inter-arrival time, depending on the delimiation approach) above which a job will be considered to be in a new session.'
+    )
+
+    parser.add_argument(
+        '--no_dynamic_reduction',
+        action="store_true",
+        help=
+        'Unless this option is specified, during the construction of the graph the algorithm dynamically avoids to add an edge between two nodes if a path already exists.'
+    )
+    parser.add_argument(
+        '--graph',
+        action="store_true",
+        help=
+        "Build a graphical representation of each session graph and save it in the output folder as a gml file"
+    )
+
+    parser.add_argument("-q", "--quiet", action="store_true")
+
+    args = parser.parse_args()
+
+    if args.last:
+        delim = 'last'
+    elif args.max:
+        delim = 'max'
+    else:
+        delim = 'arrival'
+
+    swf2sessions(input_swf=args.input_swf,
+        output_folder=args.output_folder,
+        delim_approach=delim,
+        delim_threshold=args.threshold,
+        dynamic_reduction=not (args.no_dynamic_reduction),
+        build_graph_rep=args.graph,
+        quiet=args.quiet)
\ No newline at end of file