PEP 8

fortran/build_help/sub_sizes.py

@@ -3,12 +3,13 @@ import os
 from string import Template
 import sys
 
+
 def main():
+    print("Running sub_sizes")
 
-    print ("Running sub_sizes")
     try:
         result = subprocess.check_output(["./sizes"])
-    except:
+    except OSError:
         result = subprocess.check_output(["sizes.exe"])
 
     split_result = result.split()
@@ -17,28 +18,27 @@ def main():
         if split_result[0].strip().decode() != "SIZES":
             raise ValueError("First line is not SIZES")
 
-        subs = {"FOMP_SCHED_KIND" : split_result[1].strip().decode(),
-                "FOMP_LOCK_KIND" : split_result[2].strip().decode(),
-                "FOMP_NEST_LOCK_KIND" : split_result[3].strip().decode(),
-                "FOMP_SCHED_STATIC" : split_result[4].strip().decode(),
-                "FOMP_SCHED_DYNAMIC" : split_result[5].strip().decode(),
-                "FOMP_SCHED_GUIDED" : split_result[6].strip().decode(),
-                "FOMP_SCHED_AUTO" : split_result[7].strip().decode()
+        subs = {"FOMP_SCHED_KIND": split_result[1].strip().decode(),
+                "FOMP_LOCK_KIND": split_result[2].strip().decode(),
+                "FOMP_NEST_LOCK_KIND": split_result[3].strip().decode(),
+                "FOMP_SCHED_STATIC": split_result[4].strip().decode(),
+                "FOMP_SCHED_DYNAMIC": split_result[5].strip().decode(),
+                "FOMP_SCHED_GUIDED": split_result[6].strip().decode(),
+                "FOMP_SCHED_AUTO": split_result[7].strip().decode()
                 }
     else:
         if split_result[0].strip() != "SIZES":
             raise ValueError("First line is not SIZES")
 
-        subs = {"FOMP_SCHED_KIND" : split_result[1].strip(),
-                "FOMP_LOCK_KIND" : split_result[2].strip(),
-                "FOMP_NEST_LOCK_KIND" : split_result[3].strip(),
-                "FOMP_SCHED_STATIC" : split_result[4].strip(),
-                "FOMP_SCHED_DYNAMIC" : split_result[5].strip(),
-                "FOMP_SCHED_GUIDED" : split_result[6].strip(),
-                "FOMP_SCHED_AUTO" : split_result[7].strip()
+        subs = {"FOMP_SCHED_KIND": split_result[1].strip(),
+                "FOMP_LOCK_KIND": split_result[2].strip(),
+                "FOMP_NEST_LOCK_KIND": split_result[3].strip(),
+                "FOMP_SCHED_STATIC": split_result[4].strip(),
+                "FOMP_SCHED_DYNAMIC": split_result[5].strip(),
+                "FOMP_SCHED_GUIDED": split_result[6].strip(),
+                "FOMP_SCHED_AUTO": split_result[7].strip()
                 }
 
-    
     ompgen_temp_path = os.path.join("..", "ompgen.F90.template")
     ompgen_out_path = os.path.join("..", "ompgen.F90")
 
@@ -47,11 +47,11 @@ def main():
 
         ompgen_string = ompgen_template.substitute(subs)
 
-    
     with open(ompgen_out_path, "w") as ompgen_out:
         ompgen_out.write(ompgen_string)
 
+    print("End sub_sizes")
+
 
-    print ("End sub_sizes")
 if __name__ == "__main__":
     main()