Merge pull request #121 from BBerastegui/master

Added option choices for the export format
s0md3v · Mar 21, 2019 · 25abb08 · 25abb08
2 parents 4fe6c73 + 3e3c6d2
commit 25abb08
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/core/config.py b/core/config.py
@@ -9,7 +9,7 @@
     'youtube.com',
 ]
 
-BAD_TYPES = [
+BAD_TYPES = (
     'bmp',
     'css',
     'csv',
@@ -24,4 +24,4 @@
     'svg',
     'xls',
     'xml',
-]
+)
diff --git a/photon.py b/photon.py
@@ -55,7 +55,7 @@
 parser.add_argument('-u', '--url', help='root url', dest='root')
 parser.add_argument('-c', '--cookie', help='cookie', dest='cook')
 parser.add_argument('-r', '--regex', help='regex pattern', dest='regex')
-parser.add_argument('-e', '--export', help='export format', dest='export')
+parser.add_argument('-e', '--export', help='export format', dest='export', choices=['csv', 'json'])
 parser.add_argument('-o', '--output', help='output directory', dest='output')
 parser.add_argument('-l', '--level', help='levels to crawl', dest='level',
                     type=int)