link_dotfiles.py

from __future__ import print_function
import argparse
import os
import os.path as path

EXCLUDED = frozenset(
    [".git", ".gitignore", ".DS_Store", "README.md", "archive", path.basename(__file__)]
)
FOR_REAL = False


def link_dotfiles(src_dir, dest_dir):
    for fname in os.listdir(src_dir):
        if fname in EXCLUDED:
            continue
        src = path.join(src_dir, fname)
        if path.isfile(src):
            symlink_file(fname, src_dir, dest_dir)
        # explore recursive directories
        if path.isdir(src):
            dest = path.join(dest_dir, fname)
            if not path.isdir(dest):
                if not FOR_REAL:
                    print("DRY RUN: mkdir {0}".format(dest))
                else:
                    os.mkdir(dest)
            link_dotfiles(src, dest)


def symlink_file(fname, src, dest):
    src_file = path.join(src, fname)
    dest_link = path.join(dest, fname)
    if path.lexists(dest_link):
        print("{0}: skipped, since dest exists".format(dest_link))
        return
    if not FOR_REAL:
        print("DRY RUN: ln -s {0} {1}".format(src_file, dest_link))
    else:
        os.symlink(src_file, dest_link)
        print("Linked {0}".format(dest_link))


def remove_links_at(src_dir, remove_at):
    if not path.exists(src_dir) or not path.exists(remove_at):
        print("There is no dir at {0} | {1}".format(src_dir, remove_at))
    for fname in os.listdir(src_dir):
        src = path.join(src_dir, fname)
        if path.isfile(src):
            remove_symlink(fname, remove_at)
        # explore recursive directories
        if path.isdir(src):
            dest = path.join(remove_at, fname)
            if path.isdir(dest):
                remove_links_at(src, dest)


def remove_symlink(fname, dest):
    dest_link = path.join(dest, fname)
    if path.lexists(dest_link):
        if FOR_REAL:
            print("rm {0}".format(dest_link))
            os.remove(dest_link)
        else:
            print("DRY RUN: rm {0}".format(dest_link))
    else:
        print("{0}: skipped, since dest exists".format(dest_link))


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Link dotfiles.")
    parser.add_argument(
        "--for-real", help="Makes the filesystem changes", action="store_true"
    )
    parser.add_argument(
        "--remove-at", help="Remove the symlinks made by this script at this location"
    )
    args = parser.parse_args()
    if args.for_real:
        FOR_REAL = True
    dotfile_repo = path.dirname(path.abspath(__file__))
    home = path.expanduser("~")
    if args.remove_at:
        src_dir = path.join(dotfile_repo, args.remove_at)
        remove_at = path.join(home, args.remove_at)
        remove_links_at(src_dir, remove_at)
    else:
        link_dotfiles(dotfile_repo, home)