Create mklink.py

1 files changed, 71 insertions, 0 deletions

diff --git a/mklink.py b/mklink.py
new file mode 100644
index 0000000..2522533
--- /dev/null
+++ b/mklink.py
@@ -0,0 +1,71 @@
+"""
+A wrapper script for ln.exe to make it look like the MKLINK utility
+found from Windows Vista onwards. To fully utilise this, you should
+also have a batch script that should look something like this:
+
+    @ECHO OFF
+    python %~dp0mklink.py %*
+
+Name the file "mklink.cmd" and put it in PATH. Now you can use the
+fake mklink utility like you would use the real.
+
+You can find instruction for installing ln.exe from
+
+    http://schinagl.priv.at/nt/hardlinkshellext/hardlinkshellext.html#symboliclinksforwindowsxp
+
+"""
+import argparse
+import subprocess
+
+def MyFormatter(raw):
+    """Make the help output look a little bit more like the real deal
+    (i.e., this omits the "usage: " part in the beginning of the help).
+
+    """
+    class MyFormatter_(argparse.HelpFormatter):
+        def format_help(self):
+            return raw
+    return MyFormatter_
+
+usage_str = """Creates a symbolic link.
+
+MKLINK [[/D] | [/H] | [/J]] Link Target
+
+        /D      Creates a directory symbolic link.  Default is a file
+                symbolic link.
+        /H      Creates a hard link instead of a symbolic link.
+        /J      Creates a Directory Junction.
+        Link    specifies the new symbolic link name.
+        Target  specifies the path (relative or absolute) that the new link
+                refers to.
+
+"""
+
+parser = argparse.ArgumentParser(prog='MKLINK', prefix_chars='/',
+    usage=usage_str, add_help=False,
+    formatter_class=MyFormatter(raw=usage_str))
+parser.add_argument('/?', dest='help', action='help')
+group = parser.add_mutually_exclusive_group()
+group.add_argument('/D', dest='symlink', default=False, action='store_true')
+group.add_argument('/d', dest='symlink', default=False, action='store_true')
+group.add_argument('/H', dest='hardlink', default=False, action='store_true')
+group.add_argument('/h', dest='hardlink', default=False, action='store_true')
+group.add_argument('/J', dest='junction', default=False, action='store_true')
+group.add_argument('/j', dest='junction', default=False, action='store_true')
+parser.add_argument('Link', dest='link')
+parser.add_argument('Target', dest='target')
+
+args = parser.parse_args()
+
+if (not args.symlink) and (not args.hardlink) and (not args.junction):
+    args.symlink = True
+
+if args.symlink:
+    subprocess.call(['ln.exe', '-s', args.target, args.link])
+elif args.hardlink:
+    subprocess.call(['ln.exe', args.target, args.link])
+elif args.junction:
+    subprocess.call(['ln.exe', '-j', args.target, args.link])
+else:
+    print("invalid options!")
+    sys.exit(1)