jianying-draft 1.0.0__tar.gz

jianying_draft-1.0.0/PKG-INFO

@@ -0,0 +1,6 @@
+Metadata-Version: 2.4
+Name: jianying-draft
+Version: 1.0.0
+Summary: utility
+Requires-Python: >=3.9
+Requires-Dist: cryptography>=41.0.0

jianying_draft-1.0.0/jianying_draft/__init__.py

@@ -0,0 +1,488 @@
+"""
+zcatwx-zil6sm6h-wyaaz
+======================
+
+公开 API:
+  generate_draft_json(draft_context)  -> str   把 DraftContext 还原为加密的剪映草稿 JSON
+  DraftCrypto.encrypt(plaintext)      -> str   AES-GCM 加密
+  DraftCrypto.decrypt(ciphertext)     -> str   AES-GCM 解密
+"""
+
+from __future__ import annotations
+
+import base64
+import json
+import os
+import sys
+
+from cryptography.hazmat.primitives.ciphers.aead import AESGCM
+
+# ── OOP 核心 ──────────────────────────────────────────────────────────────────
+from zcatwx_zil6sm6h_wyaaz._core import (
+    JianYingDraft, VideoClip, AudioClip, TextClip,
+    StickerClip, EffectClip, FilterClip,
+)
+
+# ── 字体名称 → ID 查找表 ──────────────────────────────────────────────────────
+def _load_font_name_to_id():
+    db_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "font_db.json")
+    try:
+        with open(db_path, "r", encoding="utf-8") as f:
+            db = json.load(f)
+        return {name: fid for fid, name in db.items()}
+    except Exception:
+        return {}
+
+_FONT_NAME_TO_ID = _load_font_name_to_id()
+
+
+def _resolve_font_id(style: dict):
+    fid = style.get("font_id")
+    if fid:
+        return fid
+    fname = style.get("font_name")
+    if fname and fname in _FONT_NAME_TO_ID:
+        return _FONT_NAME_TO_ID[fname]
+    return None
+
+
+# ── 加解密 ────────────────────────────────────────────────────────────────────
+class DraftCrypto:
+    """AES-GCM 加解密（与 Coze 节点保持相同密钥）"""
+
+    _KEY = "MySecretKeyForCozeNode20260307!!".encode("utf-8")
+
+    @classmethod
+    def encrypt(cls, plaintext: str) -> str:
+        import os as _os
+        aesgcm = AESGCM(cls._KEY)
+        iv = _os.urandom(12)
+        ciphertext = aesgcm.encrypt(iv, plaintext.encode("utf-8"), None)
+        return base64.b64encode(iv + ciphertext).decode("utf-8")
+
+    @classmethod
+    def decrypt(cls, payload: str) -> str:
+        try:
+            aesgcm = AESGCM(cls._KEY)
+            raw = base64.b64decode(payload.encode("utf-8"))
+            iv, ciphertext = raw[:12], raw[12:]
+            return aesgcm.decrypt(iv, ciphertext, None).decode("utf-8")
+        except Exception:
+            return payload  # 兼容明文透传
+
+
+# ── 时间解析 ──────────────────────────────────────────────────────────────────
+_UNIT_MULTIPLIER = {"us": 1, "s": 1_000_000}
+_tm = 1  # 由 generate_draft_json() 入口按 time_unit 字段更新
+
+
+def _parse_time(t) -> int:
+    if t is None or t == "auto":
+        return 0
+    if isinstance(t, str):
+        t = t.strip()
+        if t.endswith("us"):
+            return int(float(t[:-2]))
+        elif t.endswith("s"):
+            return int(float(t[:-1]) * 1_000_000)
+        else:
+            return int(float(t) * _tm)
+    return int(t * _tm)
+
+
+# ── 核心：DraftContext → 完整剪映草稿 JSON 字符串 ────────────────────────────
+def generate_draft_json(draft_context) -> str:
+    """
+    把 Coze 节点生成的 DraftContext（中间层 JSON）还原为完整的剪映草稿 JSON 字符串。
+
+    参数:
+        draft_context: str 或 dict
+            - str: 明文 JSON 字符串 或 AES-GCM 加密的 base64 密文
+            - dict: 已解析的 DraftContext 字典
+
+    返回:
+        str: 加密后的完整剪映草稿 JSON 字符串（可直接用 DraftCrypto.decrypt 解密）
+
+    说明:
+        - 素材路径（视频/音频/图片 URL）保留原样，不下载
+        - 本地 export_draft.py 负责下载资源并替换路径
+    """
+    global _tm
+
+    # 1. 解析输入（支持密文、明文字符串、dict）
+    if isinstance(draft_context, str):
+        decrypted = DraftCrypto.decrypt(draft_context)
+        try:
+            data = json.loads(decrypted)
+        except json.JSONDecodeError as e:
+            raise ValueError(f"DraftContext 解析失败: {e}") from e
+    else:
+        data = draft_context
+
+    # 2. 初始化项目
+    proj = data.get("project_info", {})
+    width     = proj.get("width", 1920)
+    height    = proj.get("height", 1080)
+    ratio     = proj.get("ratio", "custom")
+    save_name = proj.get("save_name", "Coze_Auto_Gen")
+
+    _tm = _UNIT_MULTIPLIER.get(data.get("time_unit", "us"), 1)
+
+    draft = JianYingDraft(width=width, height=height, ratio=ratio)
+
+    # 3. 遍历轨道
+    clip_map: dict = {}
+    for track_info in data.get("tracks", []):
+        track_type = track_info.get("type", "video")
+        oop_track_type = {"image": "video", "sticker": "video"}.get(track_type, track_type)
+        current_track = draft.add_track(oop_track_type)
+
+        for clip_data in track_info.get("clips", []):
+            clip_type = clip_data.get("type", "video")
+            clip_id   = clip_data.get("id")
+            start     = _parse_time(clip_data.get("start"))
+            duration  = _parse_time(clip_data.get("duration"))
+            clip_obj  = None
+
+            # ── A. 视频 / 图片 ──────────────────────────────────────────
+            if clip_type in ("video", "image"):
+                path = clip_data.get("path")
+                if not path:
+                    continue
+                source_start = _parse_time(clip_data.get("source_start", 0))
+                clip_obj = VideoClip(path, duration=duration, source_start=source_start, start=start)
+
+                speed_cfg = clip_data.get("speed")
+                if speed_cfg:
+                    if isinstance(speed_cfg, dict):
+                        clip_obj.set_speed(speed_cfg.get("value", 1.0),
+                                           change_pitch=speed_cfg.get("change_pitch", False))
+                    else:
+                        clip_obj.set_speed(float(speed_cfg))
+
+                transition = clip_data.get("transition")
+                if transition:
+                    clip_obj.add_transition(
+                        resource_id=transition.get("resource_id"),
+                        duration=_parse_time(transition.get("duration", 1_000_000)),
+                        overlap=transition.get("overlap", True),
+                    )
+
+                mask = clip_data.get("mask")
+                if mask:
+                    clip_obj.set_mask(
+                        mask_type=mask.get("type", "rectangle"),
+                        center_x=mask.get("center_x", 0),
+                        center_y=mask.get("center_y", 0),
+                        rotation=mask.get("rotation", 0),
+                        size_x=mask.get("size_x"),
+                        size_y=mask.get("size_y"),
+                        feather=mask.get("feather", 0),
+                        invert=mask.get("invert", False),
+                        round_corner=mask.get("round_corner", 0),
+                    )
+
+                mix_mode = clip_data.get("mix_mode")
+                if mix_mode:
+                    mode_name = mix_mode if isinstance(mix_mode, str) else mix_mode.get("name", "变亮")
+                    clip_obj.set_mix_mode(name=mode_name)
+
+                smart_crop = clip_data.get("smart_crop")
+                if smart_crop:
+                    clip_obj.set_smart_crop(
+                        ratio=smart_crop.get("ratio", "original"),
+                        stability=smart_crop.get("stability", 2),
+                        speed=smart_crop.get("speed", 1),
+                    )
+
+                video_algo = clip_data.get("video_algorithm")
+                if video_algo:
+                    clip_obj.set_video_algorithm(
+                        denoise=video_algo.get("denoise"),
+                        deflicker=video_algo.get("deflicker"),
+                        quality_enhance=video_algo.get("quality_enhance"),
+                        motion_blur=video_algo.get("motion_blur"),
+                        stabilize=video_algo.get("stabilize"),
+                    )
+
+                if clip_data.get("audio_volume") is not None:
+                    clip_obj.set_volume(clip_data["audio_volume"])
+                audio_fade = clip_data.get("audio_fade")
+                if audio_fade:
+                    clip_obj.set_audio_fade(
+                        fade_in=_parse_time(audio_fade.get("in", 0)),
+                        fade_out=_parse_time(audio_fade.get("out", 0)),
+                    )
+                if clip_data.get("audio_denoise"):
+                    clip_obj.set_audio_denoise(True)
+                if clip_data.get("loudness"):
+                    clip_obj.set_loudness(clip_data["loudness"])
+                if clip_data.get("vocal_separation") is not None:
+                    clip_obj.set_vocal_separation(clip_data["vocal_separation"])
+                if clip_data.get("vocal_beautify") is not None:
+                    clip_obj.set_vocal_beautify(clip_data["vocal_beautify"])
+                if clip_data.get("tone_effect"):
+                    clip_obj.set_tone_effect(clip_data["tone_effect"])
+                if clip_data.get("color_curves"):
+                    clip_obj.set_color_curves(True)
+
+            # ── B. 音频 ─────────────────────────────────────────────────
+            elif clip_type == "audio":
+                path = clip_data.get("path")
+                if not path:
+                    continue
+                source_start = _parse_time(clip_data.get("source_start", 0))
+                clip_obj = AudioClip(path, start=start, duration=duration, source_start=source_start)
+
+                if clip_data.get("volume") is not None:
+                    clip_obj.set_volume(clip_data["volume"])
+
+                speed_cfg = clip_data.get("speed")
+                if speed_cfg:
+                    if isinstance(speed_cfg, dict):
+                        clip_obj.set_speed(speed_cfg.get("value", 1.0),
+                                           change_pitch=speed_cfg.get("change_pitch", False))
+                    else:
+                        clip_obj.set_speed(float(speed_cfg))
+
+                fade = clip_data.get("fade")
+                if fade:
+                    clip_obj.set_fade(
+                        fade_in=_parse_time(fade.get("in", 0)),
+                        fade_out=_parse_time(fade.get("out", 0)),
+                    )
+
+                if clip_data.get("denoise"):
+                    clip_obj.set_denoise(True)
+                if clip_data.get("loudness"):
+                    clip_obj.set_loudness(clip_data["loudness"])
+                if clip_data.get("vocal_separation") is not None:
+                    clip_obj.set_vocal_separation(clip_data["vocal_separation"])
+                if clip_data.get("vocal_beautify") is not None:
+                    clip_obj.set_vocal_beautify(clip_data["vocal_beautify"])
+
+                for ve in clip_data.get("voice_effects", []):
+                    clip_obj.add_voice_effect(
+                        resource_id=ve.get("id"),
+                        type=ve.get("type", "tone"),
+                        adjust_params=ve.get("params"),
+                    )
+                tone_effect = clip_data.get("tone_effect")
+                if tone_effect and not clip_data.get("voice_effects"):
+                    clip_obj.add_voice_effect(tone_effect, type="tone")
+
+            # ── C. 文字 ─────────────────────────────────────────────────
+            elif clip_type == "text":
+                content = clip_data.get("content") or clip_data.get("text", "Text")
+                clip_obj = TextClip(content, start=start, duration=duration)
+
+                style = clip_data.get("style", {})
+                clip_obj.set_style(
+                    font_id=_resolve_font_id(style),
+                    size=style.get("size"),
+                    color=style.get("color"),
+                    alpha=style.get("alpha"),
+                    bold=style.get("bold"),
+                    italic=style.get("italic"),
+                    underline=style.get("underline"),
+                )
+                clip_obj.set_format(
+                    align=style.get("align", 1),
+                    line_spacing=style.get("line_spacing", 0),
+                    letter_spacing=style.get("letter_spacing", 0),
+                    line_max_width=style.get("line_max_width", 82),
+                    typesetting=style.get("typesetting", 0),
+                )
+
+                border = style.get("border")
+                if border:
+                    clip_obj.set_border(width=border.get("width", 0), color=border.get("color", "#000000"))
+
+                shadow = style.get("shadow")
+                if shadow:
+                    clip_obj.set_shadow(
+                        alpha=shadow.get("alpha", 0),
+                        color=shadow.get("color", "#000000"),
+                        distance=shadow.get("distance", 5.0),
+                        angle=shadow.get("angle", -45.0),
+                        diffuse=shadow.get("diffuse", 15),
+                    )
+
+                bg = style.get("background")
+                if bg:
+                    clip_obj.set_background(
+                        color=bg.get("color", "#000000"),
+                        alpha=bg.get("alpha", 50),
+                        radius=bg.get("radius", 50),
+                        width=bg.get("width", 0),
+                        height=bg.get("height", 0),
+                        off_x=bg.get("off_x", 50),
+                        off_y=bg.get("off_y", 50),
+                        style=bg.get("style", 1),
+                    )
+
+                if style.get("curve") is not None:
+                    clip_obj.set_curve(angle=style["curve"])
+
+                glow_id   = clip_data.get("glow_id") or style.get("glow_id")
+                bubble_id = clip_data.get("bubble_id") or style.get("bubble_id")
+                if glow_id or bubble_id:
+                    clip_obj.set_effect(glow_id=glow_id, bubble_id=bubble_id)
+
+                bloom = clip_data.get("bloom") or style.get("bloom")
+                if bloom:
+                    clip_obj.set_bloom(
+                        strength=bloom.get("strength", 50),
+                        range_val=bloom.get("range", 50),
+                        dir_x=bloom.get("dir_x", 0),
+                        dir_y=bloom.get("dir_y", 0),
+                        color=bloom.get("color", ""),
+                        bloom_type=bloom.get("type", "外发光"),
+                    )
+
+                rich_styles = clip_data.get("rich_styles")
+                if rich_styles:
+                    for tag, props in rich_styles.items():
+                        clip_obj.add_rich_style(tag, **props)
+
+            # ── D. 贴纸 ─────────────────────────────────────────────────
+            elif clip_type == "sticker":
+                res_id = clip_data.get("resource_id") or clip_data.get("path")
+                if not res_id:
+                    continue
+                clip_obj = StickerClip(res_id, start=start, duration=duration)
+
+            # ── E. 特效 ─────────────────────────────────────────────────
+            elif clip_type == "effect":
+                res_id = clip_data.get("resource_id") or clip_data.get("effect_title")
+                if not res_id:
+                    continue
+                clip_obj = EffectClip(
+                    resource_id=res_id,
+                    start=start,
+                    duration=duration,
+                    name=clip_data.get("name", "effect"),
+                    type=clip_data.get("effect_type", "video_effect"),
+                    render_index=clip_data.get("render_index"),
+                )
+                params = clip_data.get("params") or clip_data.get("adjust_params", [])
+                if isinstance(params, dict):
+                    for k, v in params.items():
+                        clip_obj.set_param(k, v)
+                elif isinstance(params, list):
+                    for p in params:
+                        clip_obj.set_param(p.get("name"), p.get("value"))
+
+            # ── F. 滤镜 ─────────────────────────────────────────────────
+            elif clip_type == "filter":
+                res_id = clip_data.get("resource_id")
+                if not res_id:
+                    continue
+                clip_obj = FilterClip(
+                    resource_id=res_id,
+                    start=start,
+                    duration=duration,
+                    intensity=clip_data.get("intensity", 80),
+                    name=clip_data.get("name", "filter"),
+                )
+
+            if not clip_obj:
+                continue
+
+            # ── 通用：transform / 动画 / 关键帧 ────────────────────────
+            transform = clip_data.get("transform")
+            if transform:
+                clip_obj.set_transform(
+                    x=transform.get("x", 0),
+                    y=transform.get("y", 0),
+                    scale=transform.get("scale", 100),
+                    rotation=transform.get("rotation", 0),
+                    alpha=transform.get("alpha", 100),
+                    uniform_scale=transform.get("uniform_scale", True),
+                    scale_x=transform.get("scale_x", 100),
+                    scale_y=transform.get("scale_y", 100),
+                )
+
+            for anim in clip_data.get("animations", []):
+                clip_obj.add_animation(
+                    anim_type=anim.get("type"),
+                    resource_id=anim.get("id"),
+                    duration=_parse_time(anim.get("duration", 500_000)),
+                )
+
+            for kf_group in clip_data.get("keyframes", []):
+                prop   = kf_group.get("prop")
+                for kf_point in kf_group.get("data", []):
+                    clip_obj.add_keyframe(
+                        prop,
+                        _parse_time(kf_point.get("time_offset")),
+                        kf_point.get("value"),
+                    )
+
+            current_track.add(clip_obj)
+            if clip_id:
+                clip_map[clip_id] = clip_obj
+
+    # 4. 生成完整剪映草稿 JSON（不写磁盘，直接构建 dict）
+    from zcatwx_zil6sm6h_wyaaz._core import Track
+
+    track_data = [t.export(draft) for t in draft.tracks]
+    existing_types = {t.type for t in draft.tracks}
+    for default_type in ("video", "audio"):
+        if default_type not in existing_types:
+            dummy = Track(default_type)
+            track_data.insert(0, dummy.export(draft))
+
+    duration = 0
+    for t in draft.tracks:
+        if t.segments:
+            last = t.segments[-1]
+            end = last.start_time + last.duration
+            if end > duration:
+                duration = end
+
+    draft_content = {
+        "canvas_config": {"width": draft.width, "height": draft.height, "ratio": draft.ratio},
+        "color_space": 0,
+        "config": {
+            "adjust_max_index": 1, "attachment_info": [], "combination_max_index": 1,
+            "export_range": None, "extract_audio_last_index": 1,
+            "lyrics_recognition_id": "", "lyrics_sync": True, "lyrics_taskinfo": [],
+            "maintrack_adsorb": True, "material_save_mode": 0,
+            "multi_language_current": "none", "multi_language_list": [],
+            "multi_language_main": "none", "multi_language_mode": "none",
+            "original_sound_last_index": 1, "record_audio_last_index": 1,
+            "sticker_max_index": 1, "subtitle_keywords_config": None,
+            "subtitle_recognition_id": "", "subtitle_sync": True, "subtitle_taskinfo": [],
+            "system_font_list": [], "video_mute": False, "zoom_info_params": None,
+        },
+        "cover": None, "create_time": 0, "duration": duration,
+        "extra_info": None, "fps": 30.0, "free_render_index_mode_on": False,
+        "group_container": None, "id": draft.project_id,
+        "keyframe_graph_list": [],
+        "keyframes": {
+            "adjusts": [], "audios": [], "effects": [], "filters": [],
+            "handwrites": [], "stickers": [], "texts": [], "videos": [],
+        },
+        "materials": draft.materials,
+        "mutable_config": None, "name": save_name, "new_version": "110.0.0",
+        "platform": {
+            "app_id": 3704, "app_source": "lv", "app_version": "5.9.0",
+            "device_id": "", "hard_disk_id": "", "mac_address": "",
+            "os": "mac", "os_version": "14.0",
+        },
+        "last_modified_platform": {
+            "app_id": 3704, "app_source": "lv", "app_version": "5.9.0",
+            "device_id": "", "hard_disk_id": "", "mac_address": "",
+            "os": "mac", "os_version": "14.0",
+        },
+        "relationships": [], "render_index_track_mode_on": True,
+        "retouch_cover": None, "source": "default",
+        "static_cover_image_path": "", "tracks": track_data,
+        "update_time": 0, "version": 360000,
+    }
+
+    draft_json_str = json.dumps(draft_content, ensure_ascii=False)
+
+    # 5. 加密后返回
+    return DraftCrypto.encrypt(draft_json_str)