Bilibili_iframe_convert

import re
import sys

def convert_iframe_code(input_code):
   # print(f"Received iframe code: {input_code}")  # 打印接收到的 iframe 代码，用于调试
    
    # 更新正则表达式，允许匹配多种 iframe 格式
    pattern = r'<iframe.*?src=["\']\/\/player\.bilibili\.com\/player\.html\?isOutside=true&aid=(\d+)&bvid=(\w+)&cid=(\d+)&p=(\d+).*?["\'].*?>.*?<\/iframe>'
    match = re.search(pattern, input_code)

    if match:
        aid = match.group(1)
        bvid = match.group(2)
        cid = match.group(3)
        p = match.group(4)

        # 构造新的 iframe 代码
        new_code = f'<iframe src="http://player.bilibili.com/player.html?isOutside=true&aid={aid}&bvid={bvid}&cid={cid}&p={p}&autoplay=false" ' \
                   f'scrolling="no" border="0" frameborder="no" framespacing="0" allowfullscreen="true" ' \
                   f'style="width: 640px; height: 430px; max-width: 100%"></iframe>'
        return new_code
    else:
        raise ValueError("Invalid iframe code")

# 从命令行参数中获取输入
if __name__ == '__main__':
    if len(sys.argv) > 1:
        input_code = sys.argv[1]
        try:
            output_code = convert_iframe_code(input_code)
            print(output_code)
        except ValueError as e:
            print(e)
    else:
        print("Please provide the iframe code as an argument.")