|
1 | | -#!/usr/bin/env python3 |
2 | | -importglob |
3 | | -importjson |
4 | | -importos |
| 1 | +"""Please note that this script requires a Transifex API token to run.""" |
| 2 | +importsubprocess |
| 3 | +fromfunctoolsimportpartial |
| 4 | +frompathlibimportPath |
5 | 5 | importre |
6 | | -importsys |
7 | | -importurllib.request |
8 | 6 |
|
| 7 | +run=partial(subprocess.run,check=True) |
9 | 8 |
|
10 | | -deflist_resources(token,project): |
11 | | -auth_handler=urllib.request.HTTPBasicAuthHandler() |
12 | | -auth_handler.add_password( |
13 | | -realm="api",uri="https://api.transifex.com/",user="api",passwd=token |
| 9 | + |
| 10 | +definit_project(): |
| 11 | +run(["tx","init"]) |
| 12 | + |
| 13 | + |
| 14 | +defadd_files(version:str): |
| 15 | +run( |
| 16 | + [ |
| 17 | +"tx", |
| 18 | +"add", |
| 19 | +"remote", |
| 20 | +"--file-filter", |
| 21 | +"trans/<lang>/<resource_slug>.<ext>", |
| 22 | +f"https://www.transifex.com/python-doc/{version}/dashboard/", |
| 23 | + ] |
14 | 24 | ) |
15 | | -opener=urllib.request.build_opener(auth_handler) |
16 | | -urllib.request.install_opener(opener) |
17 | | -next_= ( |
18 | | -"https://api.transifex.com/organizations/python-doc/projects/" |
19 | | -+project |
20 | | -+"/resources/" |
| 25 | + |
| 26 | + |
| 27 | +FILTER_PATTERN=re.compile( |
| 28 | +r"^(?P<prefix>file_filter( *)=( *))(?P<resource>.+)$",re.MULTILINE |
| 29 | +) |
| 30 | + |
| 31 | + |
| 32 | +defname_replacer(match:re.Match[str]): |
| 33 | +prefix,resource=match.group("prefix","resource") |
| 34 | +override_prefix=prefix.replace("file_filter","trans.zh_CN") |
| 35 | +override_resource= ( |
| 36 | +resource.replace("trans/<lang>/","") |
| 37 | + .replace("--","/") |
| 38 | + .replace("glossary_","glossary") |
| 39 | + .replace("_",".") |
21 | 40 | ) |
22 | | -resources= [] |
23 | | -whileTrue: |
24 | | -resp=urllib.request.urlopen(next_) |
25 | | -result=json.loads(resp.read().decode("utf-8")) |
26 | | -resources.extend([i["slug"]foriinresult]) |
27 | | -link=re.findall('<([^<]*)>; rel="next"',resp.getheader("Link")or"") |
28 | | -ifnotlink: |
29 | | -break |
30 | | -next_=link[0] |
31 | | -returnresources |
32 | | - |
33 | | - |
34 | | -defrender_config(doc_dir,project,resources): |
35 | | -os.chdir(doc_dir) |
36 | | -tpl=""" |
37 | | -
|
38 | | -[{project}.{resource}] |
39 | | -trans.zh_CN = {filename} |
40 | | -source_lang = en |
41 | | -type = PO""" |
42 | | -conf="""[main] |
43 | | -host = https://www.transifex.com""" |
44 | | -forresourceinsorted(resources): |
45 | | -ifresource=="glossary_": |
46 | | -filename="glossary.po" |
47 | | -elifresource=="sphinx": |
48 | | -filename="sphinx.po" |
49 | | -else: |
50 | | -pattern=resource.replace("--","/").replace("_","?") |
51 | | -matches=glob.glob(pattern+".rst") |
52 | | -iflen(matches)==0: |
53 | | -print("missing",resource,file=sys.stderr) |
54 | | -continue |
55 | | -eliflen(matches)==1: |
56 | | -filename=matches[0].replace(".rst",".po") |
57 | | -else: |
58 | | -print("multi match",resource,pattern,matches,file=sys.stderr) |
59 | | -conf+=tpl.format(project=project,resource=resource,filename=filename) |
60 | | -returnconf |
| 41 | +returnf"{prefix}{resource}\n{override_prefix}{override_resource}" |
| 42 | + |
| 43 | + |
| 44 | +defpatch_config(): |
| 45 | +tx_config_path=Path(".tx","config") |
| 46 | + |
| 47 | +config_content=tx_config_path.read_text("utf-8") |
| 48 | +config_content=FILTER_PATTERN.sub(name_replacer,config_content) |
| 49 | +tx_config_path.write_text(config_content,"utf-8") |
61 | 50 |
|
62 | 51 |
|
63 | 52 | if__name__=="__main__": |
64 | | -importargparse |
| 53 | +fromargparseimportArgumentParser |
| 54 | +importos |
| 55 | + |
| 56 | +parser=ArgumentParser() |
65 | 57 |
|
66 | | -parser=argparse.ArgumentParser() |
67 | 58 | parser.add_argument("--token") |
68 | | -parser.add_argument("--project") |
69 | | -parser.add_argument("--doc-dir") |
70 | | -args=parser.parse_args() |
| 59 | +parser.add_argument("--version") |
| 60 | + |
| 61 | +params=parser.parse_args() |
71 | 62 |
|
72 | | -resources=list_resources(args.token,args.project) |
73 | | -conf=render_config(args.doc_dir,args.project,resources) |
74 | | -print(conf) |
| 63 | +os.environ["TX_TOKEN"]=params.token |
75 | 64 |
|
76 | | -# vim: set et ts=4 sw=4 sts=4: |
| 65 | +init_project() |
| 66 | +add_files(params.version) |
| 67 | +patch_config() |