absfuyu 4.1.0__tar.gz → 4.1.1__tar.gz

{absfuyu-4.1.0 → absfuyu-4.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: absfuyu
-Version: 4.1.0
+Version: 4.1.1
 Summary: A small collection of code
 Project-URL: Homepage, https://github.com/AbsoluteWinter/absfuyu-public
 Project-URL: Documentation, https://absolutewinter.github.io/absfuyu-docs/

{absfuyu-4.1.0 → absfuyu-4.1.1}/src/absfuyu/__init__.py

@@ -22,7 +22,7 @@ Using in cmd (`absfuyu[cli]` required):
 __title__ = "absfuyu"
 __author__ = "AbsoluteWinter"
 __license__ = "MIT License"
-__version__ = "4.1.0"
+__version__ = "4.1.1"
 __all__ = [
     "core",
     "config",

{absfuyu-4.1.0 → absfuyu-4.1.1}/src/absfuyu/cli/do_group.py

@@ -3,7 +3,7 @@ ABSFUYU CLI
 -----------
 Do
 
-Version: 1.3.0
+Version: 1.3.1
 Date updated: 01/02/2025 (dd/mm/yyyy)
 """
 
@@ -18,7 +18,7 @@ from absfuyu import __title__
 from absfuyu.cli.color import COLOR
 from absfuyu.core import __package_feature__
 from absfuyu.general.human import Human2
-from absfuyu.tools.checksum import checksum_operation
+from absfuyu.tools.checksum import Checksum
 from absfuyu.util.zipped import Zipper
 from absfuyu.version import PkgVersion
 
@@ -109,6 +109,26 @@ def unzip_files_in_dir(dir: str) -> None:
     show_default=True,
     help="Hash mode",
 )
+@click.option(
+    "--save-result",
+    "-s",
+    "save_result",
+    type=bool,
+    default=False,
+    is_flag=True,
+    show_default=True,
+    help="Save checksum result to file",
+)
+@click.option(
+    "--recursive",
+    "-r",
+    "recursive_mode",
+    type=bool,
+    default=False,
+    is_flag=True,
+    show_default=True,
+    help="Do checksum for every file in the folder (including child folder)",
+)
 @click.option(
     "--compare",
     "-c",
@@ -121,11 +141,14 @@ def unzip_files_in_dir(dir: str) -> None:
 def file_checksum(
     file_path: str,
     hash_mode: str | Literal["md5", "sha1", "sha256", "sha512"],
+    save_result: bool,
+    recursive_mode: bool,
     hash_to_compare: str,
 ) -> None:
     """Checksum for file"""
-
-    res = checksum_operation(file_path, hash_mode=hash_mode)
+    # print(hash_mode, save_result, recursive_mode)
+    instance = Checksum(file_path, hash_mode=hash_mode, save_result_to_file=save_result)
+    res = instance.checksum(recursive=recursive_mode)
     if hash_to_compare:
         print(res == hash_to_compare)
     else:

{absfuyu-4.1.0 → absfuyu-4.1.1}/src/absfuyu/core.py

@@ -3,7 +3,7 @@ Absfuyu: Core
 -------------
 Contain type hints and other stuffs
 
-Version: 2.3.0
+Version: 2.3.1
 Date updated: 01/02/2025 (dd/mm/yyyy)
 """
 
@@ -25,7 +25,6 @@ __package_feature__ = ["beautiful", "extra", "res", "full", "dev"]
 ###########################################################################
 from importlib import import_module
 from importlib.resources import files
-from typing import Iterable
 
 
 class CLITextColor:
@@ -54,5 +53,5 @@ try:
     tqdm = import_module("tqdm").tqdm
 except ModuleNotFoundError:
 
-    def tqdm(iterable: Iterable, **kwargs):
+    def tqdm(iterable, **kwargs):
         return iterable

absfuyu-4.1.1/src/absfuyu/tools/checksum.py

@@ -0,0 +1,143 @@
+"""
+Absufyu: Checksum
+-----------------
+Check MD5, SHA256, ...
+
+Version: 1.1.0
+Date updated: 01/02/2025 (dd/mm/yyyy)
+"""
+
+# Module level
+###########################################################################
+__all__ = ["Checksum", "checksum_operation"]
+
+
+# Library
+###########################################################################
+import hashlib
+from pathlib import Path
+from typing import Literal
+
+from absfuyu.core import tqdm
+
+# Function
+###########################################################################
+
+
+# Deprecated
+def checksum_operation(
+    file: Path | str,
+    hash_mode: str | Literal["md5", "sha1", "sha256", "sha512"] = "sha256",
+) -> str:
+    """This performs checksum"""
+    if hash_mode.lower() == "md5":
+        hash_engine = hashlib.md5()
+    elif hash_mode.lower() == "sha1":
+        hash_engine = hashlib.sha1()
+    elif hash_mode.lower() == "sha256":
+        hash_engine = hashlib.sha256()
+    elif hash_mode.lower() == "sha512":
+        hash_engine = hashlib.sha512()
+    else:
+        hash_engine = hashlib.md5()
+
+    with open(Path(file), "rb") as f:
+        while True:
+            data = f.read(4096)
+            if len(data) == 0:
+                break
+            else:
+                hash_engine.update(data)
+    return hash_engine.hexdigest()
+
+
+class Checksum:
+    def __init__(
+        self,
+        path: str | Path,
+        hash_mode: str | Literal["md5", "sha1", "sha256", "sha512"] = "sha256",
+        save_result_to_file: bool = False,
+    ) -> None:
+        self.path = Path(path)
+        self.hash_mode = hash_mode
+        self.save_result_to_file = save_result_to_file
+        self.checksum_result_file_name = "checksum_results.txt"
+
+    def _get_hash_engine(self):
+        hash_mode = self.hash_mode
+        if hash_mode.lower() == "md5":
+            hash_engine = hashlib.md5()
+        elif hash_mode.lower() == "sha1":
+            hash_engine = hashlib.sha1()
+        elif hash_mode.lower() == "sha256":
+            hash_engine = hashlib.sha256()
+        elif hash_mode.lower() == "sha512":
+            hash_engine = hashlib.sha512()
+        else:
+            hash_engine = hashlib.md5()
+        return hash_engine
+
+    def _checksum_operation(
+        self,
+        file: Path | str,
+    ) -> str:
+        """This performs checksum"""
+
+        hash_engine = self._get_hash_engine().copy()
+        with open(Path(file), "rb") as f:
+            # Read and hash the file in 4K chunks. Reading the whole
+            # file at once might consume a lot of memory if it is
+            # large.
+            while True:
+                data = f.read(4096)
+                if len(data) == 0:
+                    break
+                else:
+                    hash_engine.update(data)
+        return hash_engine.hexdigest()  # type: ignore
+
+    def checksum(self, recursive: bool = True) -> str:
+        """Perform checksum
+
+        Parameters
+        ----------
+        recursive : bool, optional
+            Do checksum for every file in the folder (including child folder), by default True
+
+        Returns
+        -------
+        str
+            Checksum hash
+        """
+        if self.path.absolute().is_dir():
+            new_path = self.path.joinpath(self.checksum_result_file_name)
+            # List of files
+            if recursive:
+                file_list: list[Path] = [
+                    x for x in self.path.glob("**/*") if x.is_file()
+                ]
+            else:
+                file_list = [x for x in self.path.glob("*") if x.is_file()]
+
+            # Checksum
+            res = []
+            for x in tqdm(file_list, desc="Calculating hash", unit_scale=True):
+                name = x.relative_to(self.path)
+                res.append(f"{self._checksum_operation(x)} | {name}")
+            output = "\n".join(res)
+        else:
+            new_path = self.path.with_name(self.checksum_result_file_name)
+            output = self._checksum_operation(self.path)
+
+        # Save result
+        if self.save_result_to_file:
+            with open(new_path, "w", encoding="utf-8") as f:
+                f.write(output)
+
+        return output
+
+
+# Run
+###########################################################################
+if __name__ == "__main__":
+    pass

absfuyu-4.1.0/src/absfuyu/tools/checksum.py

@@ -1,56 +0,0 @@
-"""
-Absufyu: Checksum
------------------
-Check MD5, SHA256, ...
-
-Version: 1.0.0
-Date updated: 01/02/2025 (dd/mm/yyyy)
-"""
-
-# Module level
-###########################################################################
-__all__ = ["checksum_operation"]
-
-
-# Library
-###########################################################################
-import hashlib
-from pathlib import Path
-from typing import Literal
-
-
-# Function
-###########################################################################
-def checksum_operation(
-    file: Path | str,
-    hash_mode: str | Literal["md5", "sha1", "sha256", "sha512"] = "sha256",
-) -> str:
-    """This performs checksum"""
-    if hash_mode.lower() == "md5":
-        hash_engine = hashlib.md5()
-    elif hash_mode.lower() == "sha1":
-        hash_engine = hashlib.sha1()
-    elif hash_mode.lower() == "sha256":
-        hash_engine = hashlib.sha256()
-    elif hash_mode.lower() == "sha512":
-        hash_engine = hashlib.sha512()
-    else:
-        hash_engine = hashlib.md5()
-
-    with open(Path(file), "rb") as f:
-        # Read and hash the file in 4K chunks. Reading the whole
-        # file at once might consume a lot of memory if it is
-        # large.
-        while True:
-            data = f.read(4096)
-            if len(data) == 0:
-                break
-            else:
-                hash_engine.update(data)
-    return hash_engine.hexdigest()
-
-
-# Run
-###########################################################################
-if __name__ == "__main__":
-    pass