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()