Fix yaml parsing

This commit is contained in:
aerinon
2022-05-19 12:20:30 -06:00
parent 041443be3d
commit e40c4290c9
2 changed files with 10 additions and 16 deletions

View File

@@ -334,12 +334,9 @@ class CustomSettings(object):
def load_yaml(path): def load_yaml(path):
try: if os.path.exists(Path(path)):
return yaml.load(path, Loader=yaml.SafeLoader) with open(path, "r", encoding="utf-8") as f:
except yaml.YAMLError as exc: return yaml.load(f, Loader=yaml.SafeLoader)
if os.path.exists(Path(path)): elif urllib.parse.urlparse(path).scheme in ['http', 'https']:
with open(path, "r", encoding="utf-8") as f: return yaml.load(urllib.request.urlopen(path), Loader=yaml.FullLoader)
return yaml.load(f, Loader=yaml.SafeLoader)
elif urllib.parse.urlparse(path).scheme in ['http', 'https']:
return yaml.load(urllib.request.urlopen(path), Loader=yaml.FullLoader)

View File

@@ -9,14 +9,11 @@ import yaml
def get_weights(path): def get_weights(path):
try: if os.path.exists(Path(path)):
return yaml.load(path, Loader=yaml.SafeLoader) with open(path, "r", encoding="utf-8") as f:
except yaml.YAMLError as exc: return yaml.load(f, Loader=yaml.SafeLoader)
if os.path.exists(Path(path)): elif urllib.parse.urlparse(path).scheme in ['http', 'https']:
with open(path, "r", encoding="utf-8") as f: return yaml.load(urllib.request.urlopen(path), Loader=yaml.FullLoader)
return yaml.load(f, Loader=yaml.SafeLoader)
elif urllib.parse.urlparse(path).scheme in ['http', 'https']:
return yaml.load(urllib.request.urlopen(path), Loader=yaml.FullLoader)
def roll_settings(weights): def roll_settings(weights):