X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/69ad8b5035ac9a4043201622eade4264bcc5b2f3..36b4f5eccf81cc7a650d3bbfcc975d42b3d905c4:/po/xml2po.py

diff --git a/po/xml2po.py b/po/xml2po.py
index 3fef7f82..cfbeaf4b 100755
--- a/po/xml2po.py
+++ b/po/xml2po.py
@@ -1,29 +1,64 @@
 #!/usr/bin/python
 import sys
+import os
+import string
 from xml.sax import make_parser
-from xml.sax.handler import ContentHandler
+from xml.sax.handler import ContentHandler, property_lexical_handler
+try:
+	from _xmlplus.sax.saxlib import LexicalHandler
+	no_comments = False
+except ImportError:
+	class LexicalHandler:
+		pass
+	no_comments = True
 
-class parseXML(ContentHandler):
+class parseXML(ContentHandler, LexicalHandler):
 	def __init__(self, attrlist):
 		self.isPointsElement, self.isReboundsElement = 0, 0
 		self.attrlist = attrlist
+		self.last_comment = None
 
-	def startElement(self, name, attrs):
-		if (attrs.has_key('text')):
-			attrlist[attrs.get('text', "")] = "foo"
+	def comment(self, comment):
+		if comment.find("TRANSLATORS:") != -1:
+			self.last_comment = comment
 
-sys.argv[1]
+	def startElement(self, name, attrs):
+		for x in ["text", "title", "value", "caption"]:
+			try:
+				attrlist.add((attrs[x], self.last_comment))
+				self.last_comment = None
+			except KeyError:
+				pass
 
 parser = make_parser()
 
-attrlist = {}		
+attrlist = set()
+
 contentHandler = parseXML(attrlist)
 parser.setContentHandler(contentHandler)
-parser.parse(sys.argv[1])
+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, v in attrlist.items():
-	print
-	print '#: ' + sys.argv[1]
-	print 'msgid "' + str(k) + '"'
-	print 'msgstr ""'
+	for (k,c) in attrlist:
+		print
+		print '#: ' + arg
+		string.replace(k, "\\n", "\"\n\"")
+		if c:
+			for l in c.split('\n'):
+				print "#. ", l
+		if str(k).strip() != "":
+			print 'msgid "' + str(k) + '"'
+			print 'msgstr ""'
 
+	attrlist = set()