/[gentoo-x86]/eclass/python.eclass
Gentoo

Diff of /eclass/python.eclass

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

Revision 1.54 Revision 1.68
1# Copyright 1999-2008 Gentoo Foundation 1# Copyright 1999-2009 Gentoo Foundation
2# Distributed under the terms of the GNU General Public License v2 2# Distributed under the terms of the GNU General Public License v2
3# $Header: /var/cvsroot/gentoo-x86/eclass/python.eclass,v 1.54 2008/10/30 05:21:46 zmedico Exp $ 3# $Header: /var/cvsroot/gentoo-x86/eclass/python.eclass,v 1.68 2009/08/31 00:07:37 arfrever Exp $
4 4
5# @ECLASS: python.eclass 5# @ECLASS: python.eclass
6# @MAINTAINER: 6# @MAINTAINER:
7# python@gentoo.org 7# python@gentoo.org
8# 8#
9# original author: Alastair Tse <liquidx@gentoo.org> 9# original author: Alastair Tse <liquidx@gentoo.org>
10# @BLURB: A Utility Eclass that should be inherited by anything that deals with Python or Python modules. 10# @BLURB: A Utility Eclass that should be inherited by anything that deals with Python or Python modules.
11# @DESCRIPTION: 11# @DESCRIPTION:
12# Some useful functions for dealing with python. 12# Some useful functions for dealing with Python.
13inherit alternatives multilib
14 13
14inherit multilib
15 15
16if [[ -n "${NEED_PYTHON}" ]] ; then 16if [[ -n "${NEED_PYTHON}" ]] ; then
17 DEPEND=">=dev-lang/python-${NEED_PYTHON}" 17 PYTHON_ATOM=">=dev-lang/python-${NEED_PYTHON}"
18 DEPEND="${PYTHON_ATOM}"
18 RDEPEND="${DEPEND}" 19 RDEPEND="${DEPEND}"
20else
21 PYTHON_ATOM="dev-lang/python"
22fi
23
24DEPEND="${DEPEND} >=app-shells/bash-3.2"
25if ! has "${EAPI:-0}" 0 1 2 || [[ -n "${SUPPORT_PYTHON_ABIS}" ]]; then
26 DEPEND="${DEPEND} >=app-admin/eselect-python-20090804"
19fi 27fi
20 28
21__python_eclass_test() { 29__python_eclass_test() {
22 __python_version_extract 2.3 30 __python_version_extract 2.3
23 echo -n "2.3 -> PYVER: $PYVER PYVER_MAJOR: $PYVER_MAJOR" 31 echo -n "2.3 -> PYVER: $PYVER PYVER_MAJOR: $PYVER_MAJOR"
57 tmpstr="$(${python} -V 2>&1 )" 65 tmpstr="$(${python} -V 2>&1 )"
58 export PYVER_ALL="${tmpstr#Python }" 66 export PYVER_ALL="${tmpstr#Python }"
59 __python_version_extract $PYVER_ALL 67 __python_version_extract $PYVER_ALL
60} 68}
61 69
70# @FUNCTION: PYTHON
71# @USAGE: [-a|--absolute-path] <Python_ABI="${PYTHON_ABI}">
72# @DESCRIPTION:
73# Get Python interpreter filename for specified Python ABI. If Python_ABI argument
74# is ommitted, then PYTHON_ABI environment variable must be set and is used.
75PYTHON() {
76 local absolute_path="0" slot=
77
78 while (($#)); do
79 case "$1" in
80 -a|--absolute-path)
81 absolute_path="1"
82 ;;
83 -*)
84 die "${FUNCNAME}(): Unrecognized option $1"
85 ;;
86 *)
87 break
88 ;;
89 esac
90 shift
91 done
92
93 if [[ "$#" -eq "0" ]]; then
94 if [[ -n "${PYTHON_ABI}" ]]; then
95 slot="${PYTHON_ABI}"
96 else
97 die "${FUNCNAME}(): Invalid usage"
98 fi
99 elif [[ "$#" -eq "1" ]]; then
100 slot="$1"
101 else
102 die "${FUNCNAME}(): Invalid usage"
103 fi
104
105 if [[ "${absolute_path}" == "1" ]]; then
106 echo -n "/usr/bin/python${slot}"
107 else
108 echo -n "python${slot}"
109 fi
110}
111
112# @FUNCTION: validate_PYTHON_ABIS
113# @DESCRIPTION:
114# Make sure PYTHON_ABIS variable has valid value.
115validate_PYTHON_ABIS() {
116 # Ensure that some functions cannot be accidentally successfully used in EAPI <= 2 without setting SUPPORT_PYTHON_ABIS variable.
117 if has "${EAPI:-0}" 0 1 2 && [[ -z "${SUPPORT_PYTHON_ABIS}" ]]; then
118 die "${FUNCNAME}() cannot be used in this EAPI without setting SUPPORT_PYTHON_ABIS variable"
119 fi
120
121 # Ensure that /usr/bin/python and /usr/bin/python-config are valid.
122 if [[ "$(readlink /usr/bin/python)" != "python-wrapper" ]]; then
123 die "'/usr/bin/python' isn't valid symlink"
124 fi
125 if [[ "$(</usr/bin/python-config)" != *"Gentoo python-config wrapper script"* ]]; then
126 die "'/usr/bin/python-config' isn't valid script"
127 fi
128
129 # USE_${ABI_TYPE^^} and RESTRICT_${ABI_TYPE^^}_ABIS variables hopefully will be included in EAPI >= 4.
130 if [[ -z "${PYTHON_ABIS}" ]] && has "${EAPI:-0}" 0 1 2 3; then
131 local ABI support_ABI supported_PYTHON_ABIS= restricted_ABI
132 PYTHON_ABI_SUPPORTED_VALUES="2.4 2.5 2.6 2.7 3.0 3.1 3.2"
133
134 if declare -p | grep "^declare -x USE_PYTHON=" > /dev/null; then
135 if [[ -z "${USE_PYTHON}" ]]; then
136 die "USE_PYTHON variable is empty"
137 fi
138
139 for ABI in ${USE_PYTHON}; do
140 if ! has "${ABI}" ${PYTHON_ABI_SUPPORTED_VALUES}; then
141 die "USE_PYTHON variable contains invalid value '${ABI}'"
142 fi
143 support_ABI="1"
144 for restricted_ABI in ${RESTRICT_PYTHON_ABIS}; do
145 if python -c "from fnmatch import fnmatch; exit(not fnmatch('${ABI}', '${restricted_ABI}'))"; then
146 support_ABI="0"
147 break
148 fi
149 done
150 [[ "${support_ABI}" == "1" ]] && supported_PYTHON_ABIS+=" ${ABI}"
151 done
152 export PYTHON_ABIS="${supported_PYTHON_ABIS# }"
153
154 if [[ -z "${PYTHON_ABIS//[${IFS}]/}" ]]; then
155 die "USE_PYTHON variable doesn't enable any Python version supported by ${CATEGORY}/${PF}"
156 fi
157 else
158 local restricted_ABI
159 python_version
160
161 for restricted_ABI in ${RESTRICT_PYTHON_ABIS}; do
162 if python -c "from fnmatch import fnmatch; exit(not fnmatch('${PYVER}', '${restricted_ABI}'))"; then
163 die "Active Python version isn't supported by ${CATEGORY}/${PF}"
164 fi
165 done
166 export PYTHON_ABIS="${PYVER}"
167 fi
168 fi
169
170 # Ensure that EPYTHON variable is respected.
171 local PYTHON_ABI
172 for PYTHON_ABI in ${PYTHON_ABIS}; do
173 if [[ "$(EPYTHON="$(PYTHON)" python -c 'from sys import version_info; print(".".join([str(x) for x in version_info[:2]]))')" != "${PYTHON_ABI}" ]]; then
174 die "'python' doesn't respect EPYTHON variable"
175 fi
176 done
177}
178
179# @FUNCTION: python_copy_sources
180# @USAGE: [directory]
181# @DESCRIPTION:
182# Copy unpacked sources of given package for each Python ABI.
183python_copy_sources() {
184 local dir dirs=() PYTHON_ABI
185
186 if [[ "$#" -eq "0" ]]; then
187 if [[ "${WORKDIR}" == "${S}" ]]; then
188 die "${FUNCNAME}() cannot be used"
189 fi
190 dirs="${S}"
191 else
192 dirs="$@"
193 fi
194
195 validate_PYTHON_ABIS
196 for PYTHON_ABI in ${PYTHON_ABIS}; do
197 for dir in "${dirs[@]}"; do
198 cp -lpr "${dir}" "${dir}-${PYTHON_ABI}" > /dev/null || die "Copying of sources failed"
199 done
200 done
201}
202
203# @FUNCTION: python_set_build_dir_symlink
204# @USAGE: [directory="build"]
205# @DESCRIPTION:
206# Create build directory symlink.
207python_set_build_dir_symlink() {
208 local dir="$1"
209
210 [[ -z "${PYTHON_ABI}" ]] && die "PYTHON_ABI variable not set"
211 [[ -z "${dir}" ]] && dir="build"
212
213 # Don't delete preexistent directories.
214 rm -f "${dir}" || die "Deletion of '${dir}' failed"
215 ln -s "${dir}-${PYTHON_ABI}" "${dir}" || die "Creation of '${dir}' directory symlink failed"
216}
217
218# @FUNCTION: python_execute_function
219# @USAGE: [--action-message message] [-d|--default-function] [--failure-message message] [--nonfatal] [-q|--quiet] [-s|--separate-build-dirs] <function> [arguments]
220# @DESCRIPTION:
221# Execute specified function for each value of PYTHON_ABIS, optionally passing additional
222# arguments. The specified function can use PYTHON_ABI and BUILDDIR variables.
223python_execute_function() {
224 local action action_message action_message_template= default_function="0" failure_message failure_message_template= function nonfatal="0" PYTHON_ABI quiet="0" separate_build_dirs="0"
225
226 while (($#)); do
227 case "$1" in
228 --action-message)
229 action_message_template="$2"
230 shift
231 ;;
232 -d|--default-function)
233 default_function="1"
234 ;;
235 --failure-message)
236 failure_message_template="$2"
237 shift
238 ;;
239 --nonfatal)
240 nonfatal="1"
241 ;;
242 -q|--quiet)
243 quiet="1"
244 ;;
245 -s|--separate-build-dirs)
246 separate_build_dirs="1"
247 ;;
248 -*)
249 die "${FUNCNAME}(): Unrecognized option '$1'"
250 ;;
251 *)
252 break
253 ;;
254 esac
255 shift
256 done
257
258 if [[ "${default_function}" == "0" ]]; then
259 if [[ "$#" -eq "0" ]]; then
260 die "${FUNCNAME}(): Missing function name"
261 fi
262 function="$1"
263 shift
264
265 if [[ -z "$(type -t "${function}")" ]]; then
266 die "${FUNCNAME}(): '${function}' function isn't defined"
267 fi
268 else
269 if [[ "$#" -ne "0" ]]; then
270 die "${FUNCNAME}(): '--default-function' option and function name cannot be specified simultaneously"
271 fi
272 if has "${EAPI:-0}" 0 1; then
273 die "${FUNCNAME}(): '--default-function' option cannot be used in this EAPI"
274 fi
275
276 if [[ "${EBUILD_PHASE}" == "configure" ]]; then
277 if has "${EAPI}" 2; then
278 python_default_function() {
279 econf
280 }
281 else
282 python_default_function() {
283 nonfatal econf
284 }
285 fi
286 elif [[ "${EBUILD_PHASE}" == "compile" ]]; then
287 python_default_function() {
288 emake
289 }
290 elif [[ "${EBUILD_PHASE}" == "test" ]]; then
291 python_default_function() {
292 if emake -j1 -n check &> /dev/null; then
293 emake -j1 check
294 elif emake -j1 -n test &> /dev/null; then
295 emake -j1 test
296 fi
297 }
298 elif [[ "${EBUILD_PHASE}" == "install" ]]; then
299 python_default_function() {
300 emake DESTDIR="${D}" install
301 }
302 else
303 die "${FUNCNAME}(): --default-function option cannot be used in this ebuild phase"
304 fi
305 function="python_default_function"
306 fi
307
308 if [[ "${quiet}" == "0" ]]; then
309 [[ "${EBUILD_PHASE}" == "setup" ]] && action="Setting up"
310 [[ "${EBUILD_PHASE}" == "unpack" ]] && action="Unpacking"
311 [[ "${EBUILD_PHASE}" == "prepare" ]] && action="Preparation"
312 [[ "${EBUILD_PHASE}" == "configure" ]] && action="Configuration"
313 [[ "${EBUILD_PHASE}" == "compile" ]] && action="Building"
314 [[ "${EBUILD_PHASE}" == "test" ]] && action="Testing"
315 [[ "${EBUILD_PHASE}" == "install" ]] && action="Installation"
316 [[ "${EBUILD_PHASE}" == "preinst" ]] && action="Preinstallation"
317 [[ "${EBUILD_PHASE}" == "postinst" ]] && action="Postinstallation"
318 [[ "${EBUILD_PHASE}" == "prerm" ]] && action="Preuninstallation"
319 [[ "${EBUILD_PHASE}" == "postrm" ]] && action="Postuninstallation"
320 fi
321
322 local RED GREEN BLUE NORMAL
323 if [[ "${NOCOLOR:-false}" =~ ^(false|no)$ ]]; then
324 RED=$'\e[1;31m'
325 GREEN=$'\e[1;32m'
326 BLUE=$'\e[1;34m'
327 NORMAL=$'\e[0m'
328 else
329 RED=
330 GREEN=
331 BLUE=
332 NORMAL=
333 fi
334
335 validate_PYTHON_ABIS
336 for PYTHON_ABI in ${PYTHON_ABIS}; do
337 if [[ "${quiet}" == "0" ]]; then
338 if [[ -n "${action_message_template}" ]]; then
339 action_message="$(eval echo -n "${action_message_template}")"
340 else
341 action_message="${action} of ${CATEGORY}/${PF} with Python ${PYTHON_ABI}..."
342 fi
343 echo " ${GREEN}*${NORMAL} ${BLUE}${action_message}${NORMAL}"
344 fi
345
346 if [[ "${separate_build_dirs}" == "1" ]]; then
347 export BUILDDIR="${S}-${PYTHON_ABI}"
348 pushd "${BUILDDIR}" > /dev/null || die "pushd failed"
349 else
350 export BUILDDIR="${S}"
351 fi
352
353 if ! has "${EAPI}" 0 1 2 && has "${PYTHON_ABI}" ${FAILURE_TOLERANT_PYTHON_ABIS}; then
354 EPYTHON="$(PYTHON)" nonfatal "${function}" "$@"
355 else
356 EPYTHON="$(PYTHON)" "${function}" "$@"
357 fi
358
359 if [[ "$?" != "0" ]]; then
360 if [[ -n "${failure_message_template}" ]]; then
361 failure_message="$(eval echo -n "${failure_message_template}")"
362 else
363 failure_message="${action} failed with Python ${PYTHON_ABI} in ${function}() function"
364 fi
365
366 if [[ "${nonfatal}" == "1" ]] || has "${PYTHON_ABI}" ${FAILURE_TOLERANT_PYTHON_ABIS}; then
367 local ABI enabled_PYTHON_ABIS
368 for ABI in ${PYTHON_ABIS}; do
369 [[ "${ABI}" != "${PYTHON_ABI}" ]] && enabled_PYTHON_ABIS+=" ${ABI}"
370 done
371 export PYTHON_ABIS="${enabled_PYTHON_ABIS# }"
372 if [[ "${quiet}" == "0" ]]; then
373 ewarn "${RED}${failure_message}${NORMAL}"
374 fi
375 else
376 die "${failure_message}"
377 fi
378 fi
379
380 if [[ "${separate_build_dirs}" == "1" ]]; then
381 popd > /dev/null || die "popd failed"
382 fi
383 unset BUILDDIR
384 done
385
386 if [[ "${default_function}" == "1" ]]; then
387 unset -f python_default_function
388 fi
389}
390
391
392# @ECLASS-VARIABLE: PYTHON_USE_WITH
393# @DESCRIPTION:
394# Set this to a space separated list of use flags
395# the python slot in use must be built with.
396
397# @ECLASS-VARIABLE: PYTHON_USE_WITH_OR
398# @DESCRIPTION:
399# Set this to a space separated list of use flags
400# of which one must be turned on for the slot of
401# in use.
402
403# @ECLASS-VARIABLE: PYTHON_USE_WITH_OPT
404# @DESCRIPTION:
405# Set this if you need to make either PYTHON_USE_WITH or
406# PYTHON_USE_WITH_OR atoms conditional under a use flag.
407
408# @FUNCTION: python_pkg_setup
409# @DESCRIPTION:
410# Makes sure PYTHON_USE_WITH or PYTHON_USE_WITH_OR listed use flags
411# are respected. Only exported if one of those variables is set.
412if ! has "${EAPI:-0}" 0 1 && [[ -n ${PYTHON_USE_WITH} || -n ${PYTHON_USE_WITH_OR} ]]; then
413 python_pkg_setup_fail() {
414 eerror "${1}"
415 die "${1}"
416 }
417
418 python_pkg_setup() {
419 [[ ${PYTHON_USE_WITH_OPT} ]] && use !${PYTHON_USE_WITH_OPT} && return
420
421 python_version
422 local failed
423 local pyatom="dev-lang/python:${PYVER}"
424
425 for use in ${PYTHON_USE_WITH}; do
426 if ! has_version "${pyatom}[${use}]"; then
427 python_pkg_setup_fail \
428 "Please rebuild ${pyatom} with use flags: ${PYTHON_USE_WITH}"
429 fi
430 done
431
432 for use in ${PYTHON_USE_WITH_OR}; do
433 if has_version "${pyatom}[${use}]"; then
434 return
435 fi
436 done
437
438 if [[ ${PYTHON_USE_WITH_OR} ]]; then
439 python_pkg_setup_fail \
440 "Please rebuild ${pyatom} with one of: ${PYTHON_USE_WITH_OR}"
441 fi
442 }
443
444 EXPORT_FUNCTIONS pkg_setup
445
446 if [[ ${PYTHON_USE_WITH} ]]; then
447 PYTHON_USE_WITH_ATOM="${PYTHON_ATOM}[${PYTHON_USE_WITH/ /,}]"
448 elif [[ ${PYTHON_USE_WITH_OR} ]]; then
449 PYTHON_USE_WITH_ATOM="|| ( "
450 for use in ${PYTHON_USE_WITH_OR}; do
451 PYTHON_USE_WITH_ATOM="
452 ${PYTHON_USE_WITH_ATOM}
453 ${PYTHON_ATOM}[${use}]"
454 done
455 PYTHON_USE_WITH_ATOM="${PYTHON_USE_WITH_ATOM} )"
456 fi
457 if [[ ${PYTHON_USE_WITH_OPT} ]]; then
458 PYTHON_USE_WITH_ATOM="${PYTHON_USE_WITH_OPT}? ( ${PYTHON_USE_WITH_ATOM} )"
459 fi
460 DEPEND="${PYTHON_USE_WITH_ATOM}"
461 RDEPEND="${PYTHON_USE_WITH_ATOM}"
462fi
463
464# @ECLASS-VARIABLE: PYTHON_DEFINE_DEFAULT_FUNCTIONS
465# @DESCRIPTION:
466# Set this to define default functions for the following ebuild phases:
467# src_prepare, src_configure, src_compile, src_test, src_install.
468if ! has "${EAPI:-0}" 0 1 && [[ -n "${PYTHON_DEFINE_DEFAULT_FUNCTIONS}" ]]; then
469 python_src_prepare() {
470 python_copy_sources
471 }
472
473 for python_default_function in src_configure src_compile src_test src_install; do
474 eval "python_${python_default_function}() { python_execute_function -d -s; }"
475 done
476 unset python_default_function
477
478 EXPORT_FUNCTIONS src_prepare src_configure src_compile src_test src_install
479fi
480
62# @FUNCTION: python_disable_pyc 481# @FUNCTION: python_disable_pyc
63# @DESCRIPTION: 482# @DESCRIPTION:
64# Tells python not to automatically recompile modules to .pyc/.pyo 483# Tell Python not to automatically recompile modules to .pyc/.pyo
65# even if the timestamps/version stamps don't match. This is done 484# even if the timestamps/version stamps don't match. This is done
66# to protect sandbox. 485# to protect sandbox.
67#
68# note: supported by >=dev-lang/python-2.2.3-r3 only.
69#
70python_disable_pyc() { 486python_disable_pyc() {
71 export PYTHONDONTWRITEBYTECODE=1 # For 2.6 and above 487 export PYTHONDONTWRITEBYTECODE="1"
72 export PYTHON_DONTCOMPILE=1 # For 2.5 and below
73} 488}
74 489
75# @FUNCTION: python_enable_pyc 490# @FUNCTION: python_enable_pyc
76# @DESCRIPTION: 491# @DESCRIPTION:
77# Tells python to automatically recompile modules to .pyc/.pyo if the 492# Tell Python to automatically recompile modules to .pyc/.pyo if the
78# timestamps/version stamps change 493# timestamps/version stamps have changed.
79python_enable_pyc() { 494python_enable_pyc() {
80 unset PYTHONDONTWRITEBYTECODE 495 unset PYTHONDONTWRITEBYTECODE
81 unset PYTHON_DONTCOMPILE
82} 496}
83 497
84python_disable_pyc 498python_disable_pyc
85 499
86# @FUNCTION: python_need_rebuild 500# @FUNCTION: python_need_rebuild
89python_need_rebuild() { 503python_need_rebuild() {
90 python_version 504 python_version
91 export PYTHON_NEED_REBUILD=${PYVER} 505 export PYTHON_NEED_REBUILD=${PYVER}
92} 506}
93 507
508# @FUNCTION: python_get_includedir
509# @DESCRIPTION:
510# Run without arguments, returns the Python include directory.
511python_get_includedir() {
512 if [[ -n "${PYTHON_ABI}" ]]; then
513 echo "/usr/include/python${PYTHON_ABI}"
514 else
515 python_version
516 echo "/usr/include/python${PYVER}"
517 fi
518}
519
94# @FUNCTION: python_get_libdir 520# @FUNCTION: python_get_libdir
95# @DESCRIPTION: 521# @DESCRIPTION:
96# Run without arguments, returns the python library dir 522# Run without arguments, returns the Python library directory.
97python_get_libdir() { 523python_get_libdir() {
524 if [[ -n "${PYTHON_ABI}" ]]; then
525 echo "/usr/$(get_libdir)/python${PYTHON_ABI}"
526 else
98 python_version 527 python_version
99 echo "/usr/$(get_libdir)/python${PYVER}" 528 echo "/usr/$(get_libdir)/python${PYVER}"
529 fi
100} 530}
101 531
102# @FUNCTION: python_get_sitedir 532# @FUNCTION: python_get_sitedir
103# @DESCRIPTION: 533# @DESCRIPTION:
104# Run without arguments, returns the python site-packages dir 534# Run without arguments, returns the Python site-packages directory.
105python_get_sitedir() { 535python_get_sitedir() {
106 echo "$(python_get_libdir)/site-packages" 536 echo "$(python_get_libdir)/site-packages"
107}
108
109# @FUNCTION: python_makesym
110# @DESCRIPTION:
111# Run without arguments, it will create the /usr/bin/python symlinks
112# to the latest installed version
113python_makesym() {
114 alternatives_auto_makesym "/usr/bin/python" "python[0-9].[0-9]"
115 alternatives_auto_makesym "/usr/bin/python2" "python2.[0-9]"
116} 537}
117 538
118# @FUNCTION: python_tkinter_exists 539# @FUNCTION: python_tkinter_exists
119# @DESCRIPTION: 540# @DESCRIPTION:
120# Run without arguments, checks if python was compiled with Tkinter 541# Run without arguments, checks if python was compiled with Tkinter
128 die "missing tkinter support with installed python" 549 die "missing tkinter support with installed python"
129 fi 550 fi
130} 551}
131 552
132# @FUNCTION: python_mod_exists 553# @FUNCTION: python_mod_exists
133# @USAGE: < module > 554# @USAGE: <module>
134# @DESCRIPTION: 555# @DESCRIPTION:
135# Run with the module name as an argument. it will check if a 556# Run with the module name as an argument. it will check if a
136# python module is installed and loadable. it will return 557# python module is installed and loadable. it will return
137# TRUE(0) if the module exists, and FALSE(1) if the module does 558# TRUE(0) if the module exists, and FALSE(1) if the module does
138# not exist. 559# not exist.
141# if python_mod_exists gtk; then 562# if python_mod_exists gtk; then
142# echo "gtk support enabled" 563# echo "gtk support enabled"
143# fi 564# fi
144python_mod_exists() { 565python_mod_exists() {
145 [[ "$1" ]] || die "${FUNCNAME} requires an argument!" 566 [[ "$1" ]] || die "${FUNCNAME} requires an argument!"
146 python -c "import $1" >/dev/null 2>&1 567 python -c "import $1" &>/dev/null
147} 568}
148 569
149# @FUNCTION: python_mod_compile 570# @FUNCTION: python_mod_compile
150# @USAGE: < file > [more files ...] 571# @USAGE: <file> [more files ...]
151# @DESCRIPTION: 572# @DESCRIPTION:
152# Given filenames, it will pre-compile the module's .pyc and .pyo. 573# Given filenames, it will pre-compile the module's .pyc and .pyo.
153# This function should only be run in pkg_postinst() 574# This function should only be run in pkg_postinst()
154# 575#
155# Example: 576# Example:
156# python_mod_compile /usr/lib/python2.3/site-packages/pygoogle.py 577# python_mod_compile /usr/lib/python2.3/site-packages/pygoogle.py
157# 578#
158python_mod_compile() { 579python_mod_compile() {
580 if ! has "${EAPI:-0}" 0 1 2 || [[ -n "${SUPPORT_PYTHON_ABIS}" ]]; then
581 die "${FUNCNAME}() cannot be used in this EAPI"
582 fi
583
159 local f myroot myfiles=() 584 local f myroot myfiles=()
160 585
161 # Check if phase is pkg_postinst() 586 # Check if phase is pkg_postinst()
162 [[ ${EBUILD_PHASE} != postinst ]] &&\ 587 [[ ${EBUILD_PHASE} != postinst ]] &&\
163 die "${FUNCNAME} should only be run in pkg_postinst()" 588 die "${FUNCNAME} should only be run in pkg_postinst()"
177 [[ -f "${myroot}/${f}" ]] && myfiles+=("${myroot}/${f}") 602 [[ -f "${myroot}/${f}" ]] && myfiles+=("${myroot}/${f}")
178 done 603 done
179 604
180 if ((${#myfiles[@]})); then 605 if ((${#myfiles[@]})); then
181 python${PYVER} ${myroot}/usr/$(get_libdir)/python${PYVER}/py_compile.py "${myfiles[@]}" 606 python${PYVER} ${myroot}/usr/$(get_libdir)/python${PYVER}/py_compile.py "${myfiles[@]}"
182 python${PYVER} -O ${myroot}/usr/$(get_libdir)/python${PYVER}/py_compile.py "${myfiles[@]}" 607 python${PYVER} -O ${myroot}/usr/$(get_libdir)/python${PYVER}/py_compile.py "${myfiles[@]}" &> /dev/null
183 else 608 else
184 ewarn "No files to compile!" 609 ewarn "No files to compile!"
185 fi 610 fi
186} 611}
187 612
188# @FUNCTION: python_mod_optimize 613# @FUNCTION: python_mod_optimize
189# @USAGE: [ path ] 614# @USAGE: [options] [directory|file]
190# @DESCRIPTION: 615# @DESCRIPTION:
191# If no arguments supplied, it will recompile all modules under 616# If no arguments supplied, it will recompile not recursively all modules
192# sys.path (eg. /usr/lib/python2.3, /usr/lib/python2.3/site-packages/ ..) 617# under sys.path (eg. /usr/lib/python2.6, /usr/lib/python2.6/site-packages).
193# no recursively
194# 618#
195# If supplied with arguments, it will recompile all modules recursively 619# If supplied with arguments, it will recompile all modules recursively
196# in the supplied directory 620# in the supplied directory.
197# This function should only be run in pkg_postinst() 621# This function should only be run in pkg_postinst().
198# 622#
199# Options passed to this function are passed to compileall.py 623# Options passed to this function are passed to compileall.py.
200# 624#
201# Example: 625# Example:
202# python_mod_optimize /usr/share/codegen 626# python_mod_optimize ctypesgencore
203python_mod_optimize() { 627python_mod_optimize() {
204 local myroot mydirs=() myfiles=() myopts=()
205
206 # Check if phase is pkg_postinst() 628 # Check if phase is pkg_postinst().
207 [[ ${EBUILD_PHASE} != postinst ]] &&\ 629 [[ ${EBUILD_PHASE} != "postinst" ]] && die "${FUNCNAME} should only be run in pkg_postinst()"
208 die "${FUNCNAME} should only be run in pkg_postinst()"
209 630
631 if ! has "${EAPI:-0}" 0 1 2 || [[ -n "${SUPPORT_PYTHON_ABIS}" ]]; then
632 local dir file options=() other_dirs=() other_files=() PYTHON_ABI return_code root site_packages_absolute_dirs=() site_packages_dirs=() site_packages_absolute_files=() site_packages_files=()
633
634 # Strip trailing slash from ROOT.
635 root="${ROOT%/}"
636
637 # Respect ROOT and options passed to compileall.py.
638 while (($#)); do
639 case "$1" in
640 -l|-f|-q)
641 options+=("$1")
642 ;;
643 -d|-x)
644 options+=("$1" "$2")
645 shift
646 ;;
647 -*)
648 ewarn "${FUNCNAME}: Ignoring compile option $1"
649 ;;
650 *)
651 if [[ "$1" =~ ^/usr/lib(32|64)?/python[[:digit:]]+\.[[:digit:]]+ ]]; then
652 die "${FUNCNAME} doesn't support absolute paths of directories/files in site-packages directories"
653 elif [[ "$1" =~ ^/ ]]; then
654 if [[ -d "${root}/$1" ]]; then
655 other_dirs+=("${root}/$1")
656 elif [[ -f "${root}/$1" ]]; then
657 other_files+=("${root}/$1")
658 elif [[ -e "${root}/$1" ]]; then
659 ewarn "'${root}/$1' is not a file or a directory!"
660 else
661 ewarn "'${root}/$1' doesn't exist!"
662 fi
663 else
664 for PYTHON_ABI in ${PYTHON_ABIS}; do
665 if [[ -d "${root}$(python_get_sitedir)/$1" ]]; then
666 site_packages_dirs+=("$1")
667 break
668 elif [[ -f "${root}$(python_get_sitedir)/$1" ]]; then
669 site_packages_files+=("$1")
670 break
671 elif [[ -e "${root}$(python_get_sitedir)/$1" ]]; then
672 ewarn "'$1' is not a file or a directory!"
673 else
674 ewarn "'$1' doesn't exist!"
675 fi
676 done
677 fi
678 ;;
679 esac
680 shift
681 done
682
683 # Set additional options.
684 options+=("-q")
685
686 for PYTHON_ABI in ${PYTHON_ABIS}; do
687 if ((${#site_packages_dirs[@]})) || ((${#site_packages_files[@]})); then
688 return_code="0"
689 ebegin "Compilation and optimization of Python modules for Python ${PYTHON_ABI}"
690 if ((${#site_packages_dirs[@]})); then
691 for dir in "${site_packages_dirs[@]}"; do
692 site_packages_absolute_dirs+=("${root}$(python_get_sitedir)/${dir}")
693 done
694 "$(PYTHON)" "${root}$(python_get_libdir)/compileall.py" "${options[@]}" "${site_packages_absolute_dirs[@]}" || return_code="1"
695 "$(PYTHON)" -O "${root}$(python_get_libdir)/compileall.py" "${options[@]}" "${site_packages_absolute_dirs[@]}" &> /dev/null || return_code="1"
696 fi
697 if ((${#site_packages_files[@]})); then
698 for file in "${site_packages_files[@]}"; do
699 site_packages_absolute_files+=("${root}$(python_get_sitedir)/${file}")
700 done
701 "$(PYTHON)" "${root}$(python_get_libdir)/py_compile.py" "${site_packages_absolute_files[@]}" || return_code="1"
702 "$(PYTHON)" -O "${root}$(python_get_libdir)/py_compile.py" "${site_packages_absolute_files[@]}" &> /dev/null || return_code="1"
703 fi
704 eend "${return_code}"
705 fi
706 unset site_packages_absolute_dirs site_packages_absolute_files
707 done
708
709 # Don't use PYTHON_ABI in next calls to python_get_libdir().
710 unset PYTHON_ABI
711
712 if ((${#other_dirs[@]})) || ((${#other_files[@]})); then
713 return_code="0"
714 ebegin "Compilation and optimization of Python modules placed outside of site-packages directories for Python ${PYVER}..."
715 if ((${#other_dirs[@]})); then
716 python${PYVER} "${root}$(python_get_libdir)/compileall.py" "${options[@]}" "${other_dirs[@]}" || return_code="1"
717 python${PYVER} -O "${root}$(python_get_libdir)/compileall.py" "${options[@]}" "${other_dirs[@]}" &> /dev/null || return_code="1"
718 fi
719 if ((${#other_files[@]})); then
720 python${PYVER} "${root}$(python_get_libdir)/py_compile.py" "${other_files[@]}" || return_code="1"
721 python${PYVER} -O "${root}$(python_get_libdir)/py_compile.py" "${other_files[@]}" &> /dev/null || return_code="1"
722 fi
723 eend "${return_code}"
724 fi
725 else
726 local myroot mydirs=() myfiles=() myopts=() return_code="0"
727
210 # strip trailing slash 728 # strip trailing slash
211 myroot="${ROOT%/}" 729 myroot="${ROOT%/}"
212 730
213 # respect ROOT and options passed to compileall.py 731 # respect ROOT and options passed to compileall.py
214 while (($#)); do 732 while (($#)); do
215 case $1 in 733 case "$1" in
216 -l|-f|-q) 734 -l|-f|-q)
217 myopts+=("$1") 735 myopts+=("$1")
218 ;; 736 ;;
219 -d|-x) 737 -d|-x)
220 myopts+=("$1" "$2") 738 myopts+=("$1" "$2")
739 shift
740 ;;
741 -*)
742 ewarn "${FUNCNAME}: Ignoring compile option $1"
743 ;;
744 *)
745 if [[ -d "${myroot}"/$1 ]]; then
746 mydirs+=("${myroot}/$1")
747 elif [[ -f "${myroot}"/$1 ]]; then
748 # Files are passed to python_mod_compile which is ROOT-aware
749 myfiles+=("$1")
750 elif [[ -e "${myroot}/$1" ]]; then
751 ewarn "${myroot}/$1 is not a file or directory!"
752 else
753 ewarn "${myroot}/$1 doesn't exist!"
754 fi
755 ;;
756 esac
757 shift
758 done
759
760 # allow compiling for older python versions
761 if [ -n "${PYTHON_OVERRIDE_PYVER}" ]; then
762 PYVER=${PYTHON_OVERRIDE_PYVER}
763 else
764 python_version
765 fi
766
767 # set additional opts
768 myopts+=(-q)
769
770 ebegin "Byte compiling python modules for python-${PYVER} .."
771 if ((${#mydirs[@]})); then
772 python${PYVER} \
773 "${myroot}"/usr/$(get_libdir)/python${PYVER}/compileall.py \
774 "${myopts[@]}" "${mydirs[@]}" || return_code="1"
775 python${PYVER} -O \
776 "${myroot}"/usr/$(get_libdir)/python${PYVER}/compileall.py \
777 "${myopts[@]}" "${mydirs[@]}" &> /dev/null || return_code="1"
778 fi
779
780 if ((${#myfiles[@]})); then
781 python_mod_compile "${myfiles[@]}"
782 fi
783
784 eend "${return_code}"
785 fi
786}
787
788# @FUNCTION: python_mod_cleanup
789# @USAGE: [directory]
790# @DESCRIPTION:
791# Run with optional arguments, where arguments are directories of
792# python modules. If none given, it will look in /usr/lib/python[0-9].[0-9].
793#
794# It will recursively scan all compiled Python modules in the directories and
795# determine if they are orphaned (i.e. their corresponding .py files are missing.)
796# If they are, then it will remove their corresponding .pyc and .pyo files.
797#
798# This function should only be run in pkg_postrm().
799python_mod_cleanup() {
800 local PYTHON_ABI SEARCH_PATH=() root src_py
801
802 # Check if phase is pkg_postrm().
803 [[ ${EBUILD_PHASE} != "postrm" ]] && die "${FUNCNAME} should only be run in pkg_postrm()"
804
805 # Strip trailing slash from ROOT.
806 root="${ROOT%/}"
807
808 if (($#)); then
809 if ! has "${EAPI:-0}" 0 1 2 || [[ -n "${SUPPORT_PYTHON_ABIS}" ]]; then
810 while (($#)); do
811 if [[ "$1" =~ ^/usr/lib(32|64)?/python[[:digit:]]+\.[[:digit:]]+ ]]; then
812 die "${FUNCNAME} doesn't support absolute paths of directories/files in site-packages directories"
813 elif [[ "$1" =~ ^/ ]]; then
814 SEARCH_PATH+=("${root}/${1#/}")
815 else
816 for PYTHON_ABI in ${PYTHON_ABIS}; do
817 SEARCH_PATH+=("${root}$(python_get_sitedir)/$1")
818 done
819 fi
221 shift 820 shift
222 ;;
223 -*)
224 ewarn "${FUNCNAME}: Ignoring compile option $1"
225 ;;
226 *)
227 if [[ -d "${myroot}"/$1 ]]; then
228 mydirs+=("${myroot}/$1")
229 elif [[ -f "${myroot}"/$1 ]]; then
230 # Files are passed to python_mod_compile which is ROOT-aware
231 myfiles+=("$1")
232 elif [[ -e "${myroot}/$1" ]]; then
233 ewarn "${myroot}/$1 is not a file or directory!"
234 else
235 ewarn "${myroot}/$1 doesn't exist!"
236 fi
237 ;;
238 esac
239 shift
240 done 821 done
241
242 # allow compiling for older python versions
243 if [ -n "${PYTHON_OVERRIDE_PYVER}" ]; then
244 PYVER=${PYTHON_OVERRIDE_PYVER}
245 else 822 else
246 python_version
247 fi
248
249 # set additional opts
250 myopts+=(-q)
251
252 ebegin "Byte compiling python modules for python-${PYVER} .."
253 if ((${#mydirs[@]})); then
254 python${PYVER} \
255 "${myroot}"/usr/$(get_libdir)/python${PYVER}/compileall.py \
256 "${myopts[@]}" "${mydirs[@]}"
257 python${PYVER} -O \
258 "${myroot}"/usr/$(get_libdir)/python${PYVER}/compileall.py \
259 "${myopts[@]}" "${mydirs[@]}"
260 fi
261
262 if ((${#myfiles[@]})); then
263 python_mod_compile "${myfiles[@]}"
264 fi
265
266 eend $?
267}
268
269# @FUNCTION: python_mod_cleanup
270# @USAGE: [ dir ]
271# @DESCRIPTION:
272# Run with optional arguments, where arguments are directories of
273# python modules. if none given, it will look in /usr/lib/python[0-9].[0-9]
274#
275# It will recursively scan all compiled python modules in the directories
276# and determine if they are orphaned (eg. their corresponding .py is missing.)
277# if they are, then it will remove their corresponding .pyc and .pyo
278#
279# This function should only be run in pkg_postrm()
280python_mod_cleanup() {
281 local SEARCH_PATH=() myroot src_py
282
283 # Check if phase is pkg_postrm()
284 [[ ${EBUILD_PHASE} != postrm ]] &&\
285 die "${FUNCNAME} should only be run in pkg_postrm()"
286
287 # strip trailing slash
288 myroot="${ROOT%/}"
289
290 if (($#)); then
291 SEARCH_PATH=("${@#/}") 823 SEARCH_PATH=("${@#/}")
292 SEARCH_PATH=("${SEARCH_PATH[@]/#/$myroot/}") 824 SEARCH_PATH=("${SEARCH_PATH[@]/#/${root}/}")
825 fi
293 else 826 else
294 SEARCH_PATH=("${myroot}"/usr/lib*/python*/site-packages) 827 SEARCH_PATH=("${root}"/usr/lib*/python*/site-packages)
295 fi 828 fi
296 829
297 for path in "${SEARCH_PATH[@]}"; do 830 for path in "${SEARCH_PATH[@]}"; do
831 [[ ! -d "${path}" ]] && continue
298 einfo "Cleaning orphaned Python bytecode from ${path} .." 832 einfo "Cleaning orphaned Python bytecode from ${path} .."
299 find "${path}" -name '*.py[co]' -print0 | while read -rd ''; do 833 find "${path}" -name '*.py[co]' -print0 | while read -rd ''; do
300 src_py="${REPLY%[co]}" 834 src_py="${REPLY%[co]}"
301 [[ -f "${src_py}" ]] && continue 835 [[ -f "${src_py}" || (! -f "${src_py}c" && ! -f "${src_py}o") ]] && continue
302 einfo "Purging ${src_py}[co]" 836 einfo "Purging ${src_py}[co]"
303 rm -f "${src_py}"[co] 837 rm -f "${src_py}"[co]
304 done 838 done
305 839
306 # attempt to remove directories that maybe empty 840 # Attempt to remove directories that may be empty.
307 find "${path}" -type d | sort -r | while read -r dir; do 841 find "${path}" -type d | sort -r | while read -r dir; do
308 rmdir "${dir}" 2>/dev/null 842 rmdir "${dir}" 2>/dev/null && einfo "Removing empty directory ${dir}"
309 done 843 done
310 done 844 done
311} 845}

Legend:
Removed from v.1.54  
changed lines
  Added in v.1.68

  ViewVC Help
Powered by ViewVC 1.1.20