Subversion Repositories altor32

# Copyright (C) 2010-2012 Free Software Foundation, Inc.
 
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
import traceback
import os
import sys
import _gdb
 
from _gdb import *
 
class GdbOutputFile:
    def close(self):
        # Do nothing.
        return None
 
    def isatty(self):
        return False
 
    def write(self, s):
        write(s, stream=STDOUT)
 
    def writelines(self, iterable):
        for line in iterable:
            self.write(line)
 
    def flush(self):
        flush()
 
sys.stdout = GdbOutputFile()
 
class GdbOutputErrorFile:
    def close(self):
        # Do nothing.
        return None
 
    def isatty(self):
        return False
 
    def write(self, s):
        write(s, stream=STDERR)
 
    def writelines(self, iterable):
        for line in iterable:
            self.write(line)
 
    def flush(self):
        flush()
 
sys.stderr = GdbOutputErrorFile()
 
# Default prompt hook does nothing.
prompt_hook = None
 
# Ensure that sys.argv is set to something.
# We do not use PySys_SetArgvEx because it did not appear until 2.6.6.
sys.argv = ['']
 
# Initial pretty printers.
pretty_printers = []
 
# Initial type printers.
type_printers = []
 
# Convenience variable to GDB's python directory
PYTHONDIR = os.path.dirname(os.path.dirname(__file__))
 
# Auto-load all functions/commands.
 
# Packages to auto-load.
 
packages = [
    'function',
    'command'
]
 
# pkgutil.iter_modules is not available prior to Python 2.6.  Instead,
# manually iterate the list, collating the Python files in each module
# path.  Construct the module name, and import.
 
def auto_load_packages():
    for package in packages:
        location = os.path.join(os.path.dirname(__file__), package)
        if os.path.exists(location):
            py_files = filter(lambda x: x.endswith('.py')
                                        and x != '__init__.py',
                              os.listdir(location))
 
            for py_file in py_files:
                # Construct from foo.py, gdb.module.foo
                modname = "%s.%s.%s" % ( __name__, package, py_file[:-3] )
                try:
                    if modname in sys.modules:
                        # reload modules with duplicate names
                        reload(__import__(modname))
                    else:
                        __import__(modname)
                except:
                    print >> sys.stderr, traceback.format_exc()
 
auto_load_packages()
 
def GdbSetPythonDirectory(dir):
    """Update sys.path, reload gdb and auto-load packages."""
    global PYTHONDIR
 
    try:
        sys.path.remove(PYTHONDIR)
    except ValueError:
        pass
    sys.path.insert(0, dir)
 
    PYTHONDIR = dir
 
    # note that reload overwrites the gdb module without deleting existing
    # attributes
    reload(__import__(__name__))
    auto_load_packages()