upstream/ipython Files · IPython/utils/syspathcontext.py

Merge pull request from bfroehle/prun_print_stats...

Merge pull request from bfroehle/prun_print_stats %prun: Restore `stats.stream` after running `print_stream` When profiling, the output from `print_stream` is captured so that the output can be paged. To do so, the `stream` attribute is set to a `StringIO` object but was not restored after the output was captured. As a result, the stats object returned by `%prun -r` did not display any output when calling its `print_stream` method. In addition, the `pstats.Stats` class has the `stream` attribute for all currently supported Python versions (2.6, 2.7, and 3.2, at least), so the extra branch of code can be removed. Closes

Matthias BUSSONNIER - - Load All Authors

File last commit:

r5390:c82649ea


                r7959:965fc650

Download file

             syspathcontext.py
        
                    69 lines
            
             | 2.0 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / utils / syspathcontext.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # encoding: utf-8

      """

      Context managers for adding things to sys.path temporarily.

      Authors:

      * Brian Granger

      """

      #-----------------------------------------------------------------------------

      #  Copyright (C) 2008-2011  The IPython Development Team

      #

      #  Distributed under the terms of the BSD License.  The full license is in

      #  the file COPYING, distributed as part of this software.

      #-----------------------------------------------------------------------------

      #-----------------------------------------------------------------------------

      # Imports

      #-----------------------------------------------------------------------------

      import sys

      #-----------------------------------------------------------------------------

      # Code

      #-----------------------------------------------------------------------------

      class appended_to_syspath(object):

          """A context for appending a directory to sys.path for a second."""

          def __init__(self, dir):

              self.dir = dir

          def __enter__(self):

              if self.dir not in sys.path:

                  sys.path.append(self.dir)

                  self.added = True

              else:

                  self.added = False

          def __exit__(self, type, value, traceback):

              if self.added:

                  try:

                      sys.path.remove(self.dir)

                  except ValueError:

                      pass

              # Returning False causes any exceptions to be re-raised.

              return False

      class prepended_to_syspath(object):

          """A context for prepending a directory to sys.path for a second."""

          def __init__(self, dir):

              self.dir = dir

          def __enter__(self):

              if self.dir not in sys.path:

                  sys.path.insert(0,self.dir)

                  self.added = True

              else:

                  self.added = False

          def __exit__(self, type, value, traceback):

              if self.added:

                  try:

                      sys.path.remove(self.dir)

                  except ValueError:

                      pass

              # Returning False causes any exceptions to be re-raised.

              return False

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# encoding: utf-8
				"""
				Context managers for adding things to sys.path temporarily.

				Authors:

				* Brian Granger
				"""

				#-----------------------------------------------------------------------------
				# Copyright (C) 2008-2011 The IPython Development Team
				#
				# Distributed under the terms of the BSD License. The full license is in
				# the file COPYING, distributed as part of this software.
				#-----------------------------------------------------------------------------

				#-----------------------------------------------------------------------------
				# Imports
				#-----------------------------------------------------------------------------

				import sys

				#-----------------------------------------------------------------------------
				# Code
				#-----------------------------------------------------------------------------

				class appended_to_syspath(object):
				"""A context for appending a directory to sys.path for a second."""

				def __init__(self, dir):
				self.dir = dir

				def __enter__(self):
				if self.dir not in sys.path:
				sys.path.append(self.dir)
				self.added = True
				else:
				self.added = False

				def __exit__(self, type, value, traceback):
				if self.added:
				try:
				sys.path.remove(self.dir)
				except ValueError:
				pass
				# Returning False causes any exceptions to be re-raised.
				return False

				class prepended_to_syspath(object):
				"""A context for prepending a directory to sys.path for a second."""

				def __init__(self, dir):
				self.dir = dir

				def __enter__(self):
				if self.dir not in sys.path:
				sys.path.insert(0,self.dir)
				self.added = True
				else:
				self.added = False

				def __exit__(self, type, value, traceback):
				if self.added:
				try:
				sys.path.remove(self.dir)
				except ValueError:
				pass
				# Returning False causes any exceptions to be re-raised.
				return False