tx_stats.py

#!/usr/bin/python
"""Print translation percentage to update statistics."""

import json
import os
import configparser
import urllib.request
from datetime import datetime, timezone

# Get language and project from environment variables
language = os.environ.get("PYDOC_LANGUAGE")
project = os.environ.get("PYDOC_TX_PROJECT")
if language is None:
    raise ValueError("The PYDOC_LANGUAGE environment variable must be set.")
if project is None:
    raise ValueError("The PYDOC_TX_PROJECT environment variable must be set.")


# Try to read API token from TX_TOKEN env and then from ~/.transifexrc
def get_transifex_token():
    key = os.environ.get("TX_TOKEN")
    if key is None:
        config = configparser.ConfigParser()
        config.read(os.path.expanduser("~/.transifexrc"))
        try:
            key = config["https://www.transifex.com"]["token"]
        except KeyError:
            raise ValueError("Unable to retrieve Transifex API token.")
    return key


# API URL setup
url_template = (
    "https://rest.api.transifex.com/resource_language_stats"
    "?filter[project]=o%3Apython-doc%3Ap%3A{project}"
    "&filter[language]=l%3A{language}"
)

# Get the authorization key
key = get_transifex_token()

url = url_template.format(project=project, language=language)

headers = {"accept": "application/vnd.api+json", "authorization": f"Bearer {key}"}

# Initialize counters
total_strings = 0
translated_strings = 0


# Function to make an API request and handle potential errors
def fetch_data(url):
    request = urllib.request.Request(url=url, headers=headers)
    try:
        with urllib.request.urlopen(request) as response:
            return json.loads(response.read().decode("utf-8"))
    except urllib.error.URLError as e:
        raise ConnectionError(f"Error fetching data: {e}")
    except json.JSONDecodeError as e:
        raise ValueError(f"Error decoding JSON response: {e}")


# Fetch and process translation stats
while url:
    data = fetch_data(url)
    url = data["links"].get("next")
    for resource in data["data"]:
        translated_strings += resource["attributes"]["translated_strings"]
        total_strings += resource["attributes"]["total_strings"]

# Calculate translation percentage
if total_strings == 0:
    raise ValueError("Total strings cannot be zero.")

percentage = f"{(translated_strings / total_strings):.2%}"

# Print the result as JSON
print(
    json.dumps(
        {
            "translation": percentage,
            "total": total_strings,
            "updated_at": datetime.now(timezone.utc).isoformat(timespec="seconds")
            + "Z",
        }
    )
)