Improve python script that sets translation authors to be able to process all files at once

data/po/update_po_authors.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-
+# -*- coding: utf-8 -*-
 # A simple script that adds all authors from transifex, which are
 # listed in comments at the beginning of the file, to the 
 # 'translator-credits' translations - where launchpad added them
@@ -19,75 +19,79 @@ import re
 import sys
 
 if __name__ == "__main__":
-    if len(sys.argv)!=2:
+    if len(sys.argv) < 2:
         print "Usage: getpo_authors.py PATH_TO_PO_FILE"
         sys.exit(-1)
 
-    f = open(sys.argv[1], "r")
-    if not f:
-        print "Can not find", sys.argv[1]
-        exit
-    lines = f.readlines()
+    for filename in sys.argv[1:]:
+        print("Processing file ", filename)
+        f = open(filename, "r")
+        if not f:
+            print "Can not find", filename
+            exit
+        lines = f.readlines()
 
-    f.close()
+        f.close()
 
-    new_authors = []
-    found        = 0
+        new_authors = []
+        found        = 0
 
-    # Find all authors with a simple finite state machine:
-    contributions = -1
-    line_count    =  0
-    for i in lines:
-        line = i[:-1]   # remove \n
-        if line=="# Translators:":
-            found = 1
-        elif found and line[:2]=="# " and line [:14]!="# FIRST AUTHOR":
-            new_authors.append(line[2:])
-        elif line[:5]=="msgid":
-            found = 0
-        elif line[:31]== "msgstr \"Launchpad Contributions":
-            contributions = line_count
-        line_count = line_count + 1
+        # Find all authors with a simple finite state machine:
+        contributions = -1
+        line_count    =  0
+        for i in lines:
+            line = i[:-1]   # remove \n
+            if line=="# Translators:":
+                found = 1
+            elif found and line[:2]=="# " and line [:14]!="# FIRST AUTHOR":
+                new_authors.append(line[2:])
+            elif line[:5]=="msgid":
+                found = 0
+            elif line[:31]== "msgstr \"Launchpad Contributions":
+                contributions = line_count
+            line_count = line_count + 1
 
 
-    # Delete all email addresses - not sure if the authors
-    # would want them to be published
-    email=re.compile(" *<.*@.*\..*> *")   # one @ and one dot at least
-    for i in range(len(new_authors)):
-        g = email.search(new_authors[i])
-        if g:
-            new_authors[i] = new_authors[i][:g.start()] \
-                           + new_authors[i][g.end():]
+        # Delete all email addresses - not sure if the authors
+        # would want them to be published
+        email=re.compile(" *<.*@.*\..*> *")   # one @ and one dot at least
+        for i in range(len(new_authors)):
+            g = email.search(new_authors[i])
+            if g:
+                new_authors[i] = new_authors[i][:g.start()] \
+                               + new_authors[i][g.end():]
 
-    # Get the old authors from the translator-credits string:
-    if contributions>0:
-        # Ignore the first entry, which is "msgstr ...", and the
-        # last two characters, which are the '"\n'.
-        old_authors = lines[contributions][:-2].split("\\n")[1:]
-        for i in range(len(old_authors)):
-            old_authors[i] = old_authors[i].strip()
-    else:
-        old_authors=[]
-    
-    all_authors = old_authors + new_authors;
-    all_authors = sorted(all_authors, key=lambda x: x.lower())
-    all_authors_string = reduce(lambda x,y: x+"\\n"+y, all_authors, "")
+        # Get the old authors from the translator-credits string:
+        if contributions>0:
+            # Ignore the first entry, which is "msgstr ...", and the
+            # last two characters, which are the '"\n'.
+            old_authors = lines[contributions][:-2].split("\\n")[1:]
+            for i in range(len(old_authors)):
+                old_authors[i] = old_authors[i].strip()
+        else:
+            old_authors=[]
+        
+        all_authors = old_authors + new_authors;
+        all_authors = sorted(all_authors, key=lambda x: x.lower())
+        all_authors_string = reduce(lambda x,y: x+"\\n"+y, all_authors, "")
 
-    credits_line = "msgstr \"Launchpad Contributions:%s\"\n"%all_authors_string
-    # If no old authors exists, write a new entry:
-    if contributions==-1:
-        lines.append("\n")
-        lines.append("#: src/states_screens/credits.cpp:209\n")
-        lines.append("msgid \"translator-credits\"\n")
-        lines.append(credits_line)
-    else:
-        # Otherwise just replace the old contribution string
-        lines[contributions] = credits_line
+        credits_line = "msgstr \"Launchpad Contributions:%s\"\n"%all_authors_string
+        # If no old authors exists, write a new entry:
+        if contributions==-1:
+            lines.append("\n")
+            lines.append("#: src/states_screens/credits.cpp:209\n")
+            lines.append("msgid \"translator-credits\"\n")
+            lines.append(credits_line)
+        else:
+            # Otherwise just replace the old contribution string
+            lines[contributions] = credits_line
 
 
-    # Overwrite old file
-    f = open(sys.argv[1], "w")
-    for i in lines:
-        f.write(i)
-    f.close()
+        # Overwrite old file
+        f = open(filename, "w")
+        for i in lines:
+            f.write(i)
+        f.close()
+        
+        print("Done with ", filename)