3_get_diff_PO.py

#!/usr/bin/python

from __future__ import print_function
import re
import csv
import sys
import argparse
from po2csv import po2dict, dict2PO

def getDiffPO(po_file, all_items):
    new_items = {}

    # read from po_file
    current_items = {}
    po2dict(po_file, current_items)

    # use all_items as ground to find diff (than po_file)
    msgid_list = all_items.keys()
    for msgid in msgid_list:
        if msgid not in current_items:
            new_items[msgid] = all_items[msgid] 

    return new_items


parser = argparse.ArgumentParser(description='Get new items by comparing existing PO file (e.g. zh_CN.po, specified using argument --input) with new POT files (geoda.pot and xrc.pot extracted from source code). New msgid with empty msgstr will be written into a new PO file specified by argument --output')
parser.add_argument('pot_files', type=str, nargs='+', help='paths of template POT files')
parser.add_argument('--input', required=True, dest='input_po_file', type=str, help='path of an existing to-be-updated po_file')
parser.add_argument('--output', required=True, dest='output_po_file', help='path of an output po_file')
args = parser.parse_args()

if __name__ == "__main__":
    pot_files = args.pot_files
    exist_po = args.input_po_file
    output_po = args.output_po_file

    # construct a dictionary from all POT file, 
    # so that newly added msgid will be included
    all_items = {}
    for pot in pot_files:
        po2dict(pot, all_items)

    new_items = getDiffPO(exist_po, all_items)
    dict2PO(new_items, output_po)