51 lines
2.5 KiB
Python
51 lines
2.5 KiB
Python
"""保存真人资产的非破坏性修订，保留首版图片、提示词及回执。"""
|
|
import argparse
|
|
import json
|
|
from pathlib import Path
|
|
import re
|
|
import shutil
|
|
import struct
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
LIVE = ROOT / '08_真人AI短剧'
|
|
p = argparse.ArgumentParser()
|
|
p.add_argument('--id', required=True)
|
|
p.add_argument('--source', required=True)
|
|
p.add_argument('--prompt-file', required=True)
|
|
p.add_argument('--reference', action='append', default=[])
|
|
args = p.parse_args()
|
|
registry = LIVE / '02_资产/全部资产任务.json'
|
|
assets = json.loads(registry.read_text(encoding='utf8'))
|
|
asset = next(a for a in assets if a['id'] == args.id)
|
|
receipt_file = LIVE / '02_资产/生成回执' / (args.id + '.json')
|
|
old = json.loads(receipt_file.read_text(encoding='utf8'))
|
|
version = int(old.get('revision', 1)) + 1
|
|
old_path = LIVE / old['path']
|
|
new_path = old_path.with_name(re.sub(r'-V\d+-REVIEW\.png$', f'-V{version:02d}-REVIEW.png', old_path.name))
|
|
assert new_path != old_path and LIVE.resolve() in new_path.resolve().parents
|
|
assert not new_path.exists(), '修订图已存在，拒绝覆盖'
|
|
prompt = Path(args.prompt_file).read_text(encoding='utf8').rstrip('\n')
|
|
raw = Path(args.source).read_bytes()
|
|
assert raw[:8] == b'\x89PNG\r\n\x1a\n'
|
|
size = list(struct.unpack('>II', raw[16:24]))
|
|
history = receipt_file.parent / '历史'
|
|
history.mkdir(exist_ok=True)
|
|
archive = history / f'{args.id}-R{version-1:02d}.json'
|
|
assert not archive.exists(), '历史回执已存在，拒绝覆盖'
|
|
archive.write_text(json.dumps(old, ensure_ascii=False, indent=2)+'\n', encoding='utf8')
|
|
prompt_dir = LIVE / '04_生成与后期/逐项出图提示词'
|
|
original_prompt = prompt_dir / f'{args.id}-R{version-1:02d}.txt'
|
|
if not original_prompt.exists():
|
|
original_prompt.write_text(old['prompt']+'\n', encoding='utf8')
|
|
shutil.copy2(args.source, new_path)
|
|
receipt = {**old, 'path': new_path.relative_to(LIVE).as_posix(),
|
|
'source': args.source, 'size': size, 'prompt': prompt,
|
|
'revision': version, 'references': args.reference,
|
|
'previous_receipt': archive.relative_to(LIVE).as_posix(),
|
|
'status': 'REVIEW', 'approved': False}
|
|
receipt_file.write_text(json.dumps(receipt, ensure_ascii=False, indent=2)+'\n', encoding='utf8')
|
|
(prompt_dir / (args.id+'.txt')).write_text(prompt+'\n', encoding='utf8')
|
|
asset.update(path=receipt['path'], prompt=prompt, status='REVIEW')
|
|
registry.write_text(json.dumps(assets, ensure_ascii=False, indent=2)+'\n', encoding='utf8')
|
|
print(json.dumps({'id':args.id, 'revision':version, 'path':str(new_path)}, ensure_ascii=False))
|