From d536979fdc49e23c44ad2d5d16daad630c987a38 Mon Sep 17 00:00:00 2001 From: Zac Medico Date: Sat, 27 Mar 2010 18:49:52 -0700 Subject: Add support for probing shebangs and compiling python scripts that don't end with py. --- pym/portage/tests/lint/test_compile_modules.py | 35 +++++++++++++++++++++++--- 1 file changed, 31 insertions(+), 4 deletions(-) (limited to 'pym') diff --git a/pym/portage/tests/lint/test_compile_modules.py b/pym/portage/tests/lint/test_compile_modules.py index ec6edfa9b..c6e68946e 100644 --- a/pym/portage/tests/lint/test_compile_modules.py +++ b/pym/portage/tests/lint/test_compile_modules.py @@ -1,22 +1,49 @@ # Copyright 2009 Gentoo Foundation # Distributed under the terms of the GNU General Public License v2 -from portage.const import PORTAGE_PYM_PATH +import itertools +import stat + +from portage.const import PORTAGE_BIN_PATH, PORTAGE_PYM_PATH from portage.tests import TestCase from portage import os from portage import _encodings -from portage import _unicode_decode +from portage import _unicode_decode, _unicode_encode import py_compile class CompileModulesTestCase(TestCase): def testCompileModules(self): - for parent, dirs, files in os.walk(PORTAGE_PYM_PATH): + for parent, dirs, files in itertools.chain( + os.walk(PORTAGE_BIN_PATH), + os.walk(PORTAGE_PYM_PATH)): parent = _unicode_decode(parent, encoding=_encodings['fs'], errors='strict') for x in files: x = _unicode_decode(x, encoding=_encodings['fs'], errors='strict') + if x[-4:] in ('.pyc', '.pyo'): + continue + x = os.path.join(parent, x) + st = os.lstat(x) + if not stat.S_ISREG(st.st_mode): + continue + do_compile = False + cfile = x if x[-3:] == '.py': - py_compile.compile(os.path.join(parent, x), doraise=True) + do_compile = True + else: + # Check for python shebang + f = open(_unicode_encode(x, + encoding=_encodings['fs'], errors='strict'), 'rb') + line = _unicode_decode(f.readline(), + encoding=_encodings['content'], errors='replace') + f.close() + if line[:2] == '#!' and \ + 'python' in line: + do_compile = True + cfile += '.py' + if do_compile: + cfile += (__debug__ and 'c' or 'o') + py_compile.compile(x, cfile=cfile, doraise=True) -- cgit v1.2.3-1-g7c22