Incorrect file not support error fixed.

0vj · May 11, 2021 · 7413662 · 7413662
1 parent 213f854
commit 7413662
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/fbl/__main__.py b/fbl/__main__.py
@@ -69,11 +69,10 @@ def main():
         f = open(file_location, 'r')
         f.close()
         #if txt family files
-        for file_format in [".txt", ".htm", ".html", ".md"]:
-            if file_location.endswith(file_format):
-                bad_urls = find_bad_urls(find_urls(txtfamily_extract_text(file_location)))
+        if file_location.endswith((".txt", ".htm", ".html", ".md")):
+            bad_urls = find_bad_urls(find_urls(txtfamily_extract_text(file_location)))
 
-        if file_location.endswith(".pdf"):
+        elif file_location.endswith(".pdf"):
             bad_urls = find_bad_urls(find_urls(pdf_extract_text(file_location)))
 
         elif file_location.endswith(".docx"):

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="fbl",
-    version="0.8.5",
+    version="0.8.6",
     author="Amirmahdi Tafreshi",
     install_requires=["certifi==2020.12.5",
                      "chardet==4.0.0",