1 # Copyright (C) 2010-2013 Free Software Foundation, Inc.
3 # This program is free software; you can redistribute it and/or modify
4 # it under the terms of the GNU General Public License as published by
5 # the Free Software Foundation; either version 3 of the License, or
6 # (at your option) any later version.
8 # This program is distributed in the hope that it will be useful,
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # GNU General Public License for more details.
13 # You should have received a copy of the GNU General Public License
14 # along with this program. If not, see <http://www.gnu.org/licenses/>.
21 if sys.version_info[0] > 2:
22 # Python 3 moved "reload"
23 from imp import reload
27 class _GdbFile (object):
28 # These two are needed in Python 3
39 def writelines(self, iterable):
46 class GdbOutputFile (_GdbFile):
48 write(s, stream=STDOUT)
50 sys.stdout = GdbOutputFile()
52 class GdbOutputErrorFile (_GdbFile):
54 write(s, stream=STDERR)
56 sys.stderr = GdbOutputErrorFile()
58 # Default prompt hook does nothing.
61 # Ensure that sys.argv is set to something.
62 # We do not use PySys_SetArgvEx because it did not appear until 2.6.6.
65 # Initial pretty printers.
68 # Initial type printers.
71 # Convenience variable to GDB's python directory
72 PYTHONDIR = os.path.dirname(os.path.dirname(__file__))
74 # Auto-load all functions/commands.
76 # Packages to auto-load.
83 # pkgutil.iter_modules is not available prior to Python 2.6. Instead,
84 # manually iterate the list, collating the Python files in each module
85 # path. Construct the module name, and import.
87 def auto_load_packages():
88 for package in packages:
89 location = os.path.join(os.path.dirname(__file__), package)
90 if os.path.exists(location):
91 py_files = filter(lambda x: x.endswith('.py')
92 and x != '__init__.py',
95 for py_file in py_files:
96 # Construct from foo.py, gdb.module.foo
97 modname = "%s.%s.%s" % ( __name__, package, py_file[:-3] )
99 if modname in sys.modules:
100 # reload modules with duplicate names
101 reload(__import__(modname))
105 sys.stderr.write (traceback.format_exc() + "\n")
109 def GdbSetPythonDirectory(dir):
110 """Update sys.path, reload gdb and auto-load packages."""
114 sys.path.remove(PYTHONDIR)
117 sys.path.insert(0, dir)
121 # note that reload overwrites the gdb module without deleting existing
123 reload(__import__(__name__))