From c884a2523746892d15257b68f6f3681119681cf3 Mon Sep 17 00:00:00 2001 From: Anton Hvornum Date: Tue, 21 Jul 2020 19:03:24 +0000 Subject: Added a set_timezone() and fixed set_locale() in the Installer() class. Also added a mirrors.py helper to rudimentary set mirror data on the installer host --- archinstall/lib/mirrors.py | 45 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 archinstall/lib/mirrors.py (limited to 'archinstall/lib/mirrors.py') diff --git a/archinstall/lib/mirrors.py b/archinstall/lib/mirrors.py new file mode 100644 index 00000000..63eb4678 --- /dev/null +++ b/archinstall/lib/mirrors.py @@ -0,0 +1,45 @@ +from .exceptions import * +from .general import * + +def filter_mirrors_by_region(regions, *args, **kwargs): + """ + This function will change the active mirrors on the live medium by + filtering which regions are active based on `regions`. + + :param region: A series of country codes separated by `,`. For instance `SE,US` for sweden and United States. + :type region: str + """ + region_list = [] + for region in regions.split(','): + region_list.append(f'country={region}') + o = b''.join(sys_command((f"/usr/bin/wget 'https://www.archlinux.org/mirrorlist/?{'&'.join(region_list)}&protocol=https&ip_version=4&ip_version=6&use_mirror_status=on' -O /root/mirrorlist"))) + o = b''.join(sys_command(("/usr/bin/sed -i 's/#Server/Server/' /root/mirrorlist"))) + o = b''.join(sys_command(("/usr/bin/mv /root/mirrorlist /etc/pacman.d/"))) + + return True + +def insert_mirrors(mirrors, *args, **kwargs): + """ + This function will insert a given mirror-list at the top of `/etc/pacman.d/mirrorlist`. + It will not flush any other mirrors, just insert new ones. + + :param mirrors: A dictionary of `{'url' : 'country', 'url2' : 'country'}` + :type mirrors: dict + """ + original_mirrorlist = '' + with open('/etc/pacman.d/mirrorlist', 'r') as original: + original_mirrorlist = original.read() + + with open('/etc/pacman.d/mirrorlist', 'w') as new_mirrorlist: + for mirror, country in mirrors.items(): + new_mirrorlist.write(f'## {country}\n') + new_mirrorlist.write(f'Server = {mirror}\n') + new_mirrorlist.write('\n') + new_mirrorlist.write(original_mirrorlist) + + return True + +def re_rank_mirrors(top=10, *positionals, **kwargs): + if sys_command((f'/usr/bin/rankmirrors -n {top} /etc/pacman.d/mirrorlist > /etc/pacman.d/mirrorlist')).exit_code == 0: + return True + return False \ No newline at end of file -- cgit v1.2.3-54-g00ecf