mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-04-06 03:57:44 +08:00
68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
import configparser
|
|
import re
|
|
|
|
|
|
class ConfigManager:
|
|
def __init__(self):
|
|
self.config_path = 'Ref_Audio_Selector/config.ini'
|
|
self.comments = []
|
|
self.config = None
|
|
self.read_with_comments()
|
|
|
|
def read_with_comments(self):
|
|
with open(self.config_path, 'r', encoding='utf-8') as f:
|
|
lines = f.readlines()
|
|
|
|
self.comments = []
|
|
for i, line in enumerate(lines):
|
|
if line.startswith(';') or line.startswith('#'):
|
|
self.comments.append((i, line))
|
|
|
|
self.config = configparser.ConfigParser()
|
|
self.config.read_string(''.join(lines))
|
|
|
|
def write_with_comments(self):
|
|
output_lines = []
|
|
|
|
# 先写入配置项
|
|
config_str = self.config.write()
|
|
output_lines.extend(config_str.splitlines(True)) # 保持换行
|
|
|
|
# 然后插入原有注释
|
|
for index, comment in sorted(self.comments, reverse=True): # 从后往前插入，避免行号错乱
|
|
while len(output_lines) < index + 1:
|
|
output_lines.append('\n') # 补充空行
|
|
output_lines.insert(index, comment)
|
|
|
|
with open(self.config_path, 'w', encoding='utf-8') as f:
|
|
f.writelines(output_lines)
|
|
|
|
def get_base(self, key):
|
|
return self.config.get('Base', key)
|
|
|
|
def set_base(self, key, value):
|
|
self.config.set('Base', key, value)
|
|
self.write_with_comments()
|
|
|
|
def get_audio_sample(self, key):
|
|
return self.config.get('AudioSample', key)
|
|
|
|
def get_inference(self, key):
|
|
return self.config.get('Inference', key)
|
|
|
|
def get_result_check(self, key):
|
|
return self.config.get('ResultCheck', key)
|
|
|
|
def get_audio_config(self, key):
|
|
return self.config.get('AudioConfig', key)
|
|
|
|
def get_other(self, key):
|
|
return self.config.get('Other', key)
|
|
|
|
|
|
_config = ConfigManager()
|
|
|
|
|
|
def get_config():
|
|
return _config
|