Files
RiceRound-ComfyUI_CryptoCat/file_compressor.py
T
RiceRound e614d4e974 Major version update with enhanced functionality
Added environment-based configuration to streamline deployments
Improved data validation logic for more robust CRUD operations
Refined folder structure to optimize code organization
2025-01-14 04:20:56 +08:00

44 lines
1.5 KiB
Python

import json
import zlib
import os
class FileCompressor:
@staticmethod
def obfuscate(data, password):
return bytes(
[b ^ ord(password[i % len(password)]) for (i, b) in enumerate(data)]
)
@staticmethod
def compress_string(string_data, zip_path, password):
compressed_data = zlib.compress(string_data.encode("utf-8"))
obfuscated_data = FileCompressor.obfuscate(compressed_data, password)
with open(zip_path, "wb") as zipf:
zipf.write(obfuscated_data)
@staticmethod
def decompress_to_string(zip_path, password):
with open(zip_path, "rb") as zipf:
obfuscated_data = zipf.read()
compressed_data = FileCompressor.obfuscate(obfuscated_data, password)
return zlib.decompress(compressed_data).decode("utf-8")
@staticmethod
def compress_to_json(data, zip_path, password):
try:
json_string = json.dumps(data, indent=4, ensure_ascii=False)
FileCompressor.compress_string(json_string, zip_path, password)
except Exception as e:
print(f"Compression failed: {e}")
@staticmethod
def decompress_from_json(zip_path, password):
try:
if not os.path.exists(zip_path):
raise FileNotFoundError(f"{zip_path} does not exist")
json_string = FileCompressor.decompress_to_string(zip_path, password)
return json.loads(json_string)
except Exception as e:
print(f"Decompression failed: {e}")