From 31d30a82b39172bb80c8b2105d58080af4f239fe Mon Sep 17 00:00:00 2001 From: jeanrhum Date: Fri, 12 Mar 2021 18:48:51 +0100 Subject: [PATCH] Update mirror_list.py Add a mirror path paramater to class Mirror --- app/mirror_list.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/app/mirror_list.py b/app/mirror_list.py index aed081d..fa16d2c 100644 --- a/app/mirror_list.py +++ b/app/mirror_list.py @@ -4,7 +4,8 @@ from ruamel.yaml import YAML class Mirror(): - def __init__(self): + def __init__(self,mirror_path='mirrors.yaml'): + self.mirror_path=mirror_path self.load_mirrors() self._list_position = dict() self._list_max = dict() @@ -22,7 +23,7 @@ class Mirror(): yaml.indent(mapping=2, sequence=4, offset=2) yaml.preserve_quotes = True - with open('mirrors.yaml', 'r') as f: + with open(self.mirror_path, 'r') as f: config = yaml.load(f) self.mode = config['mode'] print("using mode: {}".format(self.mode))