Move script part into if __name == "__main__"

roles/slurm/files/update_config.py

@@ -73,18 +73,19 @@ def get_node_configs(limits, shapes, mgmt_info):
                     yield create_slurmconf_line(i, shape_info, shape, ad)
 
 
-# TODO Make sure that any nodes which are no longer managed due to service limit reductions are terminated.
+if __name__ == "__main__":
+    # TODO Make sure that any nodes which are no longer managed due to service limit reductions are terminated.
 
-slurm_conf_filename = "/mnt/shared/etc/slurm/slurm.conf"
+    slurm_conf_filename = "/mnt/shared/etc/slurm/slurm.conf"
 
-node_config = "\n".join(get_node_configs(get_limits(), get_shapes(), get_mgmt_info()))
+    node_config = "\n".join(get_node_configs(get_limits(), get_shapes(), get_mgmt_info()))
 
-chop = re.compile('(?<=# STARTNODES\n)(.*?)(?=\n?# ENDNODES)', re.DOTALL)
+    chop = re.compile('(?<=# STARTNODES\n)(.*?)(?=\n?# ENDNODES)', re.DOTALL)
 
-with open(slurm_conf_filename) as f:
-    all_config = f.read()
+    with open(slurm_conf_filename) as f:
+        all_config = f.read()
 
-new_config = chop.sub('{}'.format(node_config), all_config)
+    new_config = chop.sub('{}'.format(node_config), all_config)
 
-with open(slurm_conf_filename, "w") as f:
-    f.write(new_config)
+    with open(slurm_conf_filename, "w") as f:
+        f.write(new_config)