blob: 35cdfc47e0e7cf86bc77347b6d6151ad227b4e53 (
plain)
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
|
import os
import runpy
import shutil
import sys
import tempfile
import zipfile
from pathlib import PurePath
sys.argv[0] = __loader__.archive
# Set sys.executable to None. The real executable is available as
# sys.argv[0], and too many things assume sys.executable is a regular Python
# binary, which isn't available. By setting it to None we get clear errors
# when people try to use it.
sys.executable = None
# Extract the shared libraries from the zip file into a temporary directory.
# This works around the limitations of dynamic linker. Some Python libraries
# reference the .so files relatively and so extracting only the .so files
# does not work, so we extract the entire parent directory of the .so files to a
# tempdir and then add that to sys.path.
tempdir = None
with zipfile.ZipFile(__loader__.archive) as z:
# any root so files or root directories that contain so files will be
# extracted to the tempdir so the linker load them, this minimizes the
# number of files that need to be extracted to a tempdir
extract_paths = {}
for member in z.infolist():
if member.filename.endswith('.so'):
extract_paths[PurePath(member.filename).parts[0]] = member.filename
if extract_paths:
tempdir = tempfile.mkdtemp()
for member in z.infolist():
if not PurePath(member.filename).parts[0] in extract_paths.keys():
continue
if member.is_dir():
os.makedirs(os.path.join(tempdir, member.filename))
else:
z.extract(member, tempdir)
sys.path.insert(0, tempdir)
try:
runpy._run_module_as_main("ENTRY_POINT", alter_argv=False)
finally:
if tempdir is not None:
shutil.rmtree(tempdir)
|