Fix fake mylink.py was broken on win32.

1 files changed, 37 insertions, 12 deletions

diff --git a/test/fixture/mylink.py b/test/fixture/mylink.py
index fe4af58..6c56427 100644
--- a/test/fixture/mylink.py
+++ b/test/fixture/mylink.py
@@ -1,15 +1,40 @@
 import getopt
 import sys
 
-opts, args = getopt.getopt(sys.argv[1:], 'o:s:')
-for opt, arg in opts:
-    if opt == '-o':
-        out = arg
-
-with open(out, 'w') as ofp:
-    for f in args:
-        with open(f, 'r') as ifp:
-            for line in ifp.readlines():
-                if line[:5] != '#link':
-                    ofp.write(line)
-sys.exit(0)
+def fake_link():
+    opts, args = getopt.getopt(sys.argv[1:], 'o:s:')
+    for opt, arg in opts:
+        if opt == '-o':
+            out = arg
+
+    with open(out, 'w') as ofp:
+        for f in args:
+            with open(f, 'r') as ifp:
+                for line in ifp.readlines():
+                    if line[:5] != '#link':
+                        ofp.write(line)
+    sys.exit(0)
+
+def fake_win32_link():
+    args = sys.argv[1:]
+    while args:
+        a = args[0]
+        if a == '-o':
+            out = args[1]
+            args = args[2:]
+            continue
+        if not a[0] in '/-':
+            break
+        args = args[1:]
+        if a[:5].lower() == '/out:': out = a[5:]
+    with open(args[0], 'rb') as ifp, open(out, 'wb') as ofp:
+        for l in ifp.readlines():
+            if not l.startswith(b'#link'):
+                ofp.write(l)
+    sys.exit(0)
+
+if __name__ == '__main__':
+    if sys.platform == 'win32':
+        fake_win32_link()
+    else:
+        fake_link()