MainShortcuts2 2.3.2__tar.gz → 2.4.0__tar.gz

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

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

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

@@ -1,12 +1,12 @@
 [tool.poetry]
-version = "2.3.2"
+version = "2.4.0"
 name = "MainShortcuts2"
 description = "Сокращение и улучшение функций + консольные утилиты"
 authors = [ "MainPlay TG <xbox.roman6666666666@gmail.com>",]
 readme = "README.md"
 repository = "https://github.com/MainPlay-TG/MainShortcuts2.py"
 packages = [
-    { include = "MainShortcuts2", from = "src" },
+  { include = "MainShortcuts2", from = "src" },
 ]
 
 [tool.poetry.scripts]

mainshortcuts2-2.4.0/src/MainShortcuts2/__main__.py

@@ -0,0 +1,172 @@
+import argparse
+import sys
+from MainShortcuts2 import ms
+HASH_SUFFIX = ".MS2_hash"
+HASH_TYPES = ["blake2b", "blake2s", "md5", "sha1", "sha224", "sha256", "sha384", "sha3_224", "sha3_256", "sha3_384", "sha3_512", "sha512"]
+
+
+def import_example():
+  argp = argparse.ArgumentParser("ms2-import_example", description="код для импорта MainShortcuts2")
+  argp.parse_args()
+  print("from MainShortcuts2 import ms")
+
+
+def nano_json():
+  ms.utils.check_programs("nano")
+  import subprocess
+  argp = argparse.ArgumentParser("nano-json", description="форматирование JSON файлов и редактирование в GNU NANO")
+  argp.add_argument("files", nargs="+", help="пути к файлам JSON")
+  argp.add_argument("--encoding", default="utf-8", help="кодировка файлов"
+  argp.add_argument("--nano-help", action="store_true", help="показать помощь nano")
+  argp.add_argument("--no-escape", action="store_false", help="не использовать Unicode Escape")
+  argp.add_argument("-f", "--rcfile", help="использовать только этот файл для настройки nano")
+  argp.add_argument("-m", "--mode", choices=ms.json.MODES, default="p", help="режим сохранения редактирования")
+  args=argp.parse_args()
+  if args.nano_help:
+    return subprocess.call(["nano", "--help"])
+  nano_args=["nano"]
+  if args.rcfile:
+    nano_args += ["--rcfile", args.rcfile]
+  nano_args += args.files
+  for i in args.files:
+    try:
+      data=ms.json.read(i, encoding=args.encoding)
+      ms.json.write(i + ms.file.TMP_SUFFIX, data, encoding=args.encoding, ensure_ascii=False, mode="p")
+      ms.file.delete(i)
+      ms.file.move(i + ms.file.TMP_SUFFIX, i)
+    except Exception as err:
+      print(err, file=sys.stderr)
+  subprocess.call(nano_args)
+  for i in args.files:
+    try:
+      data=ms.json.read(i, encoding=args.encoding)
+      ms.json.write(i + ms.file.TMP_SUFFIX, data, encoding=args.encoding, ensure_ascii=args.no_escape, mode=args.mode)
+      ms.file.delete(i)
+      ms.file.move(i + ms.file.TMP_SUFFIX, i)
+    except Exception as err:
+      print(err, file=sys.stderr)
+
+
+def _check_nginx() -> int:
+  import subprocess
+  with subprocess.Popen(["nginx", "-t"], stderr=subprocess.PIPE) as p:
+    code=p.wait()
+    if code != 0:
+      sys.stderr.buffer.write(p.stderr.read())
+  return code
+
+
+def nginx_reload():
+  ms.utils.check_programs("nginx")
+  import subprocess
+  argp=argparse.ArgumentParser("nginx-reload", description="проверка конфига и перезагрузка Nginx")
+  argp.parse_args()
+  code=_check_nginx()
+  if code != 0:
+    sys.exit(code)
+  sys.exit(subprocess.call(["nginx", "-s", "reload"]))
+
+
+def nginx_restart():
+  ms.utils.check_programs("nginx", "systemctl")
+  import subprocess
+  argp=argparse.ArgumentParser("nginx-reload", description="проверка конфига и перезапуск Nginx через systemctl")
+  argp.parse_args()
+  code=_check_nginx()
+  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("-b", "--bar", action="store_true", help="показывать прогрессбар (нужен модуль progressbar2)")
+  argp.add_argument("-t", "--type", choices=HASH_TYPES, default="sha512", help="тип контрольной суммы")
+  args=argp.parse_args()
+  if args.bar:
+    import progressbar
+    pbar_w=[
+      progressbar.Percentage(),
+      progressbar.GranularBar(left="(", right=")"),
+      progressbar.FileTransferSpeed(),
+      progressbar.ETA(format="%(eta)8s", format_finished="%(elapsed)8s", format_na="     N/A", format_not_started="--:--:--", format_zero="00:00:00"),
+    ]
+  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)
+    if args.bar:
+      pbar=progressbar.ProgressBar(
+        max_error=False,
+        max_value=data["file"]["size"],
+        min_value=0,
+        widgets=[ms.path.Path(file).full_name] + pbar_w,
+      )
+      c=0
+      pbar.start()
+    with open(file, "rb") as f:
+      for chunk in f:
+        hash.update(chunk)
+        if args.bar:
+          c += len(chunk)
+          pbar.update(c)
+    if args.bar:
+      pbar.finish(dirty=True)
+    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("-b", "--bar", action="store_true", help="показывать прогрессбар (нужен модуль progressbar2)")
+  argp.add_argument("files", nargs="+", help="пути к файлам")
+  args=argp.parse_args()
+  if args.bar:
+    import progressbar
+    pbar_w=[
+      progressbar.Percentage(),
+      progressbar.GranularBar(left="(", right=")"),
+      progressbar.FileTransferSpeed(),
+      progressbar.ETA(format="%(eta)8s", format_finished="%(elapsed)8s", format_na="     N/A", format_not_started="--:--:--", format_zero="00:00:00"),
+    ]
+  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"])()
+    if args.bar:
+      pbar=progressbar.ProgressBar(
+        max_error=False,
+        max_value=data["file"]["size"],
+        min_value=0,
+        widgets=[ms.path.Path(file).full_name] + pbar_w,
+      )
+      c=0
+      pbar.start()
+    with open(file, "rb") as f:
+      for chunk in f:
+        hash.update(chunk)
+        if args.bar:
+          c += len(chunk)
+          pbar.update()
+    if args.bar:
+      pbar.finish(dirty=True)
+    if data["hash"]["hex"] == hash.hexdigest():
+      print("Успех: файл " + shlex.quote(file) + " не изменён")
+    else:
+      print("Ошибка: файл " + shlex.quote(file) + " изменён")

mainshortcuts2-2.4.0/src/MainShortcuts2/_any2json_regs.py

@@ -0,0 +1,136 @@
+import builtins
+from .core import ms
+decoders = {}
+encoders = {}
+
+
+def reg_decoder(type: str):
+  if builtins.type(type) != str:
+    type = type.__module__ + "." + type.__name__
+
+  def deco(func):
+    decoders[type] = func
+    return func
+  return deco
+
+
+def reg_encoder(type: str):
+  if builtins.type(type) != str:
+    type = type.__module__ + "." + type.__name__
+
+  def deco(func):
+    encoders[type] = func
+    return func
+  return deco
+
+
+@reg_decoder(type(None))
+def _(obj):
+  return None
+
+
+@reg_decoder(bool)
+def _(obj):
+  return obj
+
+
+@reg_decoder(bytes)
+def _(obj):
+  import base64
+  if obj[0] == "base64":
+    return base64.b64decode(obj[1].encode("utf-8"))
+
+
+@reg_decoder(dict)
+def _(obj):
+  result = {}
+  for k, v in obj:
+    result[ms.any2json._decode_obj(k)] = ms.any2json._decode_obj(v)
+  return result
+
+
+@reg_decoder(float)
+def _(obj):
+  return obj
+
+
+@reg_decoder(int)
+def _(obj):
+  return obj
+
+
+@reg_decoder(list)
+def _(obj):
+  return [ms.any2json._decode_obj(i) for i in obj]
+
+
+@reg_decoder(str)
+def _(obj):
+  return obj
+
+
+@reg_decoder(tuple)
+def _(obj):
+  return tuple([ms.any2json._decode_obj(i) for i in obj])
+
+
+@reg_encoder(type(None))
+def _(obj):
+  if obj is None:
+    return 0
+
+
+@reg_encoder(bool)
+def _(obj):
+  if isinstance(obj, bool):
+    return bool(obj)
+
+
+@reg_encoder(bytes)
+def _(obj):
+  if isinstance(obj, bytes):
+    from base64 import b64encode
+    result = [
+        "base64",
+        b64encode(obj).decode("utf-8"),
+    ]
+    return result
+
+
+@reg_encoder(dict)
+def _(obj):
+  if isinstance(obj, dict):
+    result = []
+    for k, v in obj.items():
+      result.append([ms.any2json._encode_obj(k), ms.any2json._encode_obj(v)])
+    return result
+
+
+@reg_encoder(float)
+def _(obj):
+  if isinstance(obj, float):
+    return float(obj)
+
+
+@reg_encoder(int)
+def _(obj):
+  if isinstance(obj, int):
+    return int(obj)
+
+
+@reg_encoder(list)
+def _(obj):
+  if isinstance(obj, list):
+    return [ms.any2json._encode_obj(i) for i in obj]
+
+
+@reg_encoder(str)
+def _(obj):
+  if isinstance(obj, str):
+    return str(obj)
+
+
+@reg_encoder(tuple)
+def _(obj):
+  if isinstance(obj, tuple):
+    return [ms.any2json._encode_obj(i) for i in obj]

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

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

mainshortcuts2-2.4.0/src/MainShortcuts2/any2json.py

@@ -0,0 +1,50 @@
+import builtins
+from .core import ms
+from ._any2json_regs import decoders, encoders, reg_decoder, reg_encoder
+
+
+def _decode_obj(obj: dict):
+  return decoders[obj["type"]](obj["data"])
+
+
+def _encode_obj(obj) -> dict:
+  result = {}
+  obj_type = builtins.type(obj)
+  if hasattr(obj_type, "__module__"):
+    if hasattr(obj_type, "__name__"):
+      type = obj_type.__module__ + "." + obj_type.__name__
+      if type in encoders:
+        result["data"] = encoders[type](obj)
+        result["type"] = type
+        return result
+  for type, func in encoders.items():
+    try:
+      encoded = func(obj)
+      if not encoded is None:
+        result["data"] = encoded
+        result["type"] = type
+        return result
+    except Exception:
+      pass
+  raise ValueError("failed to encode object " + repr(obj))
+
+
+def decode(text, **kw):
+  kw["text"] = text
+  return _decode_obj(ms.json.decode(**kw))
+
+
+def encode(data, **kw) -> str:
+  kw["data"] = _encode_obj(data)
+  return ms.json.encode(**kw)
+
+
+def read(path: str, **kw):
+  kw["path"] = path
+  return _decode_obj(ms.json.read(**kw))
+
+
+def write(path: str, data, **kw):
+  kw["data"] = _encode_obj(data)
+  kw["path"] = path
+  return ms.json.write(**kw)

{mainshortcuts2-2.3.2 → mainshortcuts2-2.4.0}/src/MainShortcuts2/core.py

@@ -33,6 +33,7 @@ class MS2:
                logger: Logger = None,
                **kw):
     self._advanced = None
+    self._any2json = None
     self._cfg = None
     self._dict = None
     self._dir = None
@@ -98,6 +99,13 @@ class MS2:
       self._advanced = advanced
     return self._advanced
 
+  @property
+  def any2json(self):
+    if self._any2json is None:
+      from . import any2json
+      self._any2json = any2json
+    return self._any2json
+
   @property
   def cfg(self):
     if self._cfg is None:

mainshortcuts2-2.3.2/src/MainShortcuts2/__main__.py

@@ -1,128 +0,0 @@
-import argparse
-import sys
-from MainShortcuts2 import ms
-HASH_SUFFIX = ".MS2_hash"
-HASH_TYPES = ["sha1", "sha256", "sha512"]
-
-
-def import_example():
-  argp = argparse.ArgumentParser("ms2-import_example", description="код для импорта MainShortcuts2")
-  argp.parse_args()
-  print("from MainShortcuts2 import ms")
-
-
-def nano_json():
-  ms.utils.check_programs("nano")
-  import subprocess
-  argp = argparse.ArgumentParser("nano-json", description="форматирование JSON файлов и редактирование в GNU NANO")
-  argp.add_argument("files", nargs="+", help="пути к файлам JSON")
-  argp.add_argument("--nano-help", action="store_true", help="показать помощь nano")
-  argp.add_argument("-f", "--rcfile", help="использовать только этот файл для настройки nano")
-  argp.add_argument("-m", "--mode", choices=ms.json.MODES, default="p", help="режим сохранения редактирования")
-  args = argp.parse_args()
-  if args.nano_help:
-    return subprocess.call(["nano", "--help"])
-  nano_args = ["nano"]
-  if args.rcfile:
-    nano_args += ["--rcfile", args.rcfile]
-  nano_args += args.files
-  for i in args.files:
-    try:
-      data = ms.json.read(i)
-      ms.json.write(i + ms.file.TMP_SUFFIX, data, ensure_ascii=False, mode="p")
-      ms.file.delete(i)
-      ms.file.move(i + ms.file.TMP_SUFFIX, i)
-    except Exception as err:
-      print(err, file=sys.stderr)
-  subprocess.call(nano_args)
-  for i in args.files:
-    try:
-      data = ms.json.read(i)
-      ms.json.write(i + ms.file.TMP_SUFFIX, data, mode=args.mode)
-      ms.file.delete(i)
-      ms.file.move(i + ms.file.TMP_SUFFIX, i)
-    except Exception as err:
-      print(err, file=sys.stderr)
-
-
-def _check_nginx() -> int:
-  import subprocess
-  with subprocess.Popen(["nginx", "-t"], stderr=subprocess.PIPE) as p:
-    code = p.wait()
-    if code != 0:
-      sys.stderr.buffer.write(p.stderr.read())
-  return code
-
-
-def nginx_reload():
-  ms.utils.check_programs("nginx")
-  import subprocess
-  argp = argparse.ArgumentParser("nginx-reload", description="проверка конфига и перезагрузка Nginx")
-  argp.parse_args()
-  code = _check_nginx()
-  if code != 0:
-    sys.exit(code)
-  sys.exit(subprocess.call(["nginx", "-s", "reload"]))
-
-
-def nginx_restart():
-  ms.utils.check_programs("nginx", "systemctl")
-  import subprocess
-  argp = argparse.ArgumentParser("nginx-reload", description="проверка конфига и перезапуск Nginx через systemctl")
-  argp.parse_args()
-  code = _check_nginx()
-  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) + " изменён")