60 lines
1.9 KiB
Python
Executable File
60 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
||
|
||
import datetime
|
||
import re
|
||
##
|
||
import iso3166
|
||
##
|
||
import classes
|
||
|
||
|
||
_strip_re = re.compile(r'^\s*(?P<num>[0-9.]+).*$')
|
||
|
||
|
||
class Ranker(classes.Ranker):
|
||
mirrorlist_url = 'https://www.archlinux.org/mirrors/status/tier/1/'
|
||
|
||
def __init__(self, *args, **kwargs):
|
||
super().__init__(*args, **kwargs)
|
||
self.get_mirrors()
|
||
self.mycountry = iso3166.countries_by_alpha2[self.my_info['country']].name
|
||
|
||
def extract_mirrors(self):
|
||
# Limit to only successful mirrors.
|
||
mirrors = self.bs.find('table', {'id': 'successful_mirrors'})
|
||
# Ayyy, thanks dude.
|
||
# Modified from https://stackoverflow.com/a/56835562/733214.
|
||
header = mirrors.find('thead').find('tr')
|
||
headers = [h.text if h.text != '' else 'details' for h in header.find_all('th')]
|
||
raw_rows = mirrors.find_all('tr')
|
||
# rows = [{headers[i]: cell.text for i, cell in enumerate(row.find_all('td'))} for row in raw_rows]
|
||
rows = [{headers[i]: cell for i, cell in enumerate(row.find_all('td'))} for row in raw_rows]
|
||
for r in rows:
|
||
for k, v in r.items():
|
||
print(v)
|
||
if k in ('Completion %', 'Mirror Score', 'μ Duration (s)', 'σ Duration (s)'):
|
||
r[k] = float(_strip_re.sub(r'\g<num>', v.text).strip())
|
||
elif k == 'μ Delay (hh:mm)':
|
||
# HOO boy. Wish they just did it in seconds.
|
||
# elif k == 'Country':
|
||
self.raw_mirrors.append(r)
|
||
# for row in rows:
|
||
# if not row:
|
||
# continue
|
||
# for k in ('Completion %', 'Mirror Score', 'μ Duration (s)', 'σ Duration (s)'):
|
||
# row[k] = float(_strip_re.sub(r'\g<num>', row[k]).strip())
|
||
|
||
return(None)
|
||
|
||
|
||
def main():
|
||
r = Ranker()
|
||
r.extract_mirrors()
|
||
import pprint
|
||
pprint.pprint(r.raw_mirrors)
|
||
return(None)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|