cinnamon-spices-makepot

#!/usr/bin/python3

import argparse
import json
import os
import subprocess
import sys

from glob import glob


def parse_args():
    """
    Get command line arguments and process translation actions
    """
    parser = argparse.ArgumentParser()
    parser.description = 'Arguments for cinnamon-spices-makepot'
    parser.add_argument('-i', '--install', action='store_true',
                        help='Compiles and installs any .po files contained in'
                             ' a spices po folder.'
                             'Use this option to test your translations'
                             ' locally before uploading to Spices.')
    parser.add_argument('-r', '--remove', action='store_true',
                        help='The opposite of install, removes local '
                             'translations.')
    parser.add_argument('-a', '--all', action='store_true',
                        help='Create or update translation templates for all '
                             'Spices')
    parser.add_argument('uuid', type=str, metavar='UUID', nargs='?',
                        help='the UUID of the Spice')
    args = parser.parse_args()
    if args.uuid and args.uuid.endswith('/'):
        args.uuid = args.uuid.replace('/', '')
    if args.all and not args.uuid:
        for file_path in os.listdir("."):
            if os.path.isdir(file_path) and not file_path.startswith("."):
                if args.install:
                    install_po(file_path, True)
                elif args.remove:
                    remove_po(file_path, True)
                else:
                    make_pot(file_path, True)
    elif args.install and args.remove:
        print('Only -i/--install OR -r/--remove may be specified. Not both.')
        sys.exit(1)
    elif args.install and args.uuid:
        install_po(args.uuid)
    elif args.remove and args.uuid:
        remove_po(args.uuid)
    elif args.uuid:
        make_pot(args.uuid)
    else:
        parser.print_help()


def install_po(uuid: str, _all: bool = False):
    """
    Install translation files locally from the po directory of the UUID
    """
    uuid_path = f'{uuid}/files/{uuid}'
    try:
        contents = os.listdir(uuid_path)
    except FileNotFoundError:
        print(f'Translations not found for: {uuid}')
        if not _all:
            sys.exit(1)
    home = os.path.expanduser("~")
    locale_inst = f'{home}/.local/share/locale'
    if 'po' in contents:
        po_dir = f'{uuid_path}/po'
        for file in os.listdir(po_dir):
            if file.endswith('.po'):
                lang = file.split(".")[0]
                locale_dir = os.path.join(locale_inst, lang, 'LC_MESSAGES')
                os.makedirs(locale_dir, mode=0o755, exist_ok=True)
                subprocess.run(['msgfmt', '-c', os.path.join(po_dir, file),
                               '-o', os.path.join(locale_dir, f'{uuid}.mo')],
                               check=True)


def remove_po(uuid: str, _all: bool = False):
    """
    Remove local translation files for the UUID
    """
    home = os.path.expanduser("~")
    locale_inst = f'{home}/.local/share/locale'
    uuid_mo_list = glob(f'{locale_inst}/**/{uuid}.mo', recursive=True)
    if not uuid_mo_list:
        print(f'No translation files found for: {uuid}')
        if not _all:
            sys.exit(1)
    for uuid_mo_file in uuid_mo_list:
        os.remove(uuid_mo_file)


def make_pot(uuid: str, _all: bool = False):
    """
    Make the translation template file for the UUID
    """
    _pwd = sys.argv[1] if not _all else f'{os.getcwd()}/{uuid}'
    _return_pwd = '../../..'
    po_dir = f'{_pwd}/files/{uuid}/po'
    uuid_dir = f'{_pwd}/files/{uuid}'
    pot_file = uuid + '.pot'
    outfile = os.path.join(po_dir, pot_file)
    if os.path.exists(outfile):
        os.remove(outfile)
    elif not os.path.exists(po_dir):
        os.mkdir(po_dir)
    subprocess.run(["cinnamon-xlet-makepot", "-o", outfile, _pwd],
                   check=True)
    os.chdir(uuid_dir)
    pot_path = f'po/{pot_file}'
    glade_list = glob('**/*.glade', recursive=True)
    for glade_file in glade_list:
        subprocess.run(["xgettext", "-jL", "Glade", "-o", pot_path,
                       glade_file], check=True)
    shell_list = glob('**/*.sh', recursive=True)
    for shell_file in shell_list:
        subprocess.run(["xgettext", "-jL", "Shell", "-o", pot_path,
                       shell_file], stderr=subprocess.DEVNULL, check=True)
    if not _all:
        os.chdir(f'{_return_pwd}')
    metadata_file = 'metadata.json' if _all else f'{uuid}/files/{uuid}/metadata.json'
    try:
        with open(metadata_file, encoding='utf-8') as meta:
            metadata = json.load(meta)
        version = str(metadata['version'])
    except (FileNotFoundError, KeyError):
        print(f"{uuid}: metadata.json or version not found")
        version = "1.0"
    address = 'https://github.com/linuxmint/cinnamon-spices-desklets/issues'
    subprocess.run(["xgettext", "-w", "79", "--package-name", uuid,
                    "--foreign-user", "--msgid-bugs-address", address,
                    "--package-version", version, "-o", outfile, outfile],
                   check=True)
    glob_path = 'po/*.po' if _all else f'{uuid}/**/*.po'
    po_list = glob(glob_path, recursive=True)
    for po_file in po_list:
        *po_dir_list, po_ext = po_file.split('/')
        po_dir_str = '/'.join(po_dir_list)
        if not os.getcwd().endswith(po_dir_str):
            os.chdir(po_dir_str)
        po_lang = po_ext.split('.')[0]
        subprocess.run(["intltool-update", "-g", uuid, "-d", po_lang],
                       check=True)
        subprocess.run(["msgattrib", "--no-obsolete", "-w", "79", "-o",
                        po_ext, po_ext],
                       check=True)
    if po_list:
        _return_pwd += '/..'
    if _all:
        os.chdir(_return_pwd)


if __name__ == "__main__":
    parse_args()