Skip to content

Commit c3d4724

Browse files
Merge pull request #3 from corneliusroemer/fix-2
fix: download latest alias_key.json using request
2 parents 539d197 + 823fc45 commit c3d4724

File tree

1 file changed

+17
-13
lines changed

1 file changed

+17
-13
lines changed

src/pango_aliasor/aliasor.py

Lines changed: 17 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)