#!/usr/bin/env python3 from __future__ import division, with_statement, print_function import distutils.spawn from contextlib import closing import json import os import sys import tarfile from urllib.request import urlopen from threading import Thread def makedirs(dirname): if not os.path.exists(dirname): os.makedirs(dirname) def get_platform(): name = sys.platform if name.startswith('darwin'): name = 'darwin64' elif name.startswith('linux'): import platform machine = platform.machine() if machine == 'armv7l': name = 'linux_armv7l' elif machine == 'aarch64': name = 'linux_aarch64' elif machine == 'x86_64': name = 'linux64' else: name = 'linux32' return name class Install(Thread): base_url = 'https://downloads.openmedialibrary.com/' status = {} failed = False def __init__(self, target, base_url=None): self.target = target if base_url: self.base_url = base_url 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:") print("\tsudo apt-get install python3-pip python3-dev libffi-dev libssl-dev libevent-2.0-5") print("\tsudo pip3 install -U pyOpenSSL") self.failed = True return apt_packages = '' dnf_packages = '' try: from PIL import Image import lxml except: apt_packages += ' python3 python3-pil python3-lxml' dnf_packages += ' python3-pillow python3-lxml' try: import gi gi.require_version('Gtk', '3.0') from gi.repository import Gtk, GLib except: apt_packages += ' gir1.2-gtk-3.0' #dnf_packages += ' webkitgtk4' 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-crypto' 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'): if 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'): if 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.abspath(target)) makedirs(target) os.chdir(target) release = self.get_release() self.status['release'] = release print("Installing Open Media Library:") platform = get_platform() for module in sorted(release['modules']): if release['modules'][module].get('platform', platform) == platform: 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(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' makedirs('updates') os.system('./ctl setup') release_path = 'data/release.json' makedirs('data') with open(release_path, 'w') as fd: json.dump(release, fd, indent=2) if sys.platform.startswith('linux'): os.system('./ctl install_launcher') self.status['progress'] = 1 self.status['done'] = True def download(self, url, filename): makedirs(os.path.dirname(filename)) print('downloading',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): path = distutils.spawn.find_executable(cmd) return path != None 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'): data_home = os.path.expanduser(os.environ.get('XDG_DATA_HOME', '~/.local/share')) target = os.path.join(data_home, '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("You can launch Open Media Library from the menu or run: %s open"%os.path.join(target, 'ctl')) #subprocess.Popen([os.path.join(target, 'ctl'), 'open'])