xtn-tools-pro 1.0.1.2.9__py3-none-any.whl → 1.0.1.3.1__py3-none-any.whl

xtn_tools_pro/utils/crypto_pro.py

@@ -0,0 +1,161 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+# 说明：
+#    加解密、编解码
+# History:
+# Date          Author    Version       Modification
+# --------------------------------------------------------------------------------------------------
+# 2024/5/13    xiatn     V00.01.000    新建
+# --------------------------------------------------------------------------------------------------
+import jwt
+import time
+import json
+import pyotp
+import base64
+import hashlib
+import secrets
+from Crypto.PublicKey import RSA
+from Crypto.Cipher import AES, PKCS1_OAEP
+from Crypto.Util.Padding import pad, unpad
+from Crypto.Random import get_random_bytes
+
+
+class XtnDeEnAir:
+    BLOCK_SIZE = AES.block_size
+
+    @staticmethod
+    def de_data(key: bytes, raw: dict) -> str:
+        """
+            加密
+        """
+        cipher = AES.new(key, AES.MODE_ECB)
+        raw["time"] = int(time.time())
+        raw_s = json.dumps(raw)
+        padded_data = pad(raw_s.encode('utf-8'), XtnDeEnAir.BLOCK_SIZE)
+        enc = cipher.encrypt(padded_data)
+        result = base64.b64encode(enc).decode('utf-8')
+        return result
+
+    @staticmethod
+    def en_data(key: bytes, raw: str) -> str:
+        """
+            解密
+        """
+        cipher = AES.new(key, AES.MODE_ECB)
+        enc = base64.b64decode(raw)
+        decrypted = cipher.decrypt(enc)
+        result = unpad(decrypted, XtnDeEnAir.BLOCK_SIZE).decode('utf-8')
+        return result
+
+
+class XtnDeEnPro:
+    BLOCK_SIZE = AES.block_size
+
+    @staticmethod
+    def encrypt_dict(key: bytes, raw: dict) -> str:
+        """
+            加密
+        """
+        raw["encrypt_time"] = int(time.time())
+        raw_json = json.dumps(data).encode('utf-8')
+        padded_data = pad(raw_json, XtnDeEnPro.BLOCK_SIZE)
+        iv = get_random_bytes(XtnDeEnPro.BLOCK_SIZE)
+        cipher = AES.new(key, AES.MODE_CBC, iv)
+        encrypted = cipher.encrypt(padded_data)
+        combined = iv + encrypted
+        result = base64.b64encode(combined).decode('utf-8')
+        # result = combined.hex() # hex
+        return result
+
+    @staticmethod
+    def decrypt_dict(key: bytes, raw: str) -> dict:
+        """
+            解密
+        """
+        try:
+            combined = base64.b64decode(raw)
+            # combined = bytes.fromhex("") # hex
+            iv = combined[:XtnDeEnPro.BLOCK_SIZE]
+            encrypted = combined[XtnDeEnPro.BLOCK_SIZE:]
+            cipher = AES.new(key, AES.MODE_CBC, iv)
+            decrypted = unpad(cipher.decrypt(encrypted), XtnDeEnPro.BLOCK_SIZE)
+            result = decrypted.decode('utf-8')
+            return json.loads(result)
+        except (ValueError, KeyError, json.JSONDecodeError) as e:
+            raise ValueError(f"解密失败: {e}")
+
+
+class XtnDeEnProMax:
+    BLOCK_SIZE = AES.block_size
+
+    @staticmethod
+    def encrypt_dict(key: bytes, raw: dict) -> str:
+        """
+            加密
+        """
+        session_key = get_random_bytes(16)
+
+        rsa_key = RSA.import_key(key)
+        rsa_cipher = PKCS1_OAEP.new(rsa_key)
+        encrypted_session_key = rsa_cipher.encrypt(session_key)
+
+        iv = get_random_bytes(16)
+        aes_cipher = AES.new(session_key, AES.MODE_CBC, iv)
+        raw["encrypt_time"] = int(time.time())
+        raw_data = json.dumps(raw).encode('utf-8')
+        encrypted_data = aes_cipher.encrypt(pad(raw_data, XtnDeEnProMax.BLOCK_SIZE))
+
+        combined = (
+                len(encrypted_session_key).to_bytes(2, 'big') +
+                encrypted_session_key +
+                iv +
+                encrypted_data
+        )
+
+        result_base64 = base64.b64encode(combined).decode()
+        # result_hex = combined.hex()  # hex
+        return result_base64
+
+    @staticmethod
+    def decrypt_dict(key: bytes, raw: str) -> dict:
+        """
+            解密
+        """
+        try:
+            combined = base64.b64decode(raw)
+            key_len = int.from_bytes(combined[:2], 'big')
+            encrypted_session_key = combined[2:2 + key_len]
+            iv = combined[2 + key_len:2 + key_len + 16]
+            encrypted_data = combined[2 + key_len + 16:]
+
+            rsa_key = RSA.import_key(key)
+            rsa_cipher = PKCS1_OAEP.new(rsa_key)
+            session_key = rsa_cipher.decrypt(encrypted_session_key)
+
+            aes_cipher = AES.new(session_key, AES.MODE_CBC, iv)
+            decrypted = unpad(aes_cipher.decrypt(encrypted_data), XtnDeEnProMax.BLOCK_SIZE)
+            result = decrypted.decode('utf-8')
+            return json.loads(result)
+        except (ValueError, KeyError, json.JSONDecodeError) as e:
+            raise ValueError(f"解密失败: {e}")
+
+
+if __name__ == '__main__':
+    key = b"d9fe4b29dc7decc8360e72e7dd15a5c8"
+    data = {
+        "task_type": "update_name",
+        "task_client": "tiktok",
+    }
+    sss = XtnDeEnAir.de_data(key, data)
+    print(sss)
+    print(XtnDeEnAir.en_data(key, sss))
+
+    key_pair = RSA.generate(2048)
+    private_key = key_pair.export_key()
+    public_key = key_pair.publickey().export_key()
+    print(private_key)
+    print(public_key)
+    sss = XtnDeEnProMax.encrypt_dict(public_key, data)
+    print(sss)
+    print(XtnDeEnProMax.decrypt_dict(private_key, sss))

xtn_tools_pro/utils/log.py

@@ -118,7 +118,7 @@ class Log:
             # 记录当前的日期，用于之后检查日期是否已经改变
             self.current_date = get_time_timestamp_to_datestr(format="%Y_%m_%d")
             # 记录日志文件的路径模板，用于之后创建新的日志文件
-            self.path_template = f"logs/{{date}}/{name}.log"
+            self.path_template = f"{path}/logs/{{date}}/{name}.log"
             path = self.path_template.format(date=self.current_date)
             # print(path)
             if is_write_to_file:

{xtn_tools_pro-1.0.1.2.9.dist-info → xtn_tools_pro-1.0.1.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xtn-tools-pro
-Version: 1.0.1.2.9
+Version: 1.0.1.3.1
 Summary: xtn 开发工具
 Author: xtn
 Author-email: czw011122@gmail.com

{xtn_tools_pro-1.0.1.2.9.dist-info → xtn_tools_pro-1.0.1.3.1.dist-info}/RECORD

@@ -55,17 +55,18 @@ xtn_tools_pro/task_pro/go_fun_v3.py,sha256=GI1OVup-izwDflAaoy_pN6xmCXat59II7v-Mo
 xtn_tools_pro/utils/__init__.py,sha256=I1_n_NP23F2lBqlF4EOlnOdLYxM8M4pbn63UhJN1hRE,418
 xtn_tools_pro/utils/audio_video.py,sha256=3Im3tGOW6IbmqDcrE1SJUg6eGyLGEyq11Qu6TCutJJw,1115
 xtn_tools_pro/utils/crypto.py,sha256=B6NC2yVxtwrbSZnngudTFRm7YRGKKEusydVVYWwxFHs,5231
+xtn_tools_pro/utils/crypto_pro.py,sha256=CJqK-fJryblKsj2uQN7_Sy8gn7HPCeage53xk1DfQLc,5331
 xtn_tools_pro/utils/file_utils.py,sha256=vsUgds8I8Z2pL9a4oCv1q0JrZsDP29qLyLyGVqj5sX8,3311
 xtn_tools_pro/utils/helpers.py,sha256=0CTMY7wfSQR_DC9-v1lkKQLpWcB1FL9V_kw_5DOcZ40,4454
-xtn_tools_pro/utils/log.py,sha256=DTUWwi5pvZrwwq94odGnR0JBhfStrc6SxcJREqf2Ls4,11031
+xtn_tools_pro/utils/log.py,sha256=MPT9q1keNYroaw6U9a-Gc0iT0ceUEVTI-Tz2GEiWLUc,11038
 xtn_tools_pro/utils/retry.py,sha256=0wjHsR5DBBKpv4naMfxiky8kprrZes4WURIfFQ4H708,1657
 xtn_tools_pro/utils/set_data.py,sha256=UR5S8xBqc4Vk8zpbrNVsxKhwaBAenaEdC7O-N3s7haw,17767
 xtn_tools_pro/utils/shell.py,sha256=s6sJedxLLQokcI1hF5YSD3qgGUPK0brNcP-D3-yv54I,3790
 xtn_tools_pro/utils/sql.py,sha256=EAKzbkZP7Q09j15Gm6o0_uq0qgQmcCQT6EAawbpp4v0,6263
 xtn_tools_pro/utils/time_utils.py,sha256=TUtzG61PeVYXhaQd6pBrXAdlz7tBispNIRQRcGhE2No,4859
-xtn_tools_pro-1.0.1.2.9.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-xtn_tools_pro-1.0.1.2.9.dist-info/METADATA,sha256=t0tt-U4bOu4a1V8WMeaReYMqt7KI2zpAAdO37MDe890,574
-xtn_tools_pro-1.0.1.2.9.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-xtn_tools_pro-1.0.1.2.9.dist-info/entry_points.txt,sha256=t8CtXWOgw7nRDW3XNlZh8MT_P4l8EzsFnyWi5b3ovrc,68
-xtn_tools_pro-1.0.1.2.9.dist-info/top_level.txt,sha256=jyB3FLDEr8zE1U7wHczTgIbvUpALhR-ULF7RVEO7O2U,14
-xtn_tools_pro-1.0.1.2.9.dist-info/RECORD,,
+xtn_tools_pro-1.0.1.3.1.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+xtn_tools_pro-1.0.1.3.1.dist-info/METADATA,sha256=Ivrk4YS7pi60sMP7xUq0323sBBR-9Cpfn3u8MkOkML4,574
+xtn_tools_pro-1.0.1.3.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+xtn_tools_pro-1.0.1.3.1.dist-info/entry_points.txt,sha256=t8CtXWOgw7nRDW3XNlZh8MT_P4l8EzsFnyWi5b3ovrc,68
+xtn_tools_pro-1.0.1.3.1.dist-info/top_level.txt,sha256=jyB3FLDEr8zE1U7wHczTgIbvUpALhR-ULF7RVEO7O2U,14
+xtn_tools_pro-1.0.1.3.1.dist-info/RECORD,,