openmedialibrary/install

#!/usr/bin/env python3
from __future__ import division, with_statement, print_function

from contextlib import closing
import json
import os
import sys
import tarfile
from urllib.request import urlopen
from threading import Thread
import subprocess

def makefolder(path):
    dirname = os.path.dirname(path)
    if not os.path.exists(dirname):
        os.makedirs(dirname)

class Install(Thread):

    base_url = 'http://downloads.openmedialibrary.com/'
    status = {}
    failed = False

    def __init__(self, target):
        self.target = target
        Thread.__init__(self)
        self.daemon = True
        self.start()

    def run(self):
        if sys.platform.startswith('linux'):
            import platform
            distro = platform.linux_distribution()
            if distro[0] == 'Ubuntu' and distro[1] < '15.10':
                try:
                    from OpenSSL.SSL import TLSv1_2_METHOD
                except:
                    print("To run Open Media Library on versions of Ubuntu before 15.10, you need to manually update pyOpenSSL.")
                    print("To install the current version of pyOpenSSL, run:\n\tsudo apt-get install python3-pip python3-dev libffi-dev libssl-dev\n\tsudo pip3 install -U pyOpenSSL")
                    self.failed = True
                    return
            apt_packages = ''
            dnf_packages = ''
            try:
                from PIL import Image
                import lxml
            except:
                apt_packages += ' python3.4 python3-pil python3-lxml'
                dnf_packages += ' python3-pillow python3-lxml'
            try:
                import OpenSSL
                from OpenSSL.SSL import TLSv1_2_METHOD
            except:
                apt_packages += ' python3-openssl'
                dnf_packages += ' python3-pyOpenSSL'
            try:
                import Crypto
            except:
                apt_packages += ' python3-crypto'
                dnf_packages += ' python3-pyCrypto'

            if not has_bin('pdftocairo'):
                apt_packages += ' poppler-utils'
                dnf_packages += ' poppler-utils'
            apt_packages = apt_packages.strip()
            dnf_packages = dnf_packages.strip()
            if has_bin('apt-get') and apt_packages:
                print('You need additional packages: %s' % apt_packages)
                print('run: sudo apt-get install ' + apt_packages)
                self.failed = True
                return
            elif has_bin('dnf') and dnf_packages:
                print('Installing additional packages: %s' % dnf_packages)
                print('run: sudo dnf install ' + dnf_packages)
                self.failed = True
                return
            elif apt_packages:
                print('You need to install Pillow, lxml and pyOpenSSL\ni.e. sudo pip3 install pillow lxml pyOpenSSL')
                if 'poppler' in apt_packages:
                    print('You need to install pdftocairo (part of poppler-utils).')
                print("After installing these packages, run install again.")
                self.failed = True
                return
        target = self.target
        target = os.path.normpath(os.path.join(os.path.abspath(target)))
        if not os.path.exists(target):
            os.makedirs(target)
        os.chdir(target)
        release = self.get_release()
        self.status['release'] = release
        for module in release['modules']:
            self.status['installing'] = module
            self.status['progress'] = 0
            self.status['size'] = 0
            package_tar = release['modules'][module]['name']
            url = self.url(package_tar)
            package_tar = os.path.join(self.target, package_tar)
            self.download(url, package_tar)
            tar = tarfile.open(package_tar)
            tar.extractall()
            tar.close()
            os.unlink(package_tar)
        os.symlink('openmedialibrary/ctl', 'ctl')
        self.status['progress'] = 0
        self.status['installing'] = 'setup'
        if sys.platform.startswith('linux'):
            os.system('./ctl install_launcher')
        with open('config/release.json', 'w') as fd:
            json.dump(release, fd, indent=2)
        os.makedirs('updates')
        os.system('./ctl setup')
        self.status['progress'] = 1
        self.status['done'] = True

    def download(self, url, filename):
        makefolder(filename)
        print(os.path.basename(filename))
        with open(filename, 'wb') as f:
            with closing(urlopen(url)) as u:
                size = int(u.headers.get('content-length', 0))
                self.status['size'] = size
                available = 0
                data = u.read(4096)
                while data:
                    if size:
                        available += len(data)
                        self.status['progress'] = available/size
                    f.write(data)
                    data = u.read(4096)

    def get_release(self):
        with closing(urlopen(self.url('release.json'))) as u:
            data = json.loads(u.read().decode('utf-8'))
        return data

    def url(self, url):
        return self.base_url + url

def has_bin(cmd):
    return subprocess.call(['which', cmd], stdout=subprocess.PIPE) == 0

if __name__ == '__main__':
    if len(sys.argv) == 1:
        if sys.platform == 'darwin':
            target = os.path.expanduser('~/Library/Application Support/Open Media Libary')
        elif sys.platform.startswith('linux'):
            target = os.path.expanduser('~/.local/share/openmedialibrary')
        else:
            target = os.path.join(os.curdir, 'openmedialibrary')
    elif len(sys.argv) != 2:
        print('usage: %s [target]' % sys.argv[0])
        sys.exit(1)
    else:
        target = sys.argv[1]
    if os.path.exists(target):
        print('%s exists, remove it before re-installing Open Media Library' % target)
        sys.exit(1)


    install = Install(target)
    install.join()
    if not install.failed:
        print("To launch Open Media Library, run: %s open"%os.path.join(target, 'ctl'))
        #subprocess.Popen([os.path.join(target, 'ctl'), 'open'])