Update add_new_plugin script with support for xpath

alertot · Oct 11, 2017 · 862d44d · 862d44d
1 parent 59ac97a
commit 862d44d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scripts/add_new_plugin.py b/scripts/add_new_plugin.py
@@ -7,14 +7,15 @@
 )
 PLUGIN_DIRECTORY = os.path.join(ROOT_DIRECTORY, 'detectem/plugins')
 PLUGIN_DIRECTORIES = [
-    d for d in os.listdir(PLUGIN_DIRECTORY) if not d.startswith('_')
+    d for d in os.listdir(PLUGIN_DIRECTORY)
+    if os.path.isdir(os.path.join(PLUGIN_DIRECTORY, d)) and d != '__pycache__'
 ]
 
 
 @click.command()
 @click.option(
     '--matcher',
-    type=click.Choice(['url', 'body', 'header']),
+    type=click.Choice(['url', 'body', 'header', 'xpath']),
     required=True,
     help='Set the matcher type.',
 )