absfuyu 4.1.0__py3-none-any.whl → 4.1.1__py3-none-any.whl

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/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/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/tools/checksum.py

@@ -3,13 +3,13 @@ Absufyu: Checksum
 -----------------
 Check MD5, SHA256, ...
 
-Version: 1.0.0
+Version: 1.1.0
 Date updated: 01/02/2025 (dd/mm/yyyy)
 """
 
 # Module level
 ###########################################################################
-__all__ = ["checksum_operation"]
+__all__ = ["Checksum", "checksum_operation"]
 
 
 # Library
@@ -18,9 +18,13 @@ 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",
@@ -38,9 +42,6 @@ def checksum_operation(
         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:
@@ -50,6 +51,92 @@ def checksum_operation(
     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__":

{absfuyu-4.1.0.dist-info → absfuyu-4.1.1.dist-info}/METADATA

@@ -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.dist-info → absfuyu-4.1.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
-absfuyu/__init__.py,sha256=H-4MDDeiuQVuYt4Qrv8ae8286dc3Vv2JHgOkU6VxCKw,638
+absfuyu/__init__.py,sha256=EGTitZIqMbg-jQFJD5nJi4papdMQsCoYnr-9EnvlUmI,638
 absfuyu/__main__.py,sha256=OpMwc35W5VANzw6gvlqJaJOl2H89i_frFZbleUQwDss,163
-absfuyu/core.py,sha256=pGN-tBD5S_p0lm3f0clPujnH9n6JZSD1cqNilBM-ZK0,1279
+absfuyu/core.py,sha256=MZQH3hpUAamGx2NM-TlIWvAnRBBwa9R9fm7F2WFe60c,1240
 absfuyu/everything.py,sha256=PGIXlqgxyADFPygohVYVIb7fZz72L_xXrlLX0WImuYg,788
 absfuyu/logger.py,sha256=Pue3oWYYSF03A-fVp3E137jzlZVI71mYqddic8myauQ,13141
 absfuyu/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -9,7 +9,7 @@ absfuyu/version.py,sha256=KMaeXNl93L4VU2RnTySoFv23IDyTvHfy84nyatFcKaE,14128
 absfuyu/cli/__init__.py,sha256=nKTbe4iPpkjtDDMJwVY36heInKDm1zyek9ags5an7s4,1105
 absfuyu/cli/color.py,sha256=7M3XqFllt26tv_NR5qKgyTId6wnVADtUB74cDYy8pOQ,497
 absfuyu/cli/config_group.py,sha256=FsyYm2apSyAA2PAD12CpHAizenRds_QlLf8j0AlLuVo,1230
-absfuyu/cli/do_group.py,sha256=_GzTrxZwVtrTl3kTd_MQTstX5DKQTjDbFgHmlghanP0,3533
+absfuyu/cli/do_group.py,sha256=HS1G9nFnqtmNVwWZiOGvu4RB6URT2VnCpEtdY9e7n2Q,4132
 absfuyu/cli/game_group.py,sha256=ySpL2hm4VCplhNY0s22kBGI5eFCdJj9fm1T58yftU74,2348
 absfuyu/config/__init__.py,sha256=4r77tFm3mgHX8H1uWWNXC_FZ8hiSJzsYAyiStZiVj5w,8641
 absfuyu/config/config.json,sha256=-ZQnmDuLq0aAFfsrQbSNR3tq5k9Eu9IVUQgYD9htIQM,646
@@ -39,7 +39,7 @@ absfuyu/pkg_data/__init__.py,sha256=VbUoFnUDiw_3bKZVvIzwf_ve0y97rmhfpkbbrP_ob_w,
 absfuyu/pkg_data/chemistry.pkl,sha256=kYWNa_PVffoDnzT8b9Jvimmf_GZshPe1D-SnEKERsLo,4655
 absfuyu/pkg_data/tarot.pkl,sha256=ssXTCC_BQgslO5F-3a9HivbxFQ6BioIe2E1frPVi2m0,56195
 absfuyu/tools/__init__.py,sha256=VDmmMLEfiRyUWPVrPYc8JUEa5TXjLguKVnyI3YavFv0,118
-absfuyu/tools/checksum.py,sha256=33DlwXRqPcatuNJ14NG9gJLe3y5RgnJm0w0dkSFurhU,1563
+absfuyu/tools/checksum.py,sha256=cLSuL_ZsoroFYYVR1GhOfBbon_iXEm1OazbjYFPJabM,4393
 absfuyu/tools/converter.py,sha256=8jqOSdkbzzupKJcBndV5q4OBu2kWFWpV3j2IswS4Vqc,10609
 absfuyu/tools/keygen.py,sha256=drj8OUDCXrLvcVf9_Shd5UMShm_mBTxa9Sg_BrAT5yM,7356
 absfuyu/tools/obfuscator.py,sha256=kiaFvMw2RwEfZvXKdbUrEYA_AuVETivoL2eVmtUWU4w,8669
@@ -54,8 +54,8 @@ absfuyu/util/performance.py,sha256=F-aLJ-5wmwkiOdl9_wZcFRKifL9zOahw52JeL4TbCuQ,9
 absfuyu/util/pkl.py,sha256=ZZf6-PFC2uRGXqARNi0PGH3A0IXwMP0sYPWZJXENvak,1559
 absfuyu/util/shorten_number.py,sha256=_R_FhzBYDvtX_B8jA0WhVyG2G_s-kyolra9Y2gH5e4w,6603
 absfuyu/util/zipped.py,sha256=CU_le1MynFwHfpajCRRJ7_A-r1jfMjt9uu72jLooW6w,3111
-absfuyu-4.1.0.dist-info/METADATA,sha256=xcjNv6lMttGHr7okEXP4HHZ2V2hGAhFV74q_zpBDBMw,3794
-absfuyu-4.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-absfuyu-4.1.0.dist-info/entry_points.txt,sha256=bW5CgJRTTWJ2Pywojo07sf-YucRPcnHzMmETh5avbX0,79
-absfuyu-4.1.0.dist-info/licenses/LICENSE,sha256=pFCHBSNSzdXwYG1AHpq7VcofI1NMQ1Fc77RzZ4Ln2O4,1097
-absfuyu-4.1.0.dist-info/RECORD,,
+absfuyu-4.1.1.dist-info/METADATA,sha256=dJJP-975rfpiV712uDMRZXbnOhDtmLGkfAWMfriu9L0,3794
+absfuyu-4.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+absfuyu-4.1.1.dist-info/entry_points.txt,sha256=bW5CgJRTTWJ2Pywojo07sf-YucRPcnHzMmETh5avbX0,79
+absfuyu-4.1.1.dist-info/licenses/LICENSE,sha256=pFCHBSNSzdXwYG1AHpq7VcofI1NMQ1Fc77RzZ4Ln2O4,1097
+absfuyu-4.1.1.dist-info/RECORD,,