diff --git a/app/main.py b/app/main.py index 6e41b3b..2f1c40b 100644 --- a/app/main.py +++ b/app/main.py @@ -17,8 +17,14 @@ from geolite2 import geolite2 from download_image_map import Parser from mirror_list import Mirror +import os +mirror_path="mirrors.yaml" +if "ARMBIAN_MIRROR_CONF" in os.environ: + mirror_path=os.environ["ARMBIAN_MIRROR_CONF"] -mirror = Mirror() +#print("Mirrors conf file:",mirror_path) + +mirror = Mirror(mirror_path) if mirror.mode == "dl_map": parser = Parser('userdata.csv') DL_MAP = parser.parsed_data 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))