xml2po.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
from __future__ import print_function
try:
	import builtins
except ImportError:
	import __builtin__ as builtins

import sys
import os
import re
from xml.sax import make_parser
from xml.sax.handler import ContentHandler, property_lexical_handler

try:
	from _xmlplus.sax.saxlib import LexicalHandler
	no_comments = False
except ImportError:
	class LexicalHandler:
		def __init__(self):
			pass

	no_comments = True


class parseXML(ContentHandler, LexicalHandler):
	def __init__(self, attrlist):
		self.isPointsElement, self.isReboundsElement = 0, 0
		self.attrlist = attrlist
		self.last_comment = None
		self.ishex = re.compile('#[0-9a-fA-F]+\Z')  # noqa: W605

	def comment(self, comment):
		if "TRANSLATORS:" in comment:
			self.last_comment = comment

	def startElement(self, name, attrs):
		for x in ["text", "title", "value", "caption", "summary", "description"]:
			try:
				k = builtins.str(attrs[x])
				if k.strip() != "" and not self.ishex.match(k):
					attrlist.add((k, self.last_comment))
					self.last_comment = None
			except KeyError:
				pass


parser = make_parser()

attrlist = set()

contentHandler = parseXML(attrlist)
parser.setContentHandler(contentHandler)
if not no_comments:
	parser.setProperty(property_lexical_handler, contentHandler)

for arg in sys.argv[1:]:
	if os.path.isdir(arg):
		for file in os.listdir(arg):
			if file.endswith(".xml"):
				parser.parse(os.path.join(arg, file))
	else:
		parser.parse(arg)

	attrlist = list(attrlist)
	attrlist.sort(key=lambda a: a[0])

	for (k, c) in attrlist:
		print()
		print('#: ' + arg)
		k.replace("\\n", "\"\n\"")
		if c:
			for line in c.split('\n'):
				print("#. ", line)
		print('msgid "' + builtins.str(k) + '"')
		print('msgstr ""')

	attrlist = set()