@@ -4,10 +4,12 @@ def __init__(self, alias_file=None):
44 import json
55
66 if alias_file is None :
7- import importlib . resources
7+ import urllib . request , json
88
9- with importlib .resources .open_text ("pango_designation" , "alias_key.json" ) as file :
10- file = json .load (file )
9+ with urllib .request .urlopen (
10+ "https://raw.githubusercontent.com/cov-lineages/pango-designation/master/pango_designation/alias_key.json"
11+ ) as data :
12+ file = json .load (data )
1113
1214 else :
1315 with open (alias_file ) as file :
@@ -22,18 +24,18 @@ def __init__(self, alias_file=None):
2224
2325 self .realias_dict = {v : k for k , v in self .alias_dict .items ()}
2426
25- def compress (self ,name ):
26- name_split = name .split ('.' )
27+ def compress (self , name ):
28+ name_split = name .split ("." )
2729 levels = len (name_split ) - 1
28- num_indirections = (levels - 1 ) // 3
30+ num_indirections = (levels - 1 ) // 3
2931 if num_indirections <= 0 :
3032 return name
31- alias = "." .join (name_split [0 :( 3 * num_indirections + 1 )])
32- ending = "." .join (name_split [(3 * num_indirections + 1 ):])
33- return self .realias_dict [alias ] + '.' + ending
33+ alias = "." .join (name_split [0 : ( 3 * num_indirections + 1 )])
34+ ending = "." .join (name_split [(3 * num_indirections + 1 ) :])
35+ return self .realias_dict [alias ] + "." + ending
3436
35- def uncompress (self ,name ):
36- name_split = name .split ('.' )
37+ def uncompress (self , name ):
38+ name_split = name .split ("." )
3739 letter = name_split [0 ]
3840 try :
3941 unaliased = self .alias_dict [letter ]
@@ -42,7 +44,9 @@ def uncompress(self,name):
4244 if len (name_split ) == 1 :
4345 return name
4446 if len (name_split ) == 2 :
45- return unaliased + '.' + name_split [1 ]
47+ return unaliased + "." + name_split [1 ]
4648 else :
47- return unaliased + '.' + "." .join (name_split [1 :])
49+ return unaliased + "." + "." .join (name_split [1 :])
50+
51+
4852# %%
0 commit comments