{
  "file_path": "poster/poster_generator.py",
  "file_size": 3654,
  "line_count": 120,
  "functions": [
    {
      "name": "__init__",
      "line_start": 32,
      "line_end": 50,
      "args": [
        {
          "name": "self"
        },
        {
          "name": "config_manager",
          "type_hint": "ConfigManager"
        },
        {
          "name": "output_manager",
          "type_hint": "OutputManager"
        }
      ],
      "return_type": null,
      "docstring": "初始化海报生成器\n\nArgs:\n    config_manager: 配置管理器\n    output_manager: 输出管理器",
      "is_async": false,
      "decorators": [],
      "code": "    def __init__(self, config_manager: ConfigManager, output_manager: OutputManager):\n        \"\"\"\n        初始化海报生成器\n\n        Args:\n            config_manager: 配置管理器\n            output_manager: 输出管理器\n        \"\"\"\n        self.config_manager = config_manager\n        self.config = config_manager.get_config('poster', PosterConfig)\n        self.output_manager = output_manager\n        self.image_processor = ImageProcessor()\n        \n        # 注册可用的模板\n        self.templates = {\n            \"vibrant\": VibrantTemplate,\n            \"business\": BusinessTemplate,\n            \"collage\": CollageTemplate,\n        }",
      "code_hash": "9adc80968510375151f85be39c66893a"
    },
    {
      "name": "generate_poster",
      "line_start": 52,
      "line_end": 101,
      "args": [
        {
          "name": "self"
        },
        {
          "name": "content",
          "type_hint": "Dict[str, Any]"
        },
        {
          "name": "topic_index",
          "type_hint": "Union[int, str]"
        },
        {
          "name": "template_name",
          "type_hint": "Optional[str]"
        }
      ],
      "return_type": "Optional[str]",
      "docstring": "为指定内容生成海报\n\nArgs:\n    content: 内容数据，包含标题、正文等\n    topic_index: 主题索引，用于文件命名\n    template_name: 模板名称，如果为None则根据配置选择\n\nReturns:\n    生成的海报文件路径，如果生成失败则返回None",
      "is_async": false,
      "decorators": [],
      "code": "    def generate_poster(self, content: Dict[str, Any], topic_index: Union[int, str], \n                       template_name: Optional[str] = None) -> Optional[str]:\n        \"\"\"\n        为指定内容生成海报\n\n        Args:\n            content: 内容数据，包含标题、正文等\n            topic_index: 主题索引，用于文件命名\n            template_name: 模板名称，如果为None则根据配置选择\n\n        Returns:\n            生成的海报文件路径，如果生成失败则返回None\n        \"\"\"\n        logger.info(f\"开始为主题 {topic_index} 生成海报...\")\n        \n        # 1. 选择模板\n        if not template_name:\n            template_name = self._select_template()\n        \n        if template_name not in self.templates:\n            logger.error(f\"模板 '{template_name}' 不存在\")\n            return None\n        \n        template_class = self.templates[template_name]\n        template = template_class(self.config.target_size)\n        \n        # 2. 准备内容\n        title = content.get('title', '')\n        text_content = content.get('content', '')\n        \n        if not title or not text_content:\n            logger.error(\"内容缺少标题或正文\")\n            return None\n        \n        # 3. 生成海报\n        try:\n            poster_image = template.generate(title, text_content)\n            \n            # 4. 保存海报\n            output_dir = self.output_manager.get_topic_dir(topic_index)\n            file_name = f\"poster_{template_name}.png\"\n            file_path = output_dir / file_name\n            \n            poster_image.save(file_path)\n            logger.info(f\"海报已保存到: {file_path}\")\n            \n            return str(file_path)\n        except Exception as e:\n            logger.error(f\"生成海报时发生错误: {e}\", exc_info=True)\n            return None",
      "code_hash": "2b0daee36c09366b3a71e972294810c6"
    },
    {
      "name": "_select_template",
      "line_start": 103,
      "line_end": 121,
      "args": [
        {
          "name": "self"
        }
      ],
      "return_type": "str",
      "docstring": "根据配置选择模板",
      "is_async": false,
      "decorators": [],
      "code": "    def _select_template(self) -> str:\n        \"\"\"根据配置选择模板\"\"\"\n        selection_method = self.config.template_selection\n        available_templates = self.config.available_templates\n        \n        # 过滤出实际可用的模板\n        valid_templates = [t for t in available_templates if t in self.templates]\n        \n        if not valid_templates:\n            logger.warning(\"没有可用的模板，使用默认模板\")\n            return next(iter(self.templates.keys()))\n        \n        if selection_method == \"random\":\n            return random.choice(valid_templates)\n        elif selection_method in valid_templates:\n            return selection_method\n        else:\n            logger.warning(f\"无效的模板选择方法 '{selection_method}'，使用随机选择\")\n            return random.choice(valid_templates) ",
      "code_hash": "3321717ec7994b4cdcf0ac6e004298c7"
    }
  ],
  "classes": [
    {
      "name": "PosterGenerator",
      "line_start": 26,
      "line_end": 121,
      "bases": [],
      "methods": [
        {
          "name": "__init__",
          "line_start": 32,
          "line_end": 50,
          "args": [
            {
              "name": "self"
            },
            {
              "name": "config_manager",
              "type_hint": "ConfigManager"
            },
            {
              "name": "output_manager",
              "type_hint": "OutputManager"
            }
          ],
          "return_type": null,
          "docstring": "初始化海报生成器\n\nArgs:\n    config_manager: 配置管理器\n    output_manager: 输出管理器",
          "is_async": false,
          "decorators": [],
          "code": "    def __init__(self, config_manager: ConfigManager, output_manager: OutputManager):\n        \"\"\"\n        初始化海报生成器\n\n        Args:\n            config_manager: 配置管理器\n            output_manager: 输出管理器\n        \"\"\"\n        self.config_manager = config_manager\n        self.config = config_manager.get_config('poster', PosterConfig)\n        self.output_manager = output_manager\n        self.image_processor = ImageProcessor()\n        \n        # 注册可用的模板\n        self.templates = {\n            \"vibrant\": VibrantTemplate,\n            \"business\": BusinessTemplate,\n            \"collage\": CollageTemplate,\n        }",
          "code_hash": "9adc80968510375151f85be39c66893a"
        },
        {
          "name": "generate_poster",
          "line_start": 52,
          "line_end": 101,
          "args": [
            {
              "name": "self"
            },
            {
              "name": "content",
              "type_hint": "Dict[str, Any]"
            },
            {
              "name": "topic_index",
              "type_hint": "Union[int, str]"
            },
            {
              "name": "template_name",
              "type_hint": "Optional[str]"
            }
          ],
          "return_type": "Optional[str]",
          "docstring": "为指定内容生成海报\n\nArgs:\n    content: 内容数据，包含标题、正文等\n    topic_index: 主题索引，用于文件命名\n    template_name: 模板名称，如果为None则根据配置选择\n\nReturns:\n    生成的海报文件路径，如果生成失败则返回None",
          "is_async": false,
          "decorators": [],
          "code": "    def generate_poster(self, content: Dict[str, Any], topic_index: Union[int, str], \n                       template_name: Optional[str] = None) -> Optional[str]:\n        \"\"\"\n        为指定内容生成海报\n\n        Args:\n            content: 内容数据，包含标题、正文等\n            topic_index: 主题索引，用于文件命名\n            template_name: 模板名称，如果为None则根据配置选择\n\n        Returns:\n            生成的海报文件路径，如果生成失败则返回None\n        \"\"\"\n        logger.info(f\"开始为主题 {topic_index} 生成海报...\")\n        \n        # 1. 选择模板\n        if not template_name:\n            template_name = self._select_template()\n        \n        if template_name not in self.templates:\n            logger.error(f\"模板 '{template_name}' 不存在\")\n            return None\n        \n        template_class = self.templates[template_name]\n        template = template_class(self.config.target_size)\n        \n        # 2. 准备内容\n        title = content.get('title', '')\n        text_content = content.get('content', '')\n        \n        if not title or not text_content:\n            logger.error(\"内容缺少标题或正文\")\n            return None\n        \n        # 3. 生成海报\n        try:\n            poster_image = template.generate(title, text_content)\n            \n            # 4. 保存海报\n            output_dir = self.output_manager.get_topic_dir(topic_index)\n            file_name = f\"poster_{template_name}.png\"\n            file_path = output_dir / file_name\n            \n            poster_image.save(file_path)\n            logger.info(f\"海报已保存到: {file_path}\")\n            \n            return str(file_path)\n        except Exception as e:\n            logger.error(f\"生成海报时发生错误: {e}\", exc_info=True)\n            return None",
          "code_hash": "2b0daee36c09366b3a71e972294810c6"
        },
        {
          "name": "_select_template",
          "line_start": 103,
          "line_end": 121,
          "args": [
            {
              "name": "self"
            }
          ],
          "return_type": "str",
          "docstring": "根据配置选择模板",
          "is_async": false,
          "decorators": [],
          "code": "    def _select_template(self) -> str:\n        \"\"\"根据配置选择模板\"\"\"\n        selection_method = self.config.template_selection\n        available_templates = self.config.available_templates\n        \n        # 过滤出实际可用的模板\n        valid_templates = [t for t in available_templates if t in self.templates]\n        \n        if not valid_templates:\n            logger.warning(\"没有可用的模板，使用默认模板\")\n            return next(iter(self.templates.keys()))\n        \n        if selection_method == \"random\":\n            return random.choice(valid_templates)\n        elif selection_method in valid_templates:\n            return selection_method\n        else:\n            logger.warning(f\"无效的模板选择方法 '{selection_method}'，使用随机选择\")\n            return random.choice(valid_templates) ",
          "code_hash": "3321717ec7994b4cdcf0ac6e004298c7"
        }
      ],
      "docstring": "海报生成器\n负责根据文本内容和配置生成海报图像",
      "decorators": [],
      "code": "class PosterGenerator:\n    \"\"\"\n    海报生成器\n    负责根据文本内容和配置生成海报图像\n    \"\"\"\n\n    def __init__(self, config_manager: ConfigManager, output_manager: OutputManager):\n        \"\"\"\n        初始化海报生成器\n\n        Args:\n            config_manager: 配置管理器\n            output_manager: 输出管理器\n        \"\"\"\n        self.config_manager = config_manager\n        self.config = config_manager.get_config('poster', PosterConfig)\n        self.output_manager = output_manager\n        self.image_processor = ImageProcessor()\n        \n        # 注册可用的模板\n        self.templates = {\n            \"vibrant\": VibrantTemplate,\n            \"business\": BusinessTemplate,\n            \"collage\": CollageTemplate,\n        }\n\n    def generate_poster(self, content: Dict[str, Any], topic_index: Union[int, str], \n                       template_name: Optional[str] = None) -> Optional[str]:\n        \"\"\"\n        为指定内容生成海报\n\n        Args:\n            content: 内容数据，包含标题、正文等\n            topic_index: 主题索引，用于文件命名\n            template_name: 模板名称，如果为None则根据配置选择\n\n        Returns:\n            生成的海报文件路径，如果生成失败则返回None\n        \"\"\"\n        logger.info(f\"开始为主题 {topic_index} 生成海报...\")\n        \n        # 1. 选择模板\n        if not template_name:\n            template_name = self._select_template()\n        \n        if template_name not in self.templates:\n            logger.error(f\"模板 '{template_name}' 不存在\")\n            return None\n        \n        template_class = self.templates[template_name]\n        template = template_class(self.config.target_size)\n        \n        # 2. 准备内容\n        title = content.get('title', '')\n        text_content = content.get('content', '')\n        \n        if not title or not text_content:\n            logger.error(\"内容缺少标题或正文\")\n            return None\n        \n        # 3. 生成海报\n        try:\n            poster_image = template.generate(title, text_content)\n            \n            # 4. 保存海报\n            output_dir = self.output_manager.get_topic_dir(topic_index)\n            file_name = f\"poster_{template_name}.png\"\n            file_path = output_dir / file_name\n            \n            poster_image.save(file_path)\n            logger.info(f\"海报已保存到: {file_path}\")\n            \n            return str(file_path)\n        except Exception as e:\n            logger.error(f\"生成海报时发生错误: {e}\", exc_info=True)\n            return None\n    \n    def _select_template(self) -> str:\n        \"\"\"根据配置选择模板\"\"\"\n        selection_method = self.config.template_selection\n        available_templates = self.config.available_templates\n        \n        # 过滤出实际可用的模板\n        valid_templates = [t for t in available_templates if t in self.templates]\n        \n        if not valid_templates:\n            logger.warning(\"没有可用的模板，使用默认模板\")\n            return next(iter(self.templates.keys()))\n        \n        if selection_method == \"random\":\n            return random.choice(valid_templates)\n        elif selection_method in valid_templates:\n            return selection_method\n        else:\n            logger.warning(f\"无效的模板选择方法 '{selection_method}'，使用随机选择\")\n            return random.choice(valid_templates) ",
      "code_hash": "aaff6457a0208a567d7dbb4767bd39d5"
    }
  ],
  "imports": [
    {
      "type": "import",
      "modules": [
        "os"
      ],
      "aliases": []
    },
    {
      "type": "import",
      "modules": [
        "logging"
      ],
      "aliases": []
    },
    {
      "type": "import",
      "modules": [
        "random"
      ],
      "aliases": []
    },
    {
      "type": "from_import",
      "module": "typing",
      "names": [
        "List",
        "Dict",
        "Any",
        "Optional",
        "Union",
        "Tuple"
      ],
      "aliases": [],
      "level": 0
    },
    {
      "type": "from_import",
      "module": "pathlib",
      "names": [
        "Path"
      ],
      "aliases": [],
      "level": 0
    },
    {
      "type": "from_import",
      "module": "PIL",
      "names": [
        "Image"
      ],
      "aliases": [],
      "level": 0
    },
    {
      "type": "from_import",
      "module": "core.config",
      "names": [
        "ConfigManager",
        "PosterConfig"
      ],
      "aliases": [],
      "level": 0
    },
    {
      "type": "from_import",
      "module": "utils.file_io",
      "names": [
        "OutputManager"
      ],
      "aliases": [],
      "level": 0
    },
    {
      "type": "from_import",
      "module": "templates.base_template",
      "names": [
        "BaseTemplate"
      ],
      "aliases": [],
      "level": 1
    },
    {
      "type": "from_import",
      "module": "templates.vibrant_template",
      "names": [
        "VibrantTemplate"
      ],
      "aliases": [],
      "level": 1
    },
    {
      "type": "from_import",
      "module": "templates.business_template",
      "names": [
        "BusinessTemplate"
      ],
      "aliases": [],
      "level": 1
    },
    {
      "type": "from_import",
      "module": "templates.collage_template",
      "names": [
        "CollageTemplate"
      ],
      "aliases": [],
      "level": 1
    },
    {
      "type": "from_import",
      "module": "utils",
      "names": [
        "ImageProcessor"
      ],
      "aliases": [],
      "level": 1
    }
  ],
  "constants": [],
  "docstring": "海报生成器模块",
  "content_hash": "9ff80767c00a954140776bb5a3144ec9"
}