summaryrefslogtreecommitdiff
blob: e9390ec7c2d70928a90900e7f95820287d398c5d (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
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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
# Copyright 1999-2016 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
# $Id: $

DESCRIPTION="Manage Python interpreter preferences"
MAINTAINER="python@gentoo.org"
VERSION=@VERSION@

CONFIG_PATH="${EROOT%/}/etc/python-exec/python-exec.conf"
ENV_D_PATH="${EROOT%/}/etc/env.d"
INTERPRETER_DIR="${EROOT%/}/usr/bin"
MAN_PATH="${EROOT%/}/usr/share/man/man1"

# Get list of all installed Python interpreters, in lexical order.
# $1 can be --pyN to filter results to pythonN.?.
get_installed_pythons() {
	local exes=(
		# note: order *matters*
		# TODO: get them outta python-exec
		"${INTERPRETER_DIR}"/jython?.?@EXEEXT@
		"${INTERPRETER_DIR}"/pypy{,3}@EXEEXT@
		"${INTERPRETER_DIR}"/python?.?@EXEEXT@
	)

	local i
	for (( i = ${#exes[@]}-1; i >= 0; --i )); do
		local exe=${exes[i]}
		[[ -x ${exe} ]] || continue
		exe=${exe##*/}
		exe=${exe%@EXEEXT@}
		# apply filters
		[[ ${1} == --py* && ${exe} != python${1:4}* ]] && continue

		echo "${exe}"
	done
}

# Get list of all preference values from python-exec.conf. This
# includes both preferred implementations (in preference order)
# and disabled interpreters.
get_all_preferences() {
	[[ -e ${CONFIG_PATH} ]] || return

	local l
	while read l; do
		# skip comments
		[[ ${l} == '#'* ]] && continue

		# note: empty lines are stripped through word splitting
		echo "${l}"
	done <"${CONFIG_PATH}"
}

# Get list of preferred Python interpreters, from python-exec.conf,
# in preference order.
# $1 can be --pyN to filter results to pythonN.?.
get_preferred_pythons() {
	local i
	for i in $(get_all_preferences); do
		# skip negative entries
		[[ ${i} == -* ]] && continue
		# apply filters
		[[ ${1} == --py* && ${i} != python${1:4}* ]] && continue

		echo "${i}"
	done
}

# Get list of explicitly disabled Python interpreters, from
# python-exec.conf, in file order.
get_disabled_pythons() {
	local i
	for i in $(get_all_preferences); do
		# process only negative entries
		[[ ${i} == -* ]] || continue

		echo "${i#-}"
	done
}

# Get combined list of preferred, installed and disabled Python
# interpreters, in preference order.
# $1 can be --pyN to filter results to pythonN.?.
get_all_pythons() {
	local targets=( $(get_installed_pythons "${@}") )
	local preferred=( $(get_preferred_pythons "${@}") )
	local disabled=( $(get_disabled_pythons "${@}") )
	local i

	# preferred first
	for i in "${preferred[@]}"; do
		echo "${i}"
	done
	# active then
	for i in "${targets[@]}"; do
		has "${i}" "${preferred[@]}" && continue
		has "${i}" "${disabled[@]}" && continue
		echo "${i}"
	done
	# disabled last
	for i in "${targets[@]}"; do
		has "${i}" "${disabled[@]}" || continue
		echo "${i}"
	done
}

# Write new preference list. Preferences need to be passed
# as parameters (${@}).
write_preferences() {
	if [[ -e ${CONFIG_PATH} ]]; then
		sed -n -e '/^#/p' "${CONFIG_PATH}" > "${CONFIG_PATH}".new || die
	fi

	local IFS=$'\n'
	echo "${*}" >> "${CONFIG_PATH}".new || die

	mv "${CONFIG_PATH}".new "${CONFIG_PATH}" || die
}

# Set a man page symlink
set_man_symlink() {
	local target=${1} x suffix

	rm -f "${MAN_PATH}"/python.1{,.gz,.bz2,.lzma,.xz,.lz} || die

	for x in .1{,.gz,.bz2,.lzma,.xz,.lz}; do
		if [[ -e "${MAN_PATH}/${target}${x}" ]]; then
			suffix=${x}
			break
		fi
	done

	if [[ ! ${suffix} ]]; then
		echo "Couldn't find a man page for ${target}; skipping." 1>&2
		return 1
	fi

	ln -nfs "${target}${suffix}" "${MAN_PATH}/python${suffix}" || die
}

# Set OSX framework symlinks
set_osx_framework() {
	local target=${1}

	# Files of Mac OS X framework
	local framework_dir="${INTERPRETER_DIR%/bin}"/lib/Python.framework
	if [[ -d ${framework_dir} ]]; then
		local version=${target#python}
		pushd "${framework_dir}" >/dev/null || die
		rm -f Headers Python Resources || die
		ln -nfs "Versions/${version}/Headers" || die
		ln -nfs "Versions/${version}/Python" || die
		ln -nfs "Versions/${version}/Resources" || die
		popd >/dev/null || die
	fi
}

# Set the content of /etc/env.d/65python-docs
set_python_docs() {
	local path target=${1#python} variable
	rm -f "${ENV_D_PATH}/65python-docs" || die
	if [[ -f ${ENV_D_PATH}/60python-docs-${target} ]]; then
		variable="PYTHONDOCS_${target//./_}"
		path="$(. "${ENV_D_PATH}/60python-docs-${target}"; echo "${!variable}")"
		if [[ -d ${path} ]]; then
			echo "PYTHONDOCS=\"${path}\"" > "${ENV_D_PATH}/65python-docs"
		fi
	fi
}

# Perform all necessary updates following preference list change.
post_update() {
	local main_cpython=$(do_show --cpython)

	# TODO: update only when necessary

	set_man_symlink "${main_cpython}"
	set_osx_framework "${main_cpython}"
	set_python_docs "${main_cpython}"
}

### edit action ###

describe_edit() {
	echo "Edit the interpreter preference list"
}

do_edit() {
	[[ ${EDITOR} ]] || die "EDITOR is not set"

	${EDITOR} "${CONFIG_PATH}"

	post_update
}

### show action ###

describe_show() {
	echo "Show the most preferred Python interpreter"
}

describe_show_options() {
	echo "--ABI         : use PYTHON_ABI variable format (deprecated)"
	echo "--cpython     : show the preferred version of CPython"
	echo "--pref-only   : consider only explicitly preferred impls"
	echo "--python2     : show the preferred version of CPython 2"
	echo "--python3     : show the preferred version of CPython 3"
}

do_show() {
	local abi filter interpreter pref_only
	while [[ ${#} -gt 0 ]]; do
		case ${1} in
			--ABI)
				abi=1
				;;
			--cpython|--py)
				filter=--py
				;;
			--pref-only)
				pref_only=1
				;;
			--python2|--py2)
				filter=--py2
				;;
			--python3|--py3)
				filter=--py3
				;;
			*)
				die -q "Unrecognized argument '$1'"
				;;
		esac
		shift
	done

	local preferred=( $(get_preferred_pythons ${filter}) )
	local installed=( $(get_installed_pythons ${filter}) )

	local all=( "${preferred[@]}" )
	# preferred are preferred, but fall back to anything
	[[ ${pref_only} ]] || all+=( "${installed[@]}" )

	local i
	for i in "${all[@]}"; do
		# skip if not installed
		has "${i}" "${installed[@]}" || continue
		interpreter=${i}
		break
	done

	if [[ ${abi} ]]; then
		echo "${interpreter#python}"
	else
		echo "${interpreter}"
	fi
}

### list action ###

describe_list() {
	echo "List installed Python interpreters"
}

describe_list_options() {
	echo "--cpython     : list only CPython interpreters"
	echo "--python2     : list only CPython 2 interpreters"
	echo "--python3     : list only CPython 3 interpreters"
}

do_list() {
	local filter
	while [[ ${#} -gt 0 ]]; do
		case ${1} in
			--cpython|--py)
				filter=--py
				;;
			--python2|--py2)
				filter=--py2
				;;
			--python3|--py3)
				filter=--py3
				;;
			*)
				die -q "Unrecognized argument '$1'"
				;;
		esac
		shift
	done

	local all=( $(get_all_pythons ${filter}) )
	local preferred=( $(get_preferred_pythons ${filter}) )
	local disabled=( $(get_disabled_pythons) )

	write_list_start "Available Python${filter+ ${filter#--py}} interpreters, in order of preference:"

	for (( i = 0; i < ${#all[@]}; ++i )); do
		if has "${all[i]}" "${disabled[@]}"; then
			all[i]=$(highlight_marker "${all[i]}" "$(highlight_warning "(disabled)")")
		elif ! has "${all[i]}" "${preferred[@]}"; then
			all[i]=$(highlight_marker "${all[i]}" "(fallback)")
		fi
	done
	write_numbered_list -m "(none found)" "${all[@]}"
}

### set action ###

describe_set() {
	echo "Set the preferred Python interpreter"
}

describe_set_options() {
	echo "--cpython     : update preference for CPython versions only"
	echo "--python2     : update preference for CPython 2 versions only"
	echo "--python3     : update preference for CPython 3 versions only"
}

describe_set_parameters() {
	echo "<target>"
}

do_set() {
	local filter
	while [[ ${#} -gt 0 ]]; do
		case ${1} in
			--cpython|--py)
				filter=--py
				;;
			--python2|--py2)
				filter=--py2
				;;
			--python3|--py3)
				filter=--py3
				;;
			*)
				break
				;;
		esac
		shift
	done

	[[ ${#} -eq 1 ]] || die "Usage: eselect python set <interpreter>"

	local target=${1}
	local targets=( $(get_all_pythons ${filter}) )
	if is_number "${target}" \
		&& [[ ${target} -ge 1 && ${target} -le ${#targets[@]} ]]
	then
		target=${targets[target-1]}
	fi

	has "${target}" "${targets[@]}" || die "Invalid target: ${target}"

	local prefs=( $(get_all_preferences) )

	local i target_idx
	for (( i = 0; i < ${#prefs[@]}; ++i )); do
		# find first positive preference matching the filter
		if [[ ! ${target_idx} ]]; then
			if [[ ( ${filter} == --py? && ${prefs[i]} == python${filter:4}* ) \
				|| ( ! ${filter} && ${prefs[i]} != -* ) ]]
			then
				target_idx=${i}
			fi
		fi

		# remove all duplicate positive and negative entries for target
		[[ ${prefs[i]#-} == ${target} ]] && unset 'prefs[i]'
	done
	# if none matched, add to the bottom
	: "${target_idx=${#prefs[@]}}"

	# add between remaining preferences, before the one matching
	# need to do this outta loop in case no pref matches
	prefs=( "${prefs[@]:0:target_idx}" "${target}" "${prefs[@]:target_idx}" )

	umask 022
	write_preferences "${prefs[@]}"
	post_update
}

### update action ###

describe_update() {
	echo "Switch to the most recent Python interpreter"
}

describe_update_options() {
	echo "--if-unset    : do not alter preferences unless there is no valid preference set"
	echo "--ignore SLOT : ignore specified Python slots"
	echo "--cpython     : update only CPython preferences"
	echo "--python2     : update only CPython 2 preferences (ignored)"
	echo "--python3     : update only CPython 3 preferences"
}

do_update() {
	local if_unset ignored_slots=() filter
	while [[ ${#} -gt 0 ]]; do
		case ${1} in
			--if-unset)
				if_unset=1
				;;
			--ignore)
				ignored_slots+=( "${2}" )
				shift
				;;
			--cpython|--py)
				filter=--py
				;;
			--python2|--py2)
				filter=--py2
				echo "Ignoring Python 2 preference update as non-meaningful"
				return 0
				;;
			--python3|--py3)
				filter=--py3
				;;
			*)
				die -q "Unrecognized argument '$1'"
				;;
		esac
		shift
	done

	if [[ ${if_unset} ]]; then
		local current=$(do_show ${filter} --pref-only)

		[[ ${current} ]] && return
	fi

	local targets=( $(get_installed_pythons ${filter}) )

	# Ignore slots
	local slot
	for slot in ${ignored_slots[@]}; do
		targets=( ${targets[@]/python${slot}/} )
	done

	if [[ ${targets[@]} ]]; then
		local target=${targets[0]}
		echo "Switching to ${target}"
		do_set ${filter} "${target}"
	else
		die -q "No Python interpreter available"
	fi
}

# vim: set ft=eselect :