Class StegaPy

StegaPy/StegaPy.py:30–236

StegaPy 核心调度与管理类，提供信息隐藏与水印功能的统一对外接口。

class StegaPy:
    """StegaPy 核心调度与管理类，提供信息隐藏与水印功能的统一对外接口。"""
    
    NAMESPACE = "StegaPy"
    
    def __init__(self, plugin: StegaPyPlugin, config: StegaPyConfig):
        """初始化 StegaPy 实例，绑定核心插件与系统配置。"""
        if plugin is None:
            raise StegaPyException(
                "未指定插件",
                StegaPyErrors.NO_PLUGIN_SPECIFIED,
                self.NAMESPACE
            )
        if config is None:
            raise StegaPyException(
                "未指定配置",
                StegaPyErrors.UNHANDLED_EXCEPTION,
                self.NAMESPACE
            )
        
        self.plugin = plugin
        self.config = config
    
    def embed_data(self, msg: bytes, msg_filename: Optional[str],
                   cover: Optional[bytes], cover_filename: Optional[str],
                   stego_filename: Optional[str]) -> bytes:
        """将机密信息（Payload）嵌入至载体图像中，并返回隐写后的图像字节流。"""
        if Purpose.DATA_HIDING not in self.plugin.get_purposes():
            raise StegaPyException(
                "插件不支持数据隐藏",
                StegaPyErrors.PLUGIN_DOES_NOT_SUPPORT_DH,
                self.NAMESPACE
            )
        
        try:
            # 压缩数据（如果启用）
            if self.config.is_use_compression():
                msg = self._compress_data(msg)
            
            # 加密数据（如果启用）
            if self.config.is_use_encryption():
                if not self.config.get_password():
                    raise StegaPyException(
                        "加密需要密码",
                        StegaPyErrors.INVALID_PASSWORD,
                        self.NAMESPACE
                    )
                crypto = CryptoUtil(self.config.get_password(), 
                                  self.config.get_encryption_algorithm())
                msg = crypto.encrypt(msg)
            
            # 使用插件嵌入数据
            return self.plugin.embed_data(msg, msg_filename, cover, 
                                         cover_filename, stego_filename)
        except StegaPyException:
            raise
        except Exception as e:
            raise StegaPyException(str(e), StegaPyErrors.UNHANDLED_EXCEPTION, self.NAMESPACE)