style fix installRules.py

src/opnsense/scripts/suricata/installRules.py

@@ -34,15 +34,16 @@ from ConfigParser import ConfigParser
 import lib.rulecache
 from lib import rule_source_directory
 
-RuleCache = lib.rulecache.RuleCache()
+if __name__ == '__main__':
+    RuleCache = lib.rulecache.RuleCache()
 
-rule_config_fn = ('%s../rules.config'%rule_source_directory)
-rule_target_dir = ('%s../opnsense.rules'%rule_source_directory)
-rule_yaml_list = ('%s../installed_rules.yaml'%rule_source_directory)
+    rule_config_fn = ('%s../rules.config'%rule_source_directory)
+    rule_target_dir = ('%s../opnsense.rules'%rule_source_directory)
+    rule_yaml_list = ('%s../installed_rules.yaml'%rule_source_directory)
 
-# parse OPNsense rule config
-rule_updates = {}
-if os.path.exists(rule_config_fn):
+    # parse OPNsense rule config
+    rule_updates = {}
+    if os.path.exists(rule_config_fn):
         cnf = ConfigParser()
         cnf.read(rule_config_fn)
         for section in cnf.sections():
@@ -52,13 +53,13 @@ if os.path.exists(rule_config_fn):
                 for rule_item in cnf.items(section):
                     rule_updates[sid][rule_item[0]] = rule_item[1]
 
-# create target rule directory if not existing
-if not os.path.exists(rule_target_dir):
+    # create target rule directory if not existing
+    if not os.path.exists(rule_target_dir):
         os.mkdir(rule_target_dir, 0o755)
 
-# install ruleset
-all_installed_files = []
-for filename in RuleCache.listLocal():
+    # install ruleset
+    all_installed_files = []
+    for filename in RuleCache.listLocal():
         output_data = []
         for rule_info_record in RuleCache.listRules(filename=filename):
             # default behavior, do not touch rule, only copy to output
@@ -84,8 +85,8 @@ for filename in RuleCache.listLocal():
         all_installed_files.append(filename.split('/')[-1])
         open('%s/%s'%(rule_target_dir, filename.split('/')[-1]), 'wb').write('\n'.join(output_data))
 
-# flush all written rule filenames into yaml file
-with open(rule_yaml_list,'wb') as f_out:
+    # flush all written rule filenames into yaml file
+    with open(rule_yaml_list,'wb') as f_out:
         f_out.write('%YAML 1.1\n')
         f_out.write('---\n')
         f_out.write('rule-files:\n')