openmedialibrary/install
2016-01-12 20:30:54 +05:30

171 lines
6.2 KiB
Python
Executable file

#!/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] < '14.10':
try:
from OpenSSL.SSL import TLSv1_2_METHOD
except:
print("Open Media Library currently only works on Ubuntu 14.10 and later")
print("You can also install a current version of pyOpenSSL:\n\tsudo apt-get install python3-pip python3-dev libffi-dev\n\tsudo pip3 install 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'
version = '4.0'
distro = platform.linux_distribution()
if distro[0] == 'Ubuntu' and distro[1] <= '14.04':
version = '3.0'
try:
import gi
gi.require_version('WebKit2', version)
from gi.repository import WebKit2, Gtk
except:
apt_packages += ' gir1.2-webkit2-' + version
dnf_packages += ' webkitgtk4'
try:
import OpenSSL
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'
if has_bin('apt-get') and apt_packages:
print('You need additional packages: %s' % apt_packages)
print('run: sudo apt-get install -y ' + 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 -y ' + 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 those 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.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:
subprocess.call([os.path.join(target, 'ctl'), 'open'])