|
1 # Copyright (C) 2005, Giovanni Bajo |
|
2 # Based on previous work under copyright (c) 2001, 2002 McMillan Enterprises, Inc. |
|
3 # |
|
4 # This program is free software; you can redistribute it and/or |
|
5 # modify it under the terms of the GNU General Public License |
|
6 # as published by the Free Software Foundation; either version 2 |
|
7 # of the License, or (at your option) any later version. |
|
8 # |
|
9 # This program is distributed in the hope that it will be useful, |
|
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
12 # GNU General Public License for more details. |
|
13 # |
|
14 # You should have received a copy of the GNU General Public License |
|
15 # along with this program; if not, write to the Free Software |
|
16 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA |
|
17 import sys, string |
|
18 |
|
19 def hook(mod): |
|
20 names = sys.builtin_module_names |
|
21 if 'posix' in names: |
|
22 removes = ['nt', 'dos', 'os2', 'mac', 'win32api'] |
|
23 elif 'nt' in names: |
|
24 removes = ['dos', 'os2', 'mac'] |
|
25 elif 'os2' in names: |
|
26 removes = ['nt', 'dos', 'mac', 'win32api'] |
|
27 elif 'dos' in names: |
|
28 removes = ['nt', 'os2', 'mac', 'win32api'] |
|
29 elif 'mac' in names: |
|
30 removes = ['nt', 'dos', 'os2', 'win32api'] |
|
31 for i in range(len(mod.imports)-1, -1, -1): |
|
32 nm = mod.imports[i][0] |
|
33 pos = string.find(nm, '.') |
|
34 if pos > -1: |
|
35 nm = nm[:pos] |
|
36 if nm in removes: |
|
37 del mod.imports[i] |
|
38 return mod |
|
39 |