setup.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""Distutils/setuptools packaging for NetworkMgr.

Installs the NetworkMgr package, the two entry scripts, the devd action
scripts, the devd and sudoers fragments, the icons and the translations.
"""

import os
import sys
from platform import system
from subprocess import run
from setuptools import setup, Command, glob
from DistUtilsExtra.command.build_extra import build_extra
from DistUtilsExtra.command.build_i18n import build_i18n
from DistUtilsExtra.command.clean_i18n import clean_i18n

__VERSION__ = '7.0'
PROGRAM_VERSION = __VERSION__

PREFIX = '/usr/local' if system() == 'FreeBSD' else sys.prefix

def datafilelist(installbase, sourcebase):
    """Map a source directory tree onto its install destination.

    Args:
        installbase (str): Directory the tree is installed under.
        sourcebase (str): Directory the tree is read from.

    Returns:
        list[tuple[str, list[str]]]: A (destination, files) pair per directory,
        in the form setup()'s data_files expects.
    """
    data_file_list = []
    for root, _, files in os.walk(sourcebase):
        file_list = [os.path.join(root, name) for name in files]
        data_file_list.append((root.replace(sourcebase, installbase), file_list))
    return data_file_list

class UpdateTranslationsCommand(Command):
    """Custom command to extract messages and update .po files."""

    description = 'Extract messages to .pot and update .po'
    user_options = []  # No custom options

    def initialize_options(self):
        """Set up the command's options. This command takes none."""

    def finalize_options(self):
        """Validate the command's options. This command takes none."""

    def run(self):
        """Extract messages into the .pot file and merge them into every .po."""
        # Define paths
        pot_file = 'po/networkmgr.pot'
        po_files = glob.glob('po/*.po')
        # Step 1: Extract messages to .pot file
        print("Extracting messages to .pot file...")
        os.system(f'xgettext --from-code=UTF-8 -L Python -o {pot_file} NetworkMgr/*.py networkmgr')
        # Step 2: Update .po files with the new .pot file
        print("Updating .po files with new translations...")
        for po_file in po_files:
            print(f"Updating {po_file}...")
            os.system(f'msgmerge -U {po_file} {pot_file}')
        print("Translation update complete.")

class CreateTranslationCommand(Command):
    """Custom command to create a new .po file for a specific language."""
    locale = None
    description = 'Create a new .po file for the specified language'
    user_options = [
        ('locale=', 'l', 'Locale code for the new translation (e.g., fr, es)')
    ]

    def initialize_options(self):
        """Set the locale option to its unset default."""
        self.locale = None

    def finalize_options(self):
        """Check that a locale was given.

        Raises:
            ValueError: If --locale was not passed on the command line.
        """
        if self.locale is None:
            raise ValueError("You must specify the locale code (e.g., --locale=fr)")

    def run(self):
        """Create a .po file for the requested locale, extracting .pot first."""
        # Define paths
        pot_file = 'po/networkmgr.pot'
        po_dir = 'po'
        po_file = os.path.join(po_dir, f'{self.locale}.po')
        # Check if the .pot file exists
        if not os.path.exists(pot_file):
            print("Extracting messages to .pot file...")
            os.system(f'xgettext --from-code=UTF-8 -L Python -o {pot_file} NetworkMgr/*.py networkmgr')
        # Create the new .po file
        if not os.path.exists(po_file):
            print(f"Creating new {po_file} for locale '{self.locale}'...")
            os.makedirs(po_dir, exist_ok=True)
            os.system(f'msginit --locale={self.locale}.UTF-8 --input={pot_file} --output-file={po_file}')
        else:
            print(f"PO file for locale '{self.locale}' already exists: {po_file}")


networkmgr_share = [
    'src/auto-switch.py',
    'src/link-up.py',
    'src/setup-nic.py',
    'src/wpa_supplicant.conf'
]

data_files = [
    (f'{PREFIX}/etc/xdg/autostart', ['src/networkmgr.desktop']),
    (f'{PREFIX}/share/networkmgr', networkmgr_share),
    (f'{PREFIX}/etc/sudoers.d', ['src/sudoers.d/networkmgr']),
    (f'{PREFIX}/etc/devd', ['src/networkmgr.conf'])
]

data_files.extend(datafilelist(f'{PREFIX}/share/icons/hicolor', 'src/icons'))
data_files.extend(datafilelist(f'{PREFIX}/share/locale', 'build/mo'))


setup(
    name="NetworkMgr",
    version=PROGRAM_VERSION,
    description="NetworkMgr is a tool to manage FreeBSD/GhostBSD network",
    license='BSD',
    author='Eric Turgeon',
    url='https://github/GhostBSD/networkmgr/',
    package_dir={'': '.'},
    data_files=data_files,
    install_requires=['setuptools'],
    packages=['NetworkMgr'],
    scripts=['networkmgr', 'networkmgr_configuration'],
    cmdclass={
        'create_translation': CreateTranslationCommand,
        'update_translations': UpdateTranslationsCommand,
        "build": build_extra,
        "build_i18n": build_i18n,
        "clean": clean_i18n
    }
)

run('gtk-update-icon-cache -f /usr/local/share/icons/hicolor', shell=True, check=False)