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
|
require "utils.rb"
module Language
module Python
def self.major_minor_version python
version = /\d\.\d/.match `#{python} --version 2>&1`
return unless version
Version.new(version.to_s)
end
def self.homebrew_site_packages(version="2.7")
HOMEBREW_PREFIX/"lib/python#{version}/site-packages"
end
def self.each_python build, &block
original_pythonpath = ENV["PYTHONPATH"]
["python", "python3"].each do |python|
next if build.without? python
version = self.major_minor_version python
ENV["PYTHONPATH"] = if Formulary.factory(python).installed?
nil
else
homebrew_site_packages(version)
end
block.call python, version if block
end
ENV["PYTHONPATH"] = original_pythonpath
end
def self.reads_brewed_pth_files? python
version = major_minor_version python
return unless homebrew_site_packages(version).directory?
probe_file = homebrew_site_packages(version)/"homebrew-pth-probe.pth"
probe_file.atomic_write("import site; site.homebrew_was_here = True")
result = quiet_system python, "-c", "import site; assert(site.homebrew_was_here)"
probe_file.unlink
result
end
def self.user_site_packages python
Pathname.new(`#{python} -c "import site; print(site.getusersitepackages())"`.chomp)
end
def self.in_sys_path? python, path
script = <<-EOS.undent
import os, sys
[os.path.realpath(p) for p in sys.path].index(os.path.realpath("#{path}"))
EOS
quiet_system python, "-c", script
end
def self.setup_install python, prefix, *args
# force-import setuptools, which monkey-patches distutils, to make
# sure that we always call a setuptools setup.py. trick borrowed from pip:
# https://github.com/pypa/pip/blob/043af83/pip/req/req_install.py#L743-L780
shim = <<-EOS.undent
import setuptools, tokenize
__file__ = 'setup.py'
exec(compile(getattr(tokenize, 'open', open)(__file__).read()
.replace('\\r\\n', '\\n'), __file__, 'exec'))
EOS
args += %w[--single-version-externally-managed --record=installed.txt]
args << "--prefix=#{prefix}"
system python, "-c", shim, "install", *args
end
end
end
|
