mirror of
https://github.com/sun-guannan/CapCutAPI.git
synced 2025-11-25 03:15:00 +08:00
74 lines
2.8 KiB
Python
74 lines
2.8 KiB
Python
from pyJianYingDraft import trange, Video_scene_effect_type, Video_character_effect_type, CapCut_Video_scene_effect_type, CapCut_Video_character_effect_type, exceptions
|
||
import pyJianYingDraft as draft
|
||
from typing import Optional, Dict, List, Union
|
||
from create_draft import get_or_create_draft
|
||
from util import generate_draft_url
|
||
from settings import IS_CAPCUT_ENV
|
||
|
||
def add_effect_impl(
|
||
effect_type: str, # 修改为字符串类型
|
||
start: float = 0,
|
||
end: float = 3.0,
|
||
draft_id: Optional[str] = None,
|
||
track_name: Optional[str] = "effect_01",
|
||
params: Optional[List[Optional[float]]] = None,
|
||
width: int = 1080,
|
||
height: int = 1920
|
||
) -> Dict[str, str]:
|
||
"""
|
||
向指定草稿添加特效
|
||
:param effect_type: 特效类型名称，将从Video_scene_effect_type或Video_character_effect_type中匹配
|
||
:param start: 开始时间（秒），默认0
|
||
:param end: 结束时间（秒），默认3秒
|
||
:param draft_id: 草稿ID，如果为None或找不到对应的zip文件，则创建新草稿
|
||
:param track_name: 轨道名称，当特效轨道仅有一条时可省略
|
||
:param params: 特效参数列表，参数列表中未提供或为None的项使用默认值
|
||
:param width: 视频宽度，默认1080
|
||
:param height: 视频高度，默认1920
|
||
:return: 更新后的草稿信息
|
||
"""
|
||
# 获取或创建草稿
|
||
draft_id, script = get_or_create_draft(
|
||
draft_id=draft_id,
|
||
width=width,
|
||
height=height
|
||
)
|
||
|
||
# 计算时间范围
|
||
duration = end - start
|
||
t_range = trange(f"{start}s", f"{duration}s")
|
||
|
||
# 动态获取特效类型对象
|
||
if IS_CAPCUT_ENV:
|
||
# 如果是CapCut环境，使用CapCut特效
|
||
effect_enum = CapCut_Video_scene_effect_type[effect_type]
|
||
if effect_enum is None:
|
||
effect_enum = CapCut_Video_character_effect_type[effect_type]
|
||
else:
|
||
# 默认使用剪映特效
|
||
effect_enum = Video_scene_effect_type[effect_type]
|
||
if effect_enum is None:
|
||
effect_enum = Video_character_effect_type[effect_type]
|
||
|
||
if effect_enum is None:
|
||
raise ValueError(f"Unknown effect type: {effect_type}")
|
||
|
||
# 添加特效轨道（仅当轨道不存在时）
|
||
if track_name is not None:
|
||
try:
|
||
imported_track=script.get_imported_track(draft.Track_type.effect, name=track_name)
|
||
# 如果没有抛出异常，说明轨道已存在
|
||
except exceptions.TrackNotFound:
|
||
# 轨道不存在，创建新轨道
|
||
script.add_track(draft.Track_type.effect, track_name=track_name)
|
||
else:
|
||
script.add_track(draft.Track_type.effect)
|
||
|
||
# 添加特效
|
||
script.add_effect(effect_enum, t_range, params=params[::-1], track_name=track_name)
|
||
|
||
return {
|
||
"draft_id": draft_id,
|
||
"draft_url": generate_draft_url(draft_id)
|
||
}
|