1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
--- setup/iso_codes.py.orig 2025-04-18 03:14:01 UTC
+++ setup/iso_codes.py
@@ -2,6 +2,7 @@ import fnmatch
# License: GPLv3 Copyright: 2023, Kovid Goyal <kovid at kovidgoyal.net>
import fnmatch
+import glob
import optparse
import os
import shutil
@@ -24,40 +25,60 @@ class ISOData(Command):
description = 'Get ISO codes name localization data'
top_level_filename = 'iso-codes-main'
_zip_data = None
+ extracted = False
def add_options(self, parser):
with suppress(optparse.OptionConflictError): # ignore if option already added
parser.add_option('--path-to-isocodes', help='Path to previously downloaded iso-codes-main.zip')
+ parser.add_option('--isocodes-extracted', default=False, action='store_true',
+ help='Isocodes is a directory')
def run(self, opts):
- if self._zip_data is None and opts.path_to_isocodes:
- with open(opts.path_to_isocodes, 'rb') as f:
- self._zip_data = f.read()
- # get top level directory
- top = {item.split('/')[0] for item in zipfile.ZipFile(self.zip_data).namelist()}
- assert len(top) == 1
- self.top_level_filename = top.pop()
+ if opts.isocodes_extracted:
+ self.top_level = opts.path_to_isocodes
+ self.extracted = True
+ else:
+ if self._zip_data is None and opts.path_to_isocodes:
+ with open(opts.path_to_isocodes, 'rb') as f:
+ self._zip_data = f.read()
+ # get top level directory
+ top = {item.split('/')[0] for item in zipfile.ZipFile(self.zip_data).namelist()}
+ assert len(top) == 1
+ self.top_level_filename = top.pop()
@property
def zip_data(self):
return self._zip_data or iso_codes_data()
def db_data(self, name: str) -> bytes:
- with zipfile.ZipFile(BytesIO(self.zip_data)) as zf:
- with zf.open(f'{self.top_level_filename}/data/{name}') as f:
+ if self.extracted:
+ src = f'{self.top_level}/data/{name}'
+ if not os.path.exists(src):
+ raise Exception(src + ' does not exist')
+ with open(src, 'rb') as f:
return f.read()
+ else:
+ with zipfile.ZipFile(BytesIO(self.zip_data)) as zf:
+ with zf.open(f'{self.top_level_filename}/data/{name}') as f:
+ return f.read()
def extract_po_files(self, name: str, output_dir: str) -> None:
name = name.split('.', 1)[0]
- pat = f'{self.top_level_filename}/{name}/*.po'
- with zipfile.ZipFile(BytesIO(self.zip_data)) as zf:
- for name in fnmatch.filter(zf.namelist(), pat):
+ if self.extracted:
+ pat = f'{self.top_level}/{name}/*.po'
+ for name in glob.glob(pat):
dest = os.path.join(output_dir, name.split('/')[-1])
- zi = zf.getinfo(name)
- with zf.open(zi) as src, open(dest, 'wb') as d:
- shutil.copyfileobj(src, d)
- date_time = time.mktime(zi.date_time + (0, 0, -1))
- os.utime(dest, (date_time, date_time))
+ shutil.copy2(name, dest)
+ else:
+ pat = f'{self.top_level_filename}/{name}/*.po'
+ with zipfile.ZipFile(BytesIO(self.zip_data)) as zf:
+ for name in fnmatch.filter(zf.namelist(), pat):
+ dest = os.path.join(output_dir, name.split('/')[-1])
+ zi = zf.getinfo(name)
+ with zf.open(zi) as src, open(dest, 'wb') as d:
+ shutil.copyfileobj(src, d)
+ date_time = time.mktime(zi.date_time + (0, 0, -1))
+ os.utime(dest, (date_time, date_time))
iso_data = ISOData()
|
