Thomas Heller | babddfc | 2006-03-08 19:56:54 +0000 | [diff] [blame] | 1 | import os |
| 2 | import sys |
| 3 | import unittest |
| 4 | |
| 5 | # Bob Ippolito: |
| 6 | """ |
| 7 | Ok.. the code to find the filename for __getattr__ should look |
| 8 | something like: |
| 9 | |
| 10 | import os |
| 11 | from macholib.dyld import dyld_find |
| 12 | |
| 13 | def find_lib(name): |
| 14 | possible = ['lib'+name+'.dylib', name+'.dylib', |
| 15 | name+'.framework/'+name] |
| 16 | for dylib in possible: |
| 17 | try: |
| 18 | return os.path.realpath(dyld_find(dylib)) |
| 19 | except ValueError: |
| 20 | pass |
| 21 | raise ValueError, "%s not found" % (name,) |
| 22 | |
| 23 | It'll have output like this: |
| 24 | |
| 25 | >>> find_lib('pthread') |
| 26 | '/usr/lib/libSystem.B.dylib' |
| 27 | >>> find_lib('z') |
| 28 | '/usr/lib/libz.1.dylib' |
| 29 | >>> find_lib('IOKit') |
| 30 | '/System/Library/Frameworks/IOKit.framework/Versions/A/IOKit' |
| 31 | |
| 32 | -bob |
| 33 | |
| 34 | """ |
| 35 | |
| 36 | from ctypes.macholib.dyld import dyld_find |
| 37 | |
| 38 | def find_lib(name): |
Tim Peters | e8d09e5 | 2006-03-09 01:15:05 +0000 | [diff] [blame] | 39 | possible = ['lib'+name+'.dylib', name+'.dylib', name+'.framework/'+name] |
| 40 | for dylib in possible: |
| 41 | try: |
| 42 | return os.path.realpath(dyld_find(dylib)) |
| 43 | except ValueError: |
| 44 | pass |
Collin Winter | a73bfee | 2007-08-30 03:47:13 +0000 | [diff] [blame] | 45 | raise ValueError("%s not found" % (name,)) |
Thomas Heller | babddfc | 2006-03-08 19:56:54 +0000 | [diff] [blame] | 46 | |
| 47 | class MachOTest(unittest.TestCase): |
| 48 | if sys.platform == "darwin": |
| 49 | def test_find(self): |
| 50 | |
Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 51 | self.assertEqual(find_lib('pthread'), |
Thomas Heller | babddfc | 2006-03-08 19:56:54 +0000 | [diff] [blame] | 52 | '/usr/lib/libSystem.B.dylib') |
| 53 | |
| 54 | result = find_lib('z') |
Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 55 | self.assertTrue(result.startswith('/usr/lib/libz.1')) |
| 56 | self.assertTrue(result.endswith('.dylib')) |
Thomas Heller | babddfc | 2006-03-08 19:56:54 +0000 | [diff] [blame] | 57 | |
Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 58 | self.assertEqual(find_lib('IOKit'), |
Thomas Heller | babddfc | 2006-03-08 19:56:54 +0000 | [diff] [blame] | 59 | '/System/Library/Frameworks/IOKit.framework/Versions/A/IOKit') |
| 60 | |
| 61 | if __name__ == "__main__": |
| 62 | unittest.main() |