Class StegaPyPlugin

StegaPy/plugin/base.py:33–115

StegaPy插件基类

class StegaPyPlugin:
    """StegaPy插件基类"""
    
    def __init__(self, config: Optional[StegaPyConfig] = None):
        """初始化插件"""
        self.config = config or self.create_config()
    
    def get_name(self) -> str:
        """获取插件名称"""
        raise NotImplementedError
    
    def get_purposes(self) -> List[Purpose]:
        """获取插件支持的用途"""
        raise NotImplementedError
    
    def get_description(self) -> str:
        """获取插件描述"""
        raise NotImplementedError
    
    def embed_data(self, msg: bytes, msg_filename: Optional[str], 
                   cover: bytes, cover_filename: Optional[str], 
                   stego_filename: Optional[str]) -> bytes:
        """嵌入数据到封面图像"""
        raise NotImplementedError
    
    def extract_msg_filename(self, stego_data: bytes, 
                             stego_filename: Optional[str]) -> str:
        """从隐写数据中提取消息文件名"""
        raise NotImplementedError
    
    def extract_data(self, stego_data: bytes, stego_filename: Optional[str],
                     orig_sig_data: Optional[bytes] = None) -> bytes:
        """从隐写数据中提取消息"""
        raise NotImplementedError
    
    def generate_signature(self) -> bytes:
        """生成签名数据（用于水印）"""
        raise NotImplementedError
    
    def check_mark(self, stego_data: bytes, stego_filename: Optional[str],
                   orig_sig_data: bytes) -> float:
        """检查水印相关性"""
        watermark_data = self.extract_data(stego_data, stego_filename, orig_sig_data)
        return self.get_watermark_correlation(orig_sig_data, watermark_data)
    
    def get_watermark_correlation(self, orig_sig_data: bytes, 
                                   watermark_data: bytes) -> float:
        """获取水印相关性"""
        raise NotImplementedError
    
    def get_high_watermark_level(self) -> float:
        """获取高水印阈值"""
        raise NotImplementedError
    
    def get_low_watermark_level(self) -> float:
        """获取低水印阈值"""
        raise NotImplementedError