|
1 | | -#!/usr/bin/env python3 |
2 | | -import glob |
3 | | -import json |
4 | | -import os |
| 1 | +"""Please note that this script requires a Transifex API token to run.""" |
| 2 | +import subprocess |
| 3 | +from functools import partial |
| 4 | +from pathlib import Path |
5 | 5 | import re |
6 | | -import sys |
7 | | -import urllib.request |
8 | 6 |
|
| 7 | +run = partial(subprocess.run, check=True) |
9 | 8 |
|
10 | | -def list_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 | +def init_project(): |
| 11 | + run(["tx", "init"]) |
| 12 | + |
| 13 | + |
| 14 | +def add_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 | +def name_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 | | - while True: |
24 | | - resp = urllib.request.urlopen(next_) |
25 | | - result = json.loads(resp.read().decode("utf-8")) |
26 | | - resources.extend([i["slug"] for i in result]) |
27 | | - link = re.findall('<([^<]*)>; rel="next"', resp.getheader("Link") or "") |
28 | | - if not link: |
29 | | - break |
30 | | - next_ = link[0] |
31 | | - return resources |
32 | | - |
33 | | - |
34 | | -def render_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 | | - for resource in sorted(resources): |
45 | | - if resource == "glossary_": |
46 | | - filename = "glossary.po" |
47 | | - elif resource == "sphinx": |
48 | | - filename = "sphinx.po" |
49 | | - else: |
50 | | - pattern = resource.replace("--", "/").replace("_", "?") |
51 | | - matches = glob.glob(pattern + ".rst") |
52 | | - if len(matches) == 0: |
53 | | - print("missing", resource, file=sys.stderr) |
54 | | - continue |
55 | | - elif len(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 | | - return conf |
| 41 | + return f"{prefix}{resource}\n{override_prefix}{override_resource}" |
| 42 | + |
| 43 | + |
| 44 | +def patch_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 | | - import argparse |
| 53 | + from argparse import ArgumentParser |
| 54 | + import os |
| 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() |
0 commit comments