main.py

import configparser
import argparse

from entity.iconfig_parser import ConfigParser
from video_builder import VideoBuilder
from entity.video_const import VideoConst
from playwright.sync_api import sync_playwright
import subprocess
import os
from logger_config import logger



def check_chromium_installed():
    with sync_playwright() as p:
        # 获取 Chromium 的可执行文件路径
        chromium_path = p.chromium.executable_path

        if os.path.exists(chromium_path):
            return True
        else:
            return False


def install_chromium():
    subprocess.run(["playwright", "install", "chromium"])


def main():

    if not check_chromium_installed():
        logger.info("初次使用,环境准备中")
        install_chromium()
        logger.info("准备完成")

    config = configparser.ConfigParser()
    config.read('./config.ini')

    parser = argparse.ArgumentParser(description='Process some integers.')
    parser.add_argument('-s', '--service', type=str, help='service["RUN_WAY_TXT","RUN_WAY_IMG","PIKA"]')
    parser.add_argument('-u', '--username', type=str, help='username')
    parser.add_argument('-p', '--password', type=str, help='password')
    parser.add_argument('-c', '--content', type=str, help='content:Prompt word or image local path')
    parser.add_argument('-i', '--image', type=str, help='content:Prompt word and image local path')

    args = parser.parse_args()
    username = args.username
    password = args.password
    content = args.content
    service = args.service
    img = args.image

    if service is None:
        service = VideoConst.RUN_WAY_MIX

    if service == VideoConst.PIKA_TXT \
            or service == VideoConst.PIKA_IMG \
            or service == VideoConst.PIKA_MIX:
        if username is None:
            username = config['PIKA']['username']
        if password is None:
            password = config['PIKA']['password']

    else:
        if username is None:
            username = config['RUNWAY']['username']
        if password is None:
            password = config['RUNWAY']['password']

    if content is None:
        content = config['CONTENT']['content']
    if img is None:
        img = config['CONTENT']['image']

    i_config = ConfigParser(username, password)
    processor = VideoBuilder.create().set_config(i_config).set_form(content, img) \
        .set_processor(service).build()
    download_link = processor.run()


if __name__ == "__main__":
    main()