openmedialibrary/install
2019-01-20 16:46:58 +05:30

199 lines
7.3 KiB
Python
Executable file

#!/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'])