python-plugins 0.2.2__py3-none-any.whl → 0.2.5__py3-none-any.whl

python_plugins/__about__.py

@@ -1 +1 @@
-__version__ = "0.2.2"
+__version__ = "0.2.5"

python_plugins/crypto/zipmix.py

@@ -0,0 +1,264 @@
+import os
+import random
+import hashlib
+import tempfile
+import filecmp
+from pathlib import Path
+import subprocess
+
+
+class ZipMix:
+    def __init__(self, zipcmd="7z"):
+        self.zipcmd = zipcmd
+        self.z7_available = self.check_7z_available()
+
+    def check_7z_available(self) -> bool:
+        try:
+            subprocess.run([self.zipcmd], capture_output=True, check=True)
+            return True
+        except (subprocess.CalledProcessError, FileNotFoundError):
+            return False
+
+    def create_mix_flag(self, pos: int, nlen: int) -> bytes:
+        """m{}i{}x{}{nlen}{}"""
+        start_bytes = pos.to_bytes(4, byteorder="big")
+        return b"m%si%sx%s%s%s" % (
+            start_bytes[0:1],
+            start_bytes[1:2],
+            start_bytes[2:3],
+            nlen.to_bytes(1, byteorder="big"),
+            start_bytes[3:4],
+        )
+
+    def int_to_bytes(self, pos: int):
+        return pos.to_bytes(4, byteorder="big")
+
+    def int_from_bytes(self, intbs):
+        return int.from_bytes(intbs, byteorder="big")
+
+    def extract_from_mix_flag(self, mix_flag):
+        fix_flag = mix_flag[0:1] + mix_flag[2:3] + mix_flag[4:5]
+        if fix_flag != b"mix":
+            raise ValueError("not match mix.")
+        nlen = self.int_from_bytes(mix_flag[6:7])
+        pos = self.int_from_bytes(
+            mix_flag[1:2] + mix_flag[3:4] + mix_flag[5:6] + mix_flag[7:8]
+        )
+        return (pos, nlen)
+
+    def insert_string_to_binary(self, data, insert_bytes):
+        n_data = len(data)
+        n_ibytes = len(insert_bytes)
+        pos_list = sorted(random.sample(range(min(n_data, 0xFFFFFFFF)), n_ibytes))
+
+        parts = []
+        pre_pos = 0
+
+        for i, curr_pos in enumerate(pos_list):
+            if i == 0:
+                end_flag = self.create_mix_flag(curr_pos, n_ibytes)
+            else:
+                parts.append(self.int_to_bytes(curr_pos))
+
+            parts.append(data[pre_pos:curr_pos])
+            parts.append(insert_bytes[i : i + 1])
+            pre_pos = curr_pos
+
+        parts.append(data[pre_pos:])
+        parts.append(end_flag)
+        return b"".join(parts)
+
+    def extract_string_from_binary(self, data):
+        start_pos, n_insert_bytes = self.extract_from_mix_flag(data[-8:])
+        parts = []
+        insert_parts = []
+        pos_list = [start_pos]
+        data_pre_pos = 0
+        curr_pos = start_pos
+
+        for k in range(n_insert_bytes):
+            data_curr_pos = curr_pos + 5 * k
+            parts.append(data[data_pre_pos:data_curr_pos])
+            insert_parts.append(data[data_curr_pos : data_curr_pos + 1])
+            if k < n_insert_bytes - 1:
+                curr_pos = self.int_from_bytes(
+                    data[data_curr_pos + 1 : data_curr_pos + 5]
+                )
+                pos_list.append(curr_pos)
+                data_pre_pos = data_curr_pos + 5
+            else:
+                curr_pos = None
+                data_pre_pos = data_curr_pos + 1
+
+        parts.append(data[data_pre_pos:-8])
+        return b"".join(insert_parts), b"".join(parts)
+
+    def _generate_password(self) -> str:
+        return hashlib.sha256(os.urandom(1000)).hexdigest()[:32]
+
+    def zip7(self, file_or_dir, archive_path=None, pwd=None, force_pwd=False,silent = True):
+        path_obj = Path(file_or_dir)
+        if not path_obj.exists():
+            return None
+
+        if archive_path is None:
+            archive_path = file_or_dir + ".7z"
+
+        if pwd is None:
+            if force_pwd:
+                pwd = self._generate_password()
+        else:
+            pwd = pwd[:32]
+
+        cmd = [self.zipcmd, "a"]
+        if silent:
+            cmd.append("-bso0")
+        if pwd:
+            cmd.extend([f"-p{pwd}", "-mhe=on"])
+        cmd.extend([archive_path, file_or_dir])
+
+        try:
+            subprocess.run(cmd, check=True)
+            if pwd is not None:
+                return ("ok", archive_path, pwd)
+            else:
+                return ("ok", archive_path)
+        except subprocess.CalledProcessError as e:
+            return ("fail", f"{e}")
+
+    def unzip7_list_archive(self, archive_path, pwd=None,silent = True):
+        cmd = [self.zipcmd, "l"]
+        if silent:
+            cmd.append("-bso0")
+        if pwd:
+            cmd.append(f"-p{pwd}")
+        cmd.append(archive_path)
+
+        try:
+            subprocess.run(cmd, check=True)
+            return ("ok", archive_path)
+        except subprocess.CalledProcessError as e:
+            return ("fail", f"{e}")
+
+    def unzip7(self, archive_path, output_path=None, pwd=None, overwrite: bool = True,silent = True):
+        archive_obj = Path(archive_path)
+
+        if output_path is None:
+            output_dir = archive_obj.parent
+        else:
+            output_dir = Path(output_path)
+
+        if not output_dir.exists():
+            output_dir.mkdir(parents=True, exist_ok=True)
+
+        if not output_dir.is_dir():
+            return ("fail", f"{output_dir} is not dir")
+
+        cmd = [self.zipcmd, "x", "-y"]
+        if silent:
+            cmd.append("-bso0")
+        if not overwrite:
+            cmd.append("-aos")
+        if pwd:
+            cmd.append(f"-p{pwd}")
+        cmd.extend([archive_path, f"-o{output_dir}"])
+
+        try:
+            subprocess.run(cmd, check=True)
+            return ("ok", f"{output_dir}")
+        except subprocess.CalledProcessError as e:
+            return ("fail", f"{e}")
+
+    def zip7mix(self, file_or_dir, archive_path=None,silent = True):
+        path_obj = Path(file_or_dir)
+        if not path_obj.exists():
+            return None
+
+        if archive_path is None:
+            archive_path = file_or_dir + ".7z"
+
+        tmp_archive_path = archive_path+".tmp"
+
+        tmp_obj = Path(tmp_archive_path)
+        if tmp_obj.exists():
+            tmp_obj.unlink()
+
+        pwd = self._generate_password()
+
+        cmd = [self.zipcmd, "a"]
+        if silent:
+            cmd.append("-bso0")
+        cmd.extend([f"-p{pwd}", "-mhe=on"])
+        cmd.extend([tmp_archive_path, file_or_dir])
+
+        try:
+            subprocess.run(cmd, check=True)
+        except subprocess.CalledProcessError as e:
+            return ("fail", f"{e}")
+        
+        with open(tmp_archive_path,"rb") as f1:
+            data = f1.read()
+
+        # remove tmp file  
+        tmp_obj.unlink()
+        
+        new_data = self.insert_string_to_binary(data,pwd.encode('utf-8'))
+
+        # append fake 7z tail
+        new_data += data[-100:]
+
+        with open(archive_path,"wb") as f2:
+            f2.write(new_data)
+
+        return ("ok", archive_path)
+
+    def unzip7mix(self, archive_path, output_path=None, overwrite: bool = True,silent = True):
+        archive_obj = Path(archive_path)
+
+        tmp_archive_path = archive_path+".tmp"
+
+        tmp_obj = Path(tmp_archive_path)
+        if tmp_obj.exists():
+            tmp_obj.unlink()
+
+        if output_path is None:
+            output_dir = archive_obj.parent
+        else:
+            output_dir = Path(output_path)
+
+        if not output_dir.exists():
+            output_dir.mkdir(parents=True, exist_ok=True)
+
+        if not output_dir.is_dir():
+            return ("fail", f"{output_dir} is not dir")
+        
+        with open(archive_path,"rb") as f1:
+            data = f1.read()
+            # remove fake 7z tail
+            data = data[:-100]
+
+        pwdbts,new_data = self.extract_string_from_binary(data)
+
+        pwd = pwdbts.decode("utf-8")
+
+        with open(tmp_archive_path,"wb") as f2:
+            f2.write(new_data)
+        
+        cmd = [self.zipcmd, "x", "-y"]
+        if silent:
+            cmd.append("-bso0")
+        if not overwrite:
+            cmd.append("-aos")
+        cmd.append(f"-p{pwd}")
+        cmd.extend([tmp_archive_path, f"-o{output_dir}"])
+
+        try:
+            subprocess.run(cmd, check=True)
+            tmp_obj.unlink()
+            return ("ok", f"{output_dir}")
+        except subprocess.CalledProcessError as e:
+            return ("fail", f"{e}")
+
+    def filecmp(self,f1,f2):
+        result = filecmp.cmp(f1, f2, shallow=False)
+        return result

python_plugins/email/__init__.py

@@ -0,0 +1 @@
+from .smtp import SmtpSSL

python_plugins/email/smtp.py

@@ -1,5 +1,6 @@
 import smtplib
 from email.message import EmailMessage
+import mimetypes
 
 
 class SmtpSSL:
@@ -12,11 +13,7 @@ class SmtpSSL:
     def send_emsg(self, data):
         emsg = EmailMessage()
         emsg["Subject"] = data["subject"]
-        emsg.set_content(data["content"])
-
-        if data.get("From") is None:
-            emsg["From"] = self.user
-
+        emsg["From"] = data.get("From", self.user)
         emsg["To"] = data["to"]
 
         if "cc" in data:
@@ -25,7 +22,27 @@ class SmtpSSL:
         if "bcc" in data:
             emsg["Bcc"] = data["bcc"]
 
-        # print(emsg)
+        # Check if HTML content is provided
+        if "html_content" in data:
+            emsg.add_alternative(data["html_content"], subtype="html")
+        else:
+            emsg.set_content(data["content"])
+
+        # Add attachments if provided
+        if "attachments" in data:
+            for file_path in data["attachments"]:
+                ctype, encoding = mimetypes.guess_type(file_path)
+                if ctype is None or encoding is not None:
+                    ctype = "application/octet-stream"
+                maintype, subtype = ctype.split("/", 1)
+
+                with open(file_path, "rb") as f:
+                    emsg.add_attachment(
+                        f.read(),
+                        maintype=maintype,
+                        subtype=subtype,
+                        filename=file_path.split("\\")[-1],
+                    )
 
         with smtplib.SMTP_SSL(self.host, self.port) as smtp:
             # smtp.set_debuglevel(1)

{python_plugins-0.2.2.dist-info → python_plugins-0.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-plugins
-Version: 0.2.2
+Version: 0.2.5
 Summary: A collection of Python functions and classes.
 Project-URL: Documentation, https://python-plugins.readthedocs.io
 Project-URL: Source, https://github.com/ojso/python-plugins

{python_plugins-0.2.2.dist-info → python_plugins-0.2.5.dist-info}/RECORD

@@ -1,4 +1,4 @@
-python_plugins/__about__.py,sha256=m6kyaNpwBcP1XYcqrelX2oS3PJuOnElOcRdBa9pEb8c,22
+python_plugins/__about__.py,sha256=Xsa3ayOMVkhUWm4t06YeyHE0apjpZefxLH4ylp0CDtU,22
 python_plugins/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 python_plugins/convert/__init__.py,sha256=mQeBTbhjuTJkyG4DhXkoX9CNP1TDEgPBFIJ2TQNl2NM,119
 python_plugins/convert/datetime_str.py,sha256=bmC9d1W7XKA1ED9XO9reUiUyfQcM_BqGoGyOF21RW9A,282
@@ -8,10 +8,11 @@ python_plugins/crypto/__init__.py,sha256=2Snt9AIwVsQ-qLNno8qGWs8IyL-AvTImvqkRFO8
 python_plugins/crypto/fernet.py,sha256=UgEAHJGBtmH9AacD-IOyr9dR8T5PKTWL8kH5yj2wKqI,1191
 python_plugins/crypto/file_to_file.py,sha256=mS9llG3j0emhCB4NY6IlKew6wpoLyVW-008ZXN1v3iw,1764
 python_plugins/crypto/str_to_list.py,sha256=T2I4p6bcZuiCsEve4XsOb1q9hFxeVPITstjPyTVoATE,2514
+python_plugins/crypto/zipmix.py,sha256=_pB-Oi7Kl4BEp8S2hynz41_Zblfpf5Z7HzWpaarHRak,8046
 python_plugins/dumps/__init__.py,sha256=TLPuPLIAuULFkiIbFfTCsuhH0CQhPRMUjBmcsr0mtqA,83
 python_plugins/dumps/postgresql_dump.py,sha256=zN5aLYLxpon2y9_5gKizRjlrRUlkAO-dG-BuKpJ4F3U,1030
-python_plugins/email/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_plugins/email/smtp.py,sha256=weSfLVPzROFqwlxDaVhjem522NVEp31lc6PFkbZE1ok,854
+python_plugins/email/__init__.py,sha256=HubZT2h3m5WEjic3NZ0H7gYp-9tUgZaDkAE9O-Tm7hM,26
+python_plugins/email/smtp.py,sha256=CY1fWxdBDqqGZ22R8yqEvFZvMeEsMnqTb4h5IZ8WAVU,1641
 python_plugins/examples/higher_order_functions.py,sha256=cBewE3cDkzaNHR6CQzdfWny0r_V4PgcUeZpHEdCp-E0,633
 python_plugins/hashes/__init__.py,sha256=PQbx8f4klFMg5Q8kehSHz1itj_djWV5hkBT4AMNNUn0,39
 python_plugins/hashes/hash.py,sha256=uSF3ohZRD7b2VAIvjjpTWC0SlY7QsFhsFZUejINrRGU,640
@@ -39,7 +40,7 @@ python_plugins/weixin/format_response.py,sha256=Eml9R1jYYOCtfXORyEWfHbCL8gauWeg2
 python_plugins/weixin/wechat.py,sha256=D6KHZSNj_19vCs5rxLSrHt5XkjkM4ZoD2z6VWD0a_b4,4296
 python_plugins/weixin/wechat_crypt.py,sha256=YOCy4F5ycfZFOJYtKSzhiFSfNb_MsQnN2Qy9u_FfZSU,4717
 python_plugins/weixin/weixin_api.py,sha256=xtnWBovJKNMdm-jXmIM7QHnMye-fZN4a4U_efejRarE,3286
-python_plugins-0.2.2.dist-info/METADATA,sha256=lXLtCipynBJOeJmj8mBv0GgiPzm1kpQaegfOPr2074k,2888
-python_plugins-0.2.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-python_plugins-0.2.2.dist-info/licenses/LICENSE.rst,sha256=X5eLIsAn1yAKd88LWTYkXUe0PmVK_Z5SD7_5NheGR-s,1104
-python_plugins-0.2.2.dist-info/RECORD,,
+python_plugins-0.2.5.dist-info/METADATA,sha256=sZM0pfD-ulzFSNGGFZUQ7ghF4zZTl9C3SsCu78f3YQU,2888
+python_plugins-0.2.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+python_plugins-0.2.5.dist-info/licenses/LICENSE.rst,sha256=X5eLIsAn1yAKd88LWTYkXUe0PmVK_Z5SD7_5NheGR-s,1104
+python_plugins-0.2.5.dist-info/RECORD,,