vim-eof-comment 0.3.20__tar.gz → 0.3.21__tar.gz

{vim_eof_comment-0.3.20 → vim_eof_comment-0.3.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vim-eof-comment
-Version: 0.3.20
+Version: 0.3.21
 Summary: Adds Vim EOF modeline comments for given filetypes in given directories
 Author-email: Guennadi Maximov C <g.maxc.fox@protonmail.com>
 Maintainer-email: Guennadi Maximov C <g.maxc.fox@protonmail.com>

{vim_eof_comment-0.3.20 → vim_eof_comment-0.3.21}/pyproject.toml

@@ -39,7 +39,7 @@ maintainers = [{ name = "Guennadi Maximov C", email = "g.maxc.fox@protonmail.com
 name = "vim-eof-comment"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.3.20"
+version = "0.3.21"
 
     [project.scripts]
     vim-eof-comment = "vim_eof_comment.eof:main"

vim_eof_comment-0.3.21/version.txt

@@ -0,0 +1 @@
+0.3.21

{vim_eof_comment-0.3.20 → vim_eof_comment-0.3.21}/vim_eof_comment/file.py

@@ -15,6 +15,64 @@ from typing import Dict, List
 from .types import BatchPairDict, BatchPathDict, LineBool
 from .util import die, error
 
+EXCLUDED_DIRS: List[str] = [
+    ".git",
+    ".mypy_cache",
+    ".ropeproject",
+    ".venv",
+    "build",
+    "venv",
+]
+
+
+def try_open(fpath: str) -> bool:
+    """
+    Try to open a file, unless a ``UnicodeDecodeError`` triggers.
+
+    Parameters
+    ----------
+    fpath : str
+        The file path to try and open.
+
+    Returns
+    -------
+    bool
+        Whether the file triggers a ``UnicodeDecodeError`` or not.
+    """
+    file = open(fpath, "rb")
+    success: bool = True
+    try:
+        file.read().decode(encoding="utf8")
+    except UnicodeDecodeError:
+        success = False
+    except Exception:
+        die("Something went wrong in `try_open()`!", code=2)
+
+    file.close()
+    return success
+
+
+def has_excluded(dir: str) -> bool:
+    """
+    Check whether a directory list contains any excluded directories.
+
+    Parameters
+    ----------
+    dir : str
+        The directory to check.
+
+    Returns
+    -------
+    bool
+        Whether an excluded directory was found.
+    """
+    dir = dir.rstrip("/")
+    for excluded in EXCLUDED_DIRS:
+        if dir.endswith(excluded):
+            return True
+
+    return False
+
 
 def bootstrap_paths(paths: List[str], exts: List[str]) -> List[BatchPairDict]:
     """
@@ -34,11 +92,22 @@ def bootstrap_paths(paths: List[str], exts: List[str]) -> List[BatchPairDict]:
     """
     result = list()
     for path in paths:
-        if not isdir(path):
+        if not isdir(path) or has_excluded(path):
             continue
 
-        file: str
+        root: str
+        dirs: List[str]
+        files: List[str]
         for root, dirs, files in walk(path):
+            excluded = False
+            for dir in dirs:
+                if has_excluded(dir):
+                    excluded = True
+                    break
+
+            if excluded:
+                continue
+
             for file in files:
                 for ext in exts:
                     if not file.endswith(ext):
@@ -66,6 +135,9 @@ def open_batch_paths(paths: List[BatchPairDict]) -> Dict[str, BatchPathDict]:
     result = dict()
     for path in paths:
         fpath, ext = path["fpath"], path["ft_ext"]
+        if not try_open(fpath):
+            continue
+
         try:
             result[fpath] = {"file": open(fpath, "rb"), "ft_ext": ext}
         except KeyboardInterrupt:

{vim_eof_comment-0.3.20 → vim_eof_comment-0.3.21}/vim_eof_comment/version.py

@@ -227,6 +227,7 @@ version_info = VersionInfo([
     (0, 3, 18),
     (0, 3, 19),
     (0, 3, 20),
+    (0, 3, 21),
 ])
 
 __version__: str = str(version_info)

{vim_eof_comment-0.3.20 → vim_eof_comment-0.3.21}/vim_eof_comment.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vim-eof-comment
-Version: 0.3.20
+Version: 0.3.21
 Summary: Adds Vim EOF modeline comments for given filetypes in given directories
 Author-email: Guennadi Maximov C <g.maxc.fox@protonmail.com>
 Maintainer-email: Guennadi Maximov C <g.maxc.fox@protonmail.com>

vim_eof_comment-0.3.20/version.txt

@@ -1 +0,0 @@
-0.3.20