MainShortcuts2 2.3.1__tar.gz → 2.3.2__tar.gz

{mainshortcuts2-2.3.1 → mainshortcuts2-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: MainShortcuts2
-Version: 2.3.1
+Version: 2.3.2
 Summary: Сокращение и улучшение функций + консольные утилиты
 Home-page: https://github.com/MainPlay-TG/MainShortcuts2.py
 Author: MainPlay TG

{mainshortcuts2-2.3.1 → mainshortcuts2-2.3.2}/pyproject.toml

@@ -1,5 +1,5 @@
 [tool.poetry]
-version = "2.3.1"
+version = "2.3.2"
 name = "MainShortcuts2"
 description = "Сокращение и улучшение функций + консольные утилиты"
 authors = [ "MainPlay TG <xbox.roman6666666666@gmail.com>",]
@@ -10,6 +10,8 @@ packages = [
 ]
 
 [tool.poetry.scripts]
+ms2-hash_check = "MainShortcuts2.__main__:hash_check"
+ms2-hash_gen = "MainShortcuts2.__main__:hash_gen"
 ms2-import_example = "MainShortcuts2.__main__:import_example"
 nano-json = "MainShortcuts2.__main__:nano_json"
 nginx-reload = "MainShortcuts2.__main__:nginx_reload"

{mainshortcuts2-2.3.1 → mainshortcuts2-2.3.2}/src/MainShortcuts2/__main__.py

@@ -1,6 +1,8 @@
 import argparse
 import sys
 from MainShortcuts2 import ms
+HASH_SUFFIX = ".MS2_hash"
+HASH_TYPES = ["sha1", "sha256", "sha512"]
 
 
 def import_example():
@@ -21,7 +23,7 @@ def nano_json():
   if args.nano_help:
     return subprocess.call(["nano", "--help"])
   nano_args = ["nano"]
-  if nano_args.rcfile:
+  if args.rcfile:
     nano_args += ["--rcfile", args.rcfile]
   nano_args += args.files
   for i in args.files:
@@ -72,3 +74,55 @@ def nginx_restart():
   if code != 0:
     sys.exit(code)
   sys.exit(subprocess.call(["systemctl", "restart", "nginx"]))
+
+
+def hash_gen():
+  import os
+  import hashlib
+  argp = argparse.ArgumentParser("ms2-hash-gen", description="создание контрольной суммы для файла")
+  argp.add_argument("files", nargs="+", help="пути к файлам")
+  argp.add_argument("-t", "--type", choices=HASH_TYPES, default="sha512", help="тип контрольной суммы")
+  args = argp.parse_args()
+  data = {}
+  data["format"] = "MainShortcuts2_hash_v1"
+  data["file"] = {}
+  data["hash"] = {}
+  data["hash"]["type"] = args.type
+  Hash = getattr(hashlib, args.type)
+  for file in args.files:
+    hash = Hash()
+    data["file"]["size"] = os.path.getsize(file)
+    with open(file, "rb") as f:
+      for chunk in f:
+        hash.update(chunk)
+    data["hash"]["hex"] = hash.hexdigest()
+    ms.json.write(file + HASH_SUFFIX, data)
+
+
+def hash_check()
+
+
+import hashlib
+import os
+import shlex
+argp = argparse.ArgumentParser("ms2-hash-check", description="проверка размера и контрольной суммы файла")
+argp.add_argument("files", nargs="+", help="пути к файлам")
+args = argp.parse_args()
+for file in args.files:
+    if file.lower().endswith(HASH_SUFFIX.lower()):
+      file = file[:0 - len(HASH_SUFFIX)]
+    if not os.path.exists(file + HASH_SUFFIX):
+      print("Ошибка: не найден файл " + shlex.quote(file + HASH_SUFFIX), file=sys.stderr)
+      continue
+    data = ms.json.read(file + HASH_SUFFIX)
+    if data["file"]["size"] != os.path.getsize(file):
+      print("Ошибка: размер файла " + shlex.quote(file) + " не совпадает")
+      continue
+    hash = getattr(hashlib, data["hash"]["type"])()
+    with open(file, "rb") as f:
+      for chunk in f:
+        hash.update(chunk)
+    if data["hash"]["hex"] == hash.hexdigest():
+      print("Успех: файл " + shlex.quote(file) + " не изменён")
+    else:
+      print("Ошибка: файл " + shlex.quote(file) + " изменён")

{mainshortcuts2-2.3.1 → mainshortcuts2-2.3.2}/src/MainShortcuts2/_module_info.py

@@ -1,2 +1,2 @@
 name = "MainShortcuts2"
-version = "2.3.1"
+version = "2.3.2"

{mainshortcuts2-2.3.1 → mainshortcuts2-2.3.2}/src/MainShortcuts2/utils.py

@@ -463,6 +463,8 @@ def check_programs(*progs: str, raise_error: bool = True) -> list[str]:
   failed.sort()
   if raise_error:
     if len(failed) > 0:
+      if len(failed) == 1:
+        raise OSError("Failed to find program " + failed[0] + " in $PATH")
       raise OSError("Failed to find programs " + (", ".join(failed)) + " in $PATH")
   return failed