from os import path
import toml
# "#969696"
def hex_to_rgb_percent(str_hex):
return int(str_hex[1:3], 16)/255, \
int(str_hex[3:5], 16)/255, \
int(str_hex[5:7], 16)/255
# [255 255 255]
def dec_to_rgb_percent(list_rgb):
return list_rgb[0]/255, list_rgb[1]/255, list_rgb[2]/255
class ConfLoader:
def __init__(self):
path_conf = f'{path.dirname(path.realpath(__file__))}/conf'
self.file_color = f'{path_conf}/color.toml'
self.file_user = f'{path_conf}/user.toml'
self.file_bili = f'{path_conf}/bili.toml'
self.file_ctrl = f'{path_conf}/ctrl.toml'
self.file_task = f'{path_conf}/task.toml'
'''
self.dict_color = self.read_color()
print(self.dict_color)
self.dict_user = self.read_user()
print(self.dict_user)
self.dict_bili = self.read_bili()
print(self.dict_bili)
print("# 初始化完成")
'''
@staticmethod
def toml_load(path):
with open(path, encoding="utf-8") as f:
return toml.load(f)
@staticmethod
def toml_dump(object, path):
with open(path, 'w', encoding="utf-8") as f:
toml.dump(object, f)
def write_user(self, dict_new, user_id):
dict_user = self.toml_load(self.file_user)
for i, value in dict_new.items():
dict_user['users'][user_id][i] = value
self.toml_dump(dict_user, self.file_user)
def read_bili(self):
return self.toml_load(self.file_bili)
def read_color(self):
dict_color = self.toml_load(self.file_color)
for i in dict_color.values():
for key, color in i.items():
if isinstance(color, str):
i[key] = hex_to_rgb_percent(color)
elif isinstance(color, list):
i[key] = dec_to_rgb_percent(color)
return dict_color
def read_user(self):
return self.toml_load(self.file_user)
def read_ctrl(self):
return self.toml_load(self.file_ctrl)
def read_task(self):
return self.toml_load(self.file_task)
var = ConfLoader()
def write_user(dict_new, user_id):
var.write_user(dict_new, user_id)
def read_bili():
return var.read_bili()
def read_color():
return var.read_color()
def read_user():
return var.read_user()
def read_ctrl():
return var.read_ctrl()
def read_task():
return var.read_task()
