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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vim-eof-comment
-Version: 0.3.19
+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.19 → 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.19"
+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.19 → vim_eof_comment-0.3.21}/vim_eof_comment/eof.py

@@ -77,7 +77,7 @@ def eof_comment_search(
         if last_line != comment_map[ext] or (newline and not had_nwl):
             verbose_print(f"{_BRIGHT}{_RED}CHANGED", verbose=verbose)
             result[path] = EOFCommentSearch(
-                state=IOWrapperBool(file=open(path, "r", encoding="utf8"), had_nwl=had_nwl),
+                state=IOWrapperBool(file=open(path, "rb"), had_nwl=had_nwl),
                 lang=ext,
                 match=matches(last_line, verbose)
             )

{vim_eof_comment-0.3.19 → 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,8 +135,11 @@ 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, "r"), "ft_ext": ext}
+            result[fpath] = {"file": open(fpath, "rb"), "ft_ext": ext}
         except KeyboardInterrupt:
             die("\nProgram interrupted!", code=1)  # Kills the program
         except FileNotFoundError:
@@ -101,7 +173,8 @@ def modify_file(file: TextIOWrapper, comments: Dict[str, str], ext: str, **kwarg
     matching: bool = kwargs.get("matching", False)
     newline: bool = kwargs.get("newline", False)
 
-    data: List[str] = file.read().split("\n")
+    bdata: str = file.read().decode(encoding="utf8")
+    data: List[str] = bdata.split("\n")
     file.close()
 
     data_len = len(data)
@@ -143,7 +216,8 @@ def get_last_line(file: TextIOWrapper) -> LineBool:
     LineBool
         An object containing both the last line in a string and a boolean indicating a newline.
     """
-    data: List[str] = file.read().split("\n")
+    bdata: str = file.read().decode(encoding="utf8")
+    data: List[str] = bdata.split("\n")
     file.close()
 
     had_newline, line = False, ""

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

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

{vim_eof_comment-0.3.19 → 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.19
+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.19/version.txt

@@ -1 +0,0 @@
-0.3.19