1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

import re
import argparse
from pathlib import Path

def find_max_used_index(text, folder, ext):
    no_prefix = folder.strip() in (".", "./")
    if no_prefix:
        pat = re.compile(r"!\[[^\]]*\]\((?:\./)?([0-9]+)\." + re.escape(ext) + r"\)")
    else:
        pat = re.compile(r"!\[[^\]]*\]\(" + re.escape(folder.rstrip('/')) + r"/([0-9]+)\." + re.escape(ext) + r"\)")
    max_idx = -1
    for m in pat.finditer(text):
        i = int(m.group(1))
        if i > max_idx:
            max_idx = i
    return max_idx

def replace_feishu_images(text, folder, start, end, alt, ext):
    remote_img_pat = re.compile(r"!\[[^\]]*\]\((https://[^)]*)\)")
    idx = start
    limit = end if end is not None else None
    no_prefix = folder.strip() in (".", "./")

    def repl(_):
        nonlocal idx
        if limit is not None and idx > limit:
            return _.group(0)
        if no_prefix:
            s = f"![{alt}]({idx}.{ext})"
        else:
            s = f"![{alt}]({folder.rstrip('/')}/{idx}.{ext})"
        idx += 1
        return s

    return remote_img_pat.sub(repl, text)

def main():
    p = argparse.ArgumentParser()
    p.add_argument('--md', required=True)
    p.add_argument('--folder', required=True)
    p.add_argument('--start', type=int)
    p.add_argument('--end', type=int)
    p.add_argument('--alt', default='img')
    p.add_argument('--ext', default='png')
    p.add_argument('--dry', action='store_true')
    args = p.parse_args()

    md_path = Path(args.md)
    text = md_path.read_text(encoding='utf-8')

    start = args.start
    if start is None:
        max_used = find_max_used_index(text, args.folder, args.ext)
        start = max_used + 1 if max_used >= 0 else 0

    new_text = replace_feishu_images(text, args.folder, start, args.end, args.alt, args.ext)

    if args.dry:
        print(new_text)
    else:
        md_path.write_text(new_text, encoding='utf-8')

if __name__ == '__main__':
    main()
    
    
# python replace_feishu_images.py --md index.md --folder . --start 1 --end 14 --ext png
    
# python replace_feishu_images.py --md 2025-RCTF.md --folder ../images/2025-RCTF --start 0 --end 76 --ext png