1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
#!/usr/bin/env python3
# Copyright (C) 2013 Vinay Sajip. New BSD License.
# Copyright (C) 2014, 2021 kaliko <kaliko@azylum.org>
#
REQ_VER = (3,6)
import sys
if sys.version_info < REQ_VER:
print('Need at least python {0}.{1} to run this script'.format(*REQ_VER), file=sys.stderr)
sys.exit(1)
import os
import os.path
import venv
from subprocess import Popen, PIPE
from threading import Thread
from urllib.parse import urlparse
from urllib.request import urlretrieve
from shutil import rmtree
class ExtendedEnvBuilder(venv.EnvBuilder):
"""
This builder installs setuptools and pip so that you can pip or
easy_install other packages into the created environment.
"""
def __init__(self, *args, **kwargs):
self.verbose = kwargs.pop('verbose', False)
super().__init__(*args, **kwargs)
def post_setup(self, context):
"""
Set up any packages which need to be pre-installed into the
environment being created.
:param context: The information for the environment creation request
being processed.
"""
os.environ['VIRTUAL_ENV'] = context.env_dir
setup = os.path.abspath(os.path.join(context.env_dir, '../setup.py'))
self.install_script(context, 'sima', setup=setup)
def reader(self, stream, context):
"""
Read lines from a subprocess' output stream and write progress
information to sys.stderr.
"""
while True:
s = stream.readline()
if not s:
break
if not self.verbose:
sys.stderr.write('.')
else:
sys.stderr.write(s.decode('utf-8'))
sys.stderr.flush()
stream.close()
def install_script(self, context, name, url=None, setup=None):
if url:
binpath = context.bin_path
_, _, path, _, _, _ = urlparse(url)
fn = os.path.split(path)[-1]
distpath = os.path.join(binpath, fn)
# Download script into the env's binaries folder
urlretrieve(url, distpath)
if url:
args = [context.env_exe, fn]
else:
args = [context.env_exe, setup, 'install']
binpath = os.path.dirname(setup)
if self.verbose:
term = '\n'
else:
term = ''
sys.stderr.write('Installing %s ...%s' % (name, term))
sys.stderr.flush()
# Install in the env
p = Popen(args, stdout=PIPE, stderr=PIPE, cwd=binpath)
t1 = Thread(target=self.reader, args=(p.stdout, 'stdout'))
t1.start()
t2 = Thread(target=self.reader, args=(p.stderr, 'stderr'))
t2.start()
p.wait()
t1.join()
t2.join()
sys.stderr.write('done.\n')
if url:
# Clean up - no longer needed
os.unlink(distpath)
def main():
root = os.path.dirname(os.path.abspath(__file__))
vdir = os.path.join(root, 'venv')
builder = ExtendedEnvBuilder(clear=True, verbose=False, with_pip=True)
builder.create(vdir)
# clean up
for residu in ['MPD_sima.egg-info', 'dist', 'build']:
if os.path.exists(os.path.join(root, residu)):
rmtree(os.path.join(root, residu))
# Write wrapper
with open(os.path.join(root, 'vmpd-sima'),'w') as fd:
fd.write('#!/bin/sh\n')
fd.write(f'. "{root}/venv/bin/activate"\n')
fd.write(f'"{root}/venv/bin/mpd-sima" "$@"')
os.chmod(os.path.join(root, 'vmpd-sima'), 0o744)
if __name__ == '__main__':
rc = 1
try:
main()
rc = 0
except ImportError as e:
print(f'Error: {e}')
sys.exit(rc)
|
