chore: 添加本地草稿服务器脚本及ugc_product_seeding账号
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
118
.claude/skills/video-from-script/scripts/local-draft-server.py
Normal file
118
.claude/skills/video-from-script/scripts/local-draft-server.py
Normal file
@@ -0,0 +1,118 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
本地草稿服务器 — 模拟 CapCut Mate API,从本地草稿文件夹提供下载。
|
||||
|
||||
用法:
|
||||
python3 local-draft-server.py [--port 8765]
|
||||
|
||||
下载器将 API base 指向 http://localhost:8765 即可下载本地草稿。
|
||||
"""
|
||||
|
||||
import os, json, sys, argparse
|
||||
from http.server import HTTPServer, BaseHTTPRequestHandler
|
||||
from urllib.parse import urlparse, parse_qs
|
||||
|
||||
DRAFT_DIR = os.path.expanduser('~/Movies/JianyingPro/User Data/Projects/com.lveditor.draft')
|
||||
|
||||
class DraftServer(BaseHTTPRequestHandler):
|
||||
def log_message(self, format, *args):
|
||||
print(f' {args[0]}')
|
||||
|
||||
def _send_json(self, data, code=200):
|
||||
body = json.dumps(data, ensure_ascii=False).encode('utf-8')
|
||||
self.send_response(code)
|
||||
self.send_header('Content-Type', 'application/json; charset=utf-8')
|
||||
self.send_header('Content-Length', len(body))
|
||||
self.send_header('Access-Control-Allow-Origin', '*')
|
||||
self.end_headers()
|
||||
self.wfile.write(body)
|
||||
|
||||
def _send_file(self, filepath):
|
||||
if not os.path.exists(filepath):
|
||||
self._send_json({'code': 404, 'message': 'File not found'}, 404)
|
||||
return
|
||||
size = os.path.getsize(filepath)
|
||||
with open(filepath, 'rb') as f:
|
||||
self.send_response(200)
|
||||
self.send_header('Content-Type', 'application/octet-stream')
|
||||
self.send_header('Content-Length', size)
|
||||
self.end_headers()
|
||||
self.wfile.write(f.read())
|
||||
|
||||
def do_GET(self):
|
||||
parsed = urlparse(self.path)
|
||||
params = parse_qs(parsed.query)
|
||||
|
||||
# /openapi/capcut-mate/v1/get_draft 或 /get_draft
|
||||
if parsed.path.endswith('/get_draft') or '/get_draft' in parsed.path:
|
||||
draft_id = params.get('draft_id', [None])[0]
|
||||
if not draft_id:
|
||||
self._send_json({'code': 1001, 'message': 'Missing draft_id'})
|
||||
return
|
||||
|
||||
draft_path = os.path.join(DRAFT_DIR, draft_id)
|
||||
if not os.path.isdir(draft_path):
|
||||
self._send_json({'code': 2001, 'message': f'Draft not found: {draft_id}'})
|
||||
return
|
||||
|
||||
# 收集所有文件 URL
|
||||
files = []
|
||||
host = self.headers.get('Host', 'localhost:8765')
|
||||
base_url = f'http://{host}/files/{draft_id}'
|
||||
|
||||
for root, dirs, filenames in os.walk(draft_path):
|
||||
for fname in filenames:
|
||||
rel = os.path.relpath(os.path.join(root, fname), draft_path)
|
||||
files.append(f'{base_url}/{rel}')
|
||||
|
||||
print(f'\n[Draft] {draft_id} -> {len(files)} files')
|
||||
self._send_json({'code': 0, 'message': '成功', 'files': files})
|
||||
return
|
||||
|
||||
# /files/NAME/... -> 提供文件下载
|
||||
if parsed.path.startswith('/files/'):
|
||||
parts = parsed.path[7:].split('/', 1) # Remove /files/
|
||||
if len(parts) >= 2:
|
||||
draft_id = parts[0]
|
||||
rel_path = parts[1]
|
||||
filepath = os.path.join(DRAFT_DIR, draft_id, rel_path)
|
||||
self._send_file(filepath)
|
||||
return
|
||||
|
||||
# 列出所有可用草稿
|
||||
if parsed.path == '/' or parsed.path == '/list':
|
||||
drafts = [d for d in sorted(os.listdir(DRAFT_DIR))
|
||||
if os.path.isdir(os.path.join(DRAFT_DIR, d))
|
||||
and not d.endswith('.tmp')
|
||||
and d.startswith('执黑')]
|
||||
self._send_json({'code': 0, 'drafts': drafts})
|
||||
return
|
||||
|
||||
self._send_json({'code': 404, 'message': 'Not found'}, 404)
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description='Local CapCut Draft Server')
|
||||
parser.add_argument('--port', type=int, default=8765, help='Server port (default: 8765)')
|
||||
args = parser.parse_args()
|
||||
|
||||
print(f'\n本地草稿服务器')
|
||||
print(f'草稿目录: {DRAFT_DIR}')
|
||||
print(f'服务地址: http://localhost:{args.port}')
|
||||
print(f'\n可用草稿:')
|
||||
for d in sorted(os.listdir(DRAFT_DIR)):
|
||||
if os.path.isdir(os.path.join(DRAFT_DIR, d)) and d.startswith('执黑') and '_v2' not in d:
|
||||
print(f' http://localhost:{args.port}/get_draft?draft_id={d}')
|
||||
print(f'\n下载器设置 API Base: http://localhost:{args.port}')
|
||||
print(f'按 Ctrl+C 停止服务器\n')
|
||||
|
||||
server = HTTPServer(('0.0.0.0', args.port), DraftServer)
|
||||
try:
|
||||
server.serve_forever()
|
||||
except KeyboardInterrupt:
|
||||
print('\n服务器已停止')
|
||||
server.shutdown()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user