pogrep/pogrep.py

#!/usr/bin/env python3
"""Find translations exemples by grepping in .po files.
"""

__version__ = "0.0.1"

import argparse
from glob import glob
import os
from textwrap import fill

import regex
import polib
from tabulate import tabulate


def find_in_po(pattern):
    table = []
    try:
        _, columns = os.popen("stty size", "r").read().split()
        available_width = int(columns) // 2 - 3
    except:
        available_width = 80 // 2 - 3

    for file in glob("**/*.po"):
        pofile = polib.pofile(file)
        for entry in pofile:
            if entry.msgstr and regex.search(pattern, entry.msgid):
                table.append(
                    [
                        fill(entry.msgid, width=available_width),
                        fill(entry.msgstr, width=available_width),
                    ]
                )
    print(tabulate(table, tablefmt="fancy_grid"))


def parse_args():
    parser = argparse.ArgumentParser(description="Find translated words.")
    parser.add_argument("pattern")
    return parser.parse_args()


def main():
    args = parse_args()
    find_in_po(args.pattern)


if __name__ == "__main__":
    main()