| Zachary Turner | 8741e31 | 2015-10-20 21:05:57 | [diff] [blame] | 1 | import inspect |
| 2 | import os |
| 3 | import sys |
| 4 | |
| Kate Stone | b9c1b51 | 2016-09-06 20:57:50 | [diff] [blame] | 5 | |
| Zachary Turner | 8741e31 | 2015-10-20 21:05:57 | [diff] [blame] | 6 | def add_third_party_module_dirs(lldb_root): |
| Kate Stone | b9c1b51 | 2016-09-06 20:57:50 | [diff] [blame] | 7 | third_party_modules_dir = os.path.join( |
| 8 | lldb_root, "third_party", "Python", "module") |
| Zachary Turner | 8741e31 | 2015-10-20 21:05:57 | [diff] [blame] | 9 | if not os.path.isdir(third_party_modules_dir): |
| 10 | return |
| 11 | |
| 12 | module_dirs = os.listdir(third_party_modules_dir) |
| 13 | for module_dir in module_dirs: |
| 14 | module_dir = os.path.join(third_party_modules_dir, module_dir) |
| Zachary Turner | 7b24ff1 | 2015-10-22 19:55:18 | [diff] [blame] | 15 | sys.path.insert(0, module_dir) |
| Zachary Turner | af383ff | 2015-10-27 22:33:47 | [diff] [blame] | 16 | |
| Kate Stone | b9c1b51 | 2016-09-06 20:57:50 | [diff] [blame] | 17 | |
| Zachary Turner | af383ff | 2015-10-27 22:33:47 | [diff] [blame] | 18 | def add_lldbsuite_packages_dir(lldb_root): |
| 19 | packages_dir = os.path.join(lldb_root, "packages", "Python") |
| 20 | sys.path.insert(0, packages_dir) |
| 21 | |
| Zachary Turner | 8741e31 | 2015-10-20 21:05:57 | [diff] [blame] | 22 | lldb_root = os.path.dirname(inspect.getfile(inspect.currentframe())) |
| Zachary Turner | af383ff | 2015-10-27 22:33:47 | [diff] [blame] | 23 | |
| Zachary Turner | 8741e31 | 2015-10-20 21:05:57 | [diff] [blame] | 24 | add_third_party_module_dirs(lldb_root) |
| Zachary Turner | af383ff | 2015-10-27 22:33:47 | [diff] [blame] | 25 | add_lldbsuite_packages_dir(lldb_root) |
