install.py

#!/usr/bin/env python
from __future__ import print_function
import sys
import os
import subprocess
import optparse


user_home_dir = os.path.expanduser("~")
dot_files_dir = os.path.dirname(__file__)
if not dot_files_dir:
    dot_files_dir = '.'
parser = optparse.OptionParser()
parser.add_option("-U", "--uninstall", 
                  action="store_true",
                  dest="uninstall", 
                  default=False,
                  help="uninstall the distribution from the home directory")
parser.add_option("-v", "--verbose", 
                  action="store_true",
                  dest="verbose", 
                  default=False,
                  help="more detailed output")
parser.add_option("-d", "--dir",
                  dest="dir",
                  default=user_home_dir,
                  help="specify a directory to install to/uninstall from",
                  metavar="DIR")

(options, args) = parser.parse_args()


class Command(object):
    warnings = []
    verbose = []


class Install(Command):
    cmd_message = "Installing dot_files to"

    def no_target(self, source, target):
        status = subprocess.call("ln -s %s %s" % (os.path.realpath(source), os.path.realpath(target)), shell=True)
        if status == 0:
            self.verbose.append("linked %s to %s" % (source, target))
        else:
            self.warnings.append("creating link %s to %s failed with status code %d" % (source, target, status))

    def target_link_same(self, source, target, real_source, real_target):
        self.verbose.append("link from %s to %s already exists" % (source, target))       
            
    def target_link_different(self, source, target, real_source, real_target):
        self.warnings.append("link at %s exists but points to %s" % (target, real_target))

    def target_different(self, source, target):
        self.warnings.append("target %s exists but is not a link" % (target, ))


class Uninstall(Command):
    cmd_message = "Uninstalling dot_files from"
    
    def no_target(self, source, target):
        self.verbose.append("target %s was missing" % (target,))

    def target_link_same(self, source, target, real_source, real_target):
        os.unlink(target)
        self.verbose.append("link from %s to %s removed" % (source, target))       
            
    def target_link_different(self, source, target, real_source, real_target):
        self.warnings.append("skipped target at %s: link points to %s" % (target, real_target))

    def target_different(self, source, target):
        self.warnings.append("skipped target at %s: not a link" % (target, ))


def obtain_machine_type():
    """Checks common locations for files holding the current version of the OS."""

    if os.path.exists('/etc/redhat-release'):
        f = open('/etc/redhat-release')
        os_version = f.read().strip()
        os_type = 'redhat'
    elif os.path.exists('/etc/debian_version'):
        f = open('/etc/debian_version')
        os_version = f.read().strip()
        os_type = 'debian'
    elif os.path.exists('/System/Library/CoreServices/SystemVersion.plist'):
        from xml.dom.minidom import parse
        f = parse('/System/Library/CoreServices/SystemVersion.plist')
        keys = f.getElementsByTagName('dict')[0].getElementsByTagName('key') 
        prod_name = ""
        prod_version = ""
        for k in keys:
            if k.childNodes[0].data.strip() == u"ProductName":
                sibling = k.nextSibling
                while sibling.__class__.__name__ != 'Element':
                    sibling = sibling.nextSibling
                if sibling.tagName == u"string":
                    prod_name = sibling.childNodes[0].data.strip()
            if k.childNodes[0].data.strip() == u"ProductVersion":
                sibling = k.nextSibling
                while sibling.__class__.__name__ != 'Element':
                    sibling = sibling.nextSibling
                if sibling.tagName == u"string":
                    prod_version = sibling.childNodes[0].data.strip()
        os_version = "%s %s" % (prod_name, prod_version)
        os_type = 'darwin'
    elif os.path.exists('/cygdrive'):
        os_version = 'Cygwin'
        os_type = 'cygwin'
    else:
        os_version = 'unknown'
        os_type = None

    return os_type, os_version


def handle_subdir(subdir):
    _source_prefix = os.path.join(dot_files_dir,  subdir)
    _target_prefix = os.path.join(options.dir,  subdir)

    if not os.path.isdir(_source_prefix):
        return
    
    if not os.path.isdir(_target_prefix):
        os.makedirs(_target_prefix)
    for entry in os.listdir(_source_prefix):
        source = os.path.join(_source_prefix, entry)
        target = os.path.join(_target_prefix, entry)

        if not os.path.exists(target):
            handler.no_target(source, target)
        elif os.path.islink(target):
            real_source = os.path.realpath(source)
            real_target = os.path.realpath(target)
            if real_source == real_target:
                handler.target_link_same(source, target, real_source, real_target)
            else:
                handler.target_link_different(source, target, real_source, real_target)
        else:
            handler.target_different(source, target)


if not options.uninstall:
    handler = Install()
else:
    handler = Uninstall()

machine_type, machine_version = obtain_machine_type()

if machine_type:
    additional_info = "%s with extras for %s" % (options.dir, machine_version)
else:
    additional_info = options.dir

print(handler.cmd_message, "%s..." % additional_info, end="")

for entry in os.listdir(dot_files_dir):
    source = os.path.join(dot_files_dir, entry)
    target = os.path.join(options.dir, ".%s" % entry)

    if entry.startswith('.') or entry.endswith('~') or \
       entry in ('install.py', 'bin', 'README.rst'):
        continue
    elif entry == 'profile_machine':
        if not machine_type:
            continue
        source = os.path.join(dot_files_dir, entry, machine_type)

    if not os.path.exists(target):
        handler.no_target(source, target)
    elif os.path.islink(target):
        real_source = os.path.realpath(source)
        real_target = os.path.realpath(target)
        if real_source == real_target:
            handler.target_link_same(source, target, real_source, real_target)
        else:
            handler.target_link_different(source, target, real_source, real_target)
    else:
        handler.target_different(source, target)

handle_subdir(".config")
handle_subdir(".local")

print("done.")

for w in handler.warnings:
    print("Warning:", w)

if options.verbose:
    for v in handler.verbose:
        print(v)