Brett Cannon | 7bf8d4e | 2009-02-07 02:33:52 +0000 | [diff] [blame] | 1 | import imp |
Brett Cannon | cb66eb0 | 2012-05-11 12:58:42 -0400 | [diff] [blame] | 2 | from importlib import machinery |
Brett Cannon | 7bf8d4e | 2009-02-07 02:33:52 +0000 | [diff] [blame] | 3 | import os |
| 4 | import sys |
| 5 | |
| 6 | PATH = None |
| 7 | EXT = None |
| 8 | FILENAME = None |
| 9 | NAME = '_testcapi' |
Brett Cannon | 7bf8d4e | 2009-02-07 02:33:52 +0000 | [diff] [blame] | 10 | try: |
| 11 | for PATH in sys.path: |
Brett Cannon | cb66eb0 | 2012-05-11 12:58:42 -0400 | [diff] [blame] | 12 | for EXT in machinery.EXTENSION_SUFFIXES: |
Brett Cannon | 7bf8d4e | 2009-02-07 02:33:52 +0000 | [diff] [blame] | 13 | FILENAME = NAME + EXT |
| 14 | FILEPATH = os.path.join(PATH, FILENAME) |
| 15 | if os.path.exists(os.path.join(PATH, FILENAME)): |
| 16 | raise StopIteration |
| 17 | else: |
| 18 | PATH = EXT = FILENAME = FILEPATH = None |
| 19 | except StopIteration: |
| 20 | pass |