工作流，更新双GPU加速

2022-05-03 11:07:45 +08:00
parent a43b47c0b8
commit 12504b9cf5
3 changed files with 427 additions and 98 deletions
--- a/config.py
+++ b/config.py
@ -0,0 +1,107 @@
+import configparser
+import os.path
+
+
+# [danmaku]
+# exec
+DANMAKU_FACTORY_EXEC = "DanmakuFactory"
+# speed
+DANMAKU_SPEED = 12
+# font
+DEFAULT_FONT_NAME = "Sarasa Term SC"
+# resolution
+VIDEO_RESOLUTION = "1280x720"
+# [ffmpeg]
+# exec
+FFMPEG_EXEC = "ffmpeg"
+# nvidia_gpu
+FFMPEG_USE_NVIDIA_GPU = True
+# intel_gpu
+FFMPEG_USE_INTEL_GPU = True
+# bitrate
+VIDEO_BITRATE = "2.5M"
+# [video]
+# title
+VIDEO_TITLE = "【永恒de草薙直播录播】直播于 {}"
+# [clip]
+# each_sec
+VIDEO_CLIP_EACH_SEC = 6000
+# overflow_sec
+VIDEO_CLIP_OVERFLOW_SEC = 5
+# [recorder]
+# bili_dir
+BILILIVE_RECORDER_DIRECTORY = "./"
+# xigua_dir
+XIGUALIVE_RECORDER_DIRECTORY = "./"
+
+
+def load_config():
+    if not os.path.exists("config.ini"):
+        write_config()
+        return False
+    config = configparser.ConfigParser()
+    config.read("config.ini", encoding="utf-8")
+    if config.has_section("danmaku"):
+        section = config['danmaku']
+        global DANMAKU_FACTORY_EXEC, DANMAKU_SPEED, DEFAULT_FONT_NAME, VIDEO_RESOLUTION
+        DANMAKU_FACTORY_EXEC = section.get('exec', DANMAKU_FACTORY_EXEC)
+        DANMAKU_SPEED = section.getfloat('speed', DANMAKU_SPEED)
+        DEFAULT_FONT_NAME = section.get('font', DEFAULT_FONT_NAME)
+        VIDEO_RESOLUTION = section.get('resolution', VIDEO_RESOLUTION)
+    if config.has_section("video"):
+        section = config['video']
+        global VIDEO_TITLE
+        VIDEO_TITLE = section.get('title', VIDEO_TITLE)
+    if config.has_section("clip"):
+        section = config['clip']
+        global VIDEO_CLIP_EACH_SEC, VIDEO_CLIP_OVERFLOW_SEC
+        VIDEO_CLIP_EACH_SEC = section.getfloat('each_sec', VIDEO_CLIP_EACH_SEC)
+        VIDEO_CLIP_OVERFLOW_SEC = section.getfloat('overflow_sec', VIDEO_CLIP_OVERFLOW_SEC)
+    if config.has_section("ffmpeg"):
+        section = config['ffmpeg']
+        global FFMPEG_EXEC, FFMPEG_USE_NVIDIA_GPU, FFMPEG_USE_INTEL_GPU, VIDEO_BITRATE
+        FFMPEG_EXEC = section.get('exec', FFMPEG_EXEC)
+        FFMPEG_USE_NVIDIA_GPU = section.getboolean('nvidia_gpu', FFMPEG_USE_NVIDIA_GPU)
+        FFMPEG_USE_INTEL_GPU = section.getboolean('intel_gpu', FFMPEG_USE_INTEL_GPU)
+        VIDEO_BITRATE = section.get('bitrate', VIDEO_BITRATE)
+    if config.has_section("recorder"):
+        global BILILIVE_RECORDER_DIRECTORY, XIGUALIVE_RECORDER_DIRECTORY
+        section = config['recorder']
+        BILILIVE_RECORDER_DIRECTORY = section.get('bili_dir', BILILIVE_RECORDER_DIRECTORY)
+        XIGUALIVE_RECORDER_DIRECTORY = section.get('xigua_dir', XIGUALIVE_RECORDER_DIRECTORY)
+    return True
+
+
+def get_config():
+    config = {
+        'danmaku': {
+            'exec': DANMAKU_FACTORY_EXEC,
+            'speed': DANMAKU_SPEED,
+            'font': DEFAULT_FONT_NAME,
+            'resolution': VIDEO_RESOLUTION,
+        },
+        'clip': {
+            'each_sec': VIDEO_CLIP_EACH_SEC,
+            'overflow_sec': VIDEO_CLIP_OVERFLOW_SEC,
+        },
+        'ffmpeg': {
+            'exec': FFMPEG_EXEC,
+            'nvidia_gpu': FFMPEG_USE_NVIDIA_GPU,
+            'intel_gpu': FFMPEG_USE_INTEL_GPU,
+            'bitrate': VIDEO_BITRATE,
+        },
+    }
+    return config
+
+
+def write_config():
+    config = configparser.ConfigParser()
+    _config = get_config()
+    for _i in _config:
+        config[_i] = _config[_i]
+    with open("config.ini", "w", encoding="utf-8") as f:
+        config.write(f)
+    return True
+
+
+load_config()