read from blacklist.txt only if it exists

utiso · Aug 31, 2018 · 87b4a9a · 87b4a9a
1 parent 6efaaed
commit 87b4a9a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dorkbot.py b/dorkbot.py
@@ -145,8 +145,9 @@ def scan(db, scanner, args):
 def get_blacklist(blacklist_file):
     pattern = "$^"
     try:
-        with open(blacklist_file, "r") as f:
-            pattern = "|".join(f.read().splitlines())
+        if os.path.isfile(blacklist_file):
+            with open(blacklist_file, "r") as f:
+                pattern = "|".join(f.read().splitlines())
         blacklist = re.compile(pattern)
     except Exception as e:
         print("ERROR reading blacklist - %s" % e, file=sys.stderr)