Replaced edit distance with NLTK package

Author: jakub-safetycli

pyproject.toml

@@ -49,7 +49,7 @@ dependencies = [
   "setuptools>=65.5.1",
   "typer>=0.12.1",
   "typing-extensions>=4.7.1",
-  "python-levenshtein>=0.25.1",
+  "nltk>=3.9",
 ]
 license = "MIT"
 license-files = ["LICENSES/*"]

safety/tool/utils.py

@@ -8,7 +8,7 @@
 from tempfile import mkstemp
 
 import typer
-from Levenshtein import distance
+import nltk
 from filelock import FileLock
 from rich.padding import Padding
 from rich.prompt import Prompt
@@ -227,7 +227,7 @@ def __check_typosquatting(self, package_name):
 
         for pkg in MOST_FREQUENTLY_DOWNLOADED_PYPI_PACKAGES:
             if (abs(len(pkg) - len(package_name)) <= max_edit_distance
-                and distance(pkg, package_name) <= max_edit_distance):
+                and nltk.edit_distance(pkg, package_name) <= max_edit_distance):
                 return (False, pkg)
 
         return (True, package_name)