install.py

#!/usr/bin/env python3
import argparse
import os
import re
import shutil
import sys
import mmap

from collections import Counter

source_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'src')
install_dir = '/usr/bin/'


def copy(src, dst):
    """Copy file <src> to <dst>, creating all necessary dirs in between"""
    try:
        assert os.path.isfile(src)
        assert not os.path.isdir(dst)
        if not os.path.isdir(os.path.dirname(dst)):
            os.makedirs(os.path.dirname(dst))
        shutil.copy(src, dst)
        return True
    except Exception:
        return False


def check_dependencies():
    """Check for required dependencies."""
    required_found = True
    print('Checking dependencies ...\n')
    print('Required dependencies:')

    print('    Python Imaging Library .......')
    try:
        from PIL import Image
        print('OK')
    except ImportError:
        required_found = False
        try:
            import Image
            assert Image.__version__ >= '1.1.5'
            required_found = True
        except AssertionError:
            print('    !!! version ' + Image.__version__ + ' found, but 1.1.5 or higher is required')
            required_found = False
        except ImportError:
            pass
        finally:
            if not required_found:
                print('    ... Not found')
    if not required_found:
        print('\nCould not find all required dependencies!')
        print('Please install them and try again.')
        sys.exit(1)
    print('')

MAJOR_VERSION_EXTRACTOR = re.compile(r'.* (\d*)\..*')
def gnome_shell_version():
    major_version = None
    gnome_session_match = MAJOR_VERSION_EXTRACTOR.search(os.popen('gnome-session --version').read())  # output is "gnome-session X.Y.Z"
    major_version = gnome_session_match and gnome_session_match.group(1)
    if not major_version:
        gnome_shell_match = MAJOR_VERSION_EXTRACTOR.search(os.popen('gnome-shell --version').read())  # output is "GNOME Shell X.Y.Z"
        major_version = gnome_shell_match and gnome_shell_match.group(1)
    return ('gnome%s' % major_version) if major_version else None

VERSION_GUSSERS = {
    'cinnamon': lambda: 'gnome3',
    'enlightenment': lambda: 'enlightenment',
    'gnome': gnome_shell_version,
    'gnome-shell': gnome_shell_version,
    'i3': lambda: 'i3',
    'mate': lambda: 'gnome3',
    'openbox': lambda: 'openbox',
    'pantheon': lambda: 'pantheon',
    'unity': lambda: 'unity',
    'xfce4': lambda: 'xfce4',
}
KNOWN_DE = '|'.join(VERSION_GUSSERS.keys())

def check_desktop_env():
    """
    Tries to guess the installed desktop environment. This is kind of heuristic and might (and most likely will)
    fail
    returns: Things like 'gnome2', 'gnome3', 'xfce4', 'openbox' or None
    """
    value = os.popen("ps -A | grep --extended-regexp --ignore-case '(%s)$'" % KNOWN_DE).read()
    if not value:
        value = os.popen("ps -A | grep --extended-regexp --ignore-case '%s'" % KNOWN_DE).read()
    counter = Counter(re.findall(r'(%s)' % KNOWN_DE, value))
    if counter:
        desktop_environment = counter.most_common(1)[0][0]
        return VERSION_GUSSERS.get(desktop_environment, lambda: None)()
    return None


def check_tumbler():
    return os.path.isfile('/etc/xdg/tumbler/tumbler.rc')


def add_thumbnailer_to_tumbler():
    with(open('/etc/xdg/tumbler/tumbler.rc', 'a+')) as tumbler_config:
        config_content = mmap.mmap(tumbler_config.fileno(), 0, access=mmap.ACCESS_READ)
        if config_content.find(b'# Ebook thumbnailer') == -1:
            tumbler_config.write('\n')
            tumbler_config.write('# Ebook thumbnailer\n')
            tumbler_config.write('[EbookThumbnailer]\n')
            tumbler_config.write('Disabled=false\n')
            tumbler_config.write('Priority=1\n')
            tumbler_config.write('Locations=\n')
            tumbler_config.write('MaxFileSize=0\n')
            print('Registered thumbnailer in tumbler.rc')
        else:
            print('Thumbnailer already present in tumbler.rc')


def install():
    check_dependencies()
    if not os.access(install_dir, os.W_OK):
        print('You do not have write permissions to %s (maybe you need to sudo)' % install_dir)
        sys.exit(1)

    print('Installing epub-thumbnailer to %s ...' % install_dir)
    if copy(os.path.join(source_dir, 'epub-thumbnailer.py'), os.path.join(install_dir, 'epub-thumbnailer')):
        print('OK')
        environment = check_desktop_env()

        if environment == 'gnome2':
            schema = os.path.join(source_dir, 'epub-thumbnailer.schemas')
            os.popen('GCONF_CONFIG_SOURCE=$(gconftool-2 --get-default-source) '
                         'gconftool-2 --makefile-install-rule "%s" 2>/dev/null' %
                            schema)
            print('\nRegistered epub archive thumbnailer in gconf (if available).')
            print('The thumbnailer is only supported by some file managers, such as Nautilus, Caja and Thunar')
            print('You might have to restart the file manager for the thumbnailer to be activated.\n')
        elif environment is not None:
            print('Installing thumbnailer hook in /usr/share/thumbnailers ...')
            if copy(os.path.join(source_dir, 'epub.thumbnailer'), '/usr/share/thumbnailers/epub.thumbnailer'):
                print('OK')
            else:
                print('Could not install')
                exit(1)
        else:
            print('\nCould not determine your desktop environment version. You can still use the thumbnailer script manually.')
            print('')
            print('For example:')
            print('')
            print('    epub-thumbnailer Lawrence\ Lessig\ -\ Free\ Culture.epub cover.png 128')
            print('')
            print('Also, please consider reporting this in https://github.com/marianosimone/epub-thumbnailer/')
            print('including information about your distribution, window manager, file manager and the result of `ps -A`')
            exit(1)
    else:
        print('Could not install')
        exit(1)

    uses_tumbler = check_tumbler()
    if uses_tumbler:
        add_thumbnailer_to_tumbler()

    print('You might have to restart your file manager for the thumbnailer to be activated.\n')

def uninstall():
    print('Uninstalling epub-thumbnailer from', install_dir, '...')
    environment = check_desktop_env()
    os.remove(os.path.join(install_dir, 'epub-thumbnailer'))
    if environment in ('gnome3', 'xfce4', 'unity'):
        print('Uninstalling epub.thumbnailer from /usr/share/thumbnailers/ ...')
        try:
            os.remove('/usr/share/thumbnailers/epub.thumbnailer')
            print('OK')
        except:
            print('Could not remove /usr/share/thumbnailers/epub.thumbnailer')


commands = {
    'install': install,
    'uninstall': uninstall
}

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Installs or uninstall epub-thumbnailer on your system')
    parser.add_argument('action', metavar='action', choices=['install', 'uninstall'], help='the action to perform')
    args = parser.parse_args()
    commands[args.action]()