From 91fde6f5e716a8543f0195932a38c5c191e0abba Mon Sep 17 00:00:00 2001
From: mcmartin <mcmartin@8092fc87-c524-0410-9efc-e669fe64eaf9>
Date: Sun, 21 Mar 2010 07:39:59 +0000
Subject: [PATCH] verify_rmp.py now works on uqm files.

git-svn-id: svn://svn.code.sf.net/p/sc2/code/trunk@3536 8092fc87-c524-0410-9efc-e669fe64eaf9
---
 tools/rmpver/verify_rmp.py | 34 ++++++++++++++++++++++------------
 1 file changed, 22 insertions(+), 12 deletions(-)

diff --git a/tools/rmpver/verify_rmp.py b/tools/rmpver/verify_rmp.py
index 37671ab01..4167c5a7e 100644
--- a/tools/rmpver/verify_rmp.py
+++ b/tools/rmpver/verify_rmp.py
@@ -1,9 +1,10 @@
 #!/usr/bin/python
 
 import sys
+import zipfile
 
-def read_rmp(fname):
-    lines = file(fname).readlines()
+def read_rmp(f):
+    lines = f.readlines()
     result = {}
     errors = []
     for (l, i) in zip(lines, xrange(1, 99999)):
@@ -44,18 +45,27 @@ def check_rmp(base, target):
         if t not in valid_types:
             print "%d: new type '%s'" % (i, t)
 
+def loot_uqm(fname):
+    zf = zipfile.ZipFile(fname)
+    files = zf.namelist()
+    rmps = [x for x in files if x.endswith(".rmp")]
+    result = {}
+    errors = []
+    for rmp in rmps:
+        (b, e) = read_rmp(zf.open(rmp))
+        result.update(b)
+        errors.extend(e)
+        for k in b:
+            (i, t, v) = b[k]
+            if v.startswith("addons/"):
+                v = v[7:]
+            if v not in files:
+                errors.append((i, "Unknown file '%s'" % v))
+    return (result, errors)
 
 if __name__ == '__main__':
-    base = sys.argv[1]
-    target = sys.argv[2]
-    (b, e) = read_rmp(base)
-    if e != []:
-        for (i, x) in e:
-            print "%d: %s" % (i, x)
-        print "Errors found in base resource index, cannot continue"
-    else:
-        (t, e) = read_rmp(target)
+    for uqm in sys.argv[1:]:
+        (b, e) = loot_uqm(uqm)
         if e != []:
             for (i, x) in e:
                 print "%d: %s" % (i, x)
-        check_rmp(b, t)