Python cbook.is_writable_file_like函数代码示例

    def _print_image(self, filename, format):
        if self.flags() & gtk.REALIZED == 0:
            # for self.window(for pixmap) and has a side effect of altering
            # figure width,height (via configure-event?)
            gtk.DrawingArea.realize(self)

        width, height = self.get_width_height()
        pixmap = gdk.Pixmap (self.window, width, height)
        self._renderer.set_pixmap (pixmap)
        self._render_figure(pixmap, width, height)

        # jpg colors don't match the display very well, png colors match
        # better
        pixbuf = gdk.Pixbuf(gdk.COLORSPACE_RGB, 0, 8, width, height)
        pixbuf.get_from_drawable(pixmap, pixmap.get_colormap(),
                                     0, 0, 0, 0, width, height)

        if is_string_like(filename):
            try:
                pixbuf.save(filename, format)
            except gobject.GError as exc:
                error_msg_gtk('Save figure failure:\n%s' % (exc,), parent=self)
        elif is_writable_file_like(filename):
            if hasattr(pixbuf, 'save_to_callback'):
                def save_callback(buf, data=None):
                    data.write(buf)
                try:
                    pixbuf.save_to_callback(save_callback, format, user_data=filename)
                except gobject.GError as exc:
                    error_msg_gtk('Save figure failure:\n%s' % (exc,), parent=self)
            else:
                raise ValueError("Saving to a Python file-like object is only supported by PyGTK >= 2.8")
        else:
            raise ValueError("filename must be a path or a file-like object")

 def print_svg(self, filename, *args, **kwargs):
     if is_string_like(filename):
         fh_to_close = svgwriter = codecs.open(filename, "w", "utf-8")
     elif is_writable_file_like(filename):
         svgwriter = codecs.EncodedFile(filename, "utf-8")
         fh_to_close = None
     else:
         raise ValueError("filename must be a path or a file-like object")
     return self._print_svg(filename, svgwriter, fh_to_close)

 def print_svgz(self, filename, *args, **kwargs):
     if is_string_like(filename):
         fh_to_close = gzipwriter = gzip.GzipFile(filename, 'w')
         svgwriter = io.TextIOWrapper(gzipwriter, 'utf-8')
     elif is_writable_file_like(filename):
         fh_to_close = gzipwriter = gzip.GzipFile(fileobj=filename, mode='w')
         svgwriter = io.TextIOWrapper(gzipwriter, 'utf-8')
     else:
         raise ValueError("filename must be a path or a file-like object")
     return self._print_svg(filename, svgwriter, fh_to_close)

 def print_svgz(self, filename, *args, **kwargs):
     if is_string_like(filename):
         gzipwriter = gzip.GzipFile(filename, "w")
         fh_to_close = svgwriter = codecs.EncodedFile(gzipwriter, "utf-8")
     elif is_writable_file_like(filename):
         fh_to_close = gzipwriter = gzip.GzipFile(fileobj=filename, mode="w")
         svgwriter = codecs.EncodedFile(gzipwriter, "utf-8")
     else:
         raise ValueError("filename must be a path or a file-like object")
     return self._print_svg(filename, svgwriter, fh_to_close)

 def print_png(self, fname_or_fh, *args, **kwargs):
     """
     Use LaTeX to compile a pgf figure to pdf and convert it to png.
     """
     if is_string_like(fname_or_fh):
         with open(fname_or_fh, "wb") as fh:
             self._print_png_to_fh(fh)
     elif is_writable_file_like(fname_or_fh):
         self._print_png_to_fh(fname_or_fh)
     else:
         raise ValueError("filename must be a path or a file-like object")

 def print_pdf(self, fname_or_fh, *args, **kwargs):
     """
     Use LaTeX to compile a Pgf generated figure to PDF.
     """
     # figure out where the pdf is to be written to
     if is_string_like(fname_or_fh):
         with open(fname_or_fh, "wb") as fh:
             self._print_pdf_to_fh(fh)
     elif is_writable_file_like(fname_or_fh):
         self._print_pdf_to_fh(fname_or_fh)
     else:
         raise ValueError("filename must be a path or a file-like object")

 def print_svg(self, filename, *args, **kwargs):
     if is_string_like(filename):
         fh_to_close = svgwriter = io.open(filename, "w", encoding="utf-8")
     elif is_writable_file_like(filename):
         if not isinstance(filename, io.TextIOBase):
             if sys.version_info[0] >= 3:
                 svgwriter = io.TextIOWrapper(filename, "utf-8")
             else:
                 svgwriter = codecs.getwriter("utf-8")(filename)
         fh_to_close = None
     else:
         raise ValueError("filename must be a path or a file-like object")
     return self._print_svg(filename, svgwriter, fh_to_close, **kwargs)

    def print_png(self, fname_or_fh, *args, **kwargs):
        """
        Use LaTeX to compile a pgf figure to pdf and convert it to png.
        """
        if kwargs.get("dryrun", False):
            self._print_pgf_to_fh(None, *args, **kwargs)
            return

        if isinstance(fname_or_fh, six.string_types):
            with open(fname_or_fh, "wb") as fh:
                self._print_png_to_fh(fh, *args, **kwargs)
        elif is_writable_file_like(fname_or_fh):
            self._print_png_to_fh(fname_or_fh, *args, **kwargs)
        else:
            raise ValueError("filename must be a path or a file-like object")

 def print_svg(self, filename, *args, **kwargs):
     if is_string_like(filename):
         fh_to_close = svgwriter = io.open(filename, 'w', encoding='utf-8')
     elif is_writable_file_like(filename):
         if not isinstance(filename, io.TextIOBase):
             if six.PY3:
                 svgwriter = io.TextIOWrapper(filename, 'utf-8')
             else:
                 svgwriter = codecs.getwriter('utf-8')(filename)
         else:
             svgwriter = filename
         fh_to_close = None
     else:
         raise ValueError("filename must be a path or a file-like object")
     return self._print_svg(filename, svgwriter, fh_to_close, **kwargs)

 def print_ipe(self, filename, *args, **kwargs):
     if isinstance(filename, string_types):
         fh_to_close = ipewriter = io.open(filename, 'w', encoding='utf-8')
     elif is_writable_file_like(filename):
         if not isinstance(filename, io.TextIOBase):
             if sys.version_info[0] >= 3:
                 ipewriter = io.TextIOWrapper(filename, 'utf-8')
             else:
                 ipewriter = codecs.getwriter('utf-8')(filename)
         else:
             ipewriter = filename
         fh_to_close = None
     else:
         raise ValueError("filename must be a path or a file-like object")
     return self._print_ipe(filename, ipewriter, fh_to_close, **kwargs)

    def print_pgf(self, fname_or_fh, *args, **kwargs):
        """
        Output pgf commands for drawing the figure so it can be included and
        rendered in latex documents.
        """
        if kwargs.get("dryrun", False): return

        # figure out where the pgf is to be written to
        if is_string_like(fname_or_fh):
            with codecs.open(fname_or_fh, "w", encoding="utf-8") as fh:
                self._print_pgf_to_fh(fh)
        elif is_writable_file_like(fname_or_fh):
            raise ValueError("saving pgf to a stream is not supported, " + \
            "consider using the pdf option of the pgf-backend")
        else:
            raise ValueError("filename must be a path")

    def print_pdf(self, fname_or_fh, *args, **kwargs):
        """
        Use LaTeX to compile a Pgf generated figure to PDF.
        """
        if kwargs.get("dryrun", False):
            self._print_pgf_to_fh(None, *args, **kwargs)
            return

        # figure out where the pdf is to be written to
        if isinstance(fname_or_fh, six.string_types):
            with open(fname_or_fh, "wb") as fh:
                self._print_pdf_to_fh(fh, *args, **kwargs)
        elif is_writable_file_like(fname_or_fh):
            self._print_pdf_to_fh(fname_or_fh, *args, **kwargs)
        else:
            raise ValueError("filename must be a path or a file-like object")

    def print_pgf(self, fname_or_fh, *args, **kwargs):
        """
        Output pgf commands for drawing the figure so it can be included and
        rendered in latex documents.
        """
        if kwargs.get("dryrun", False):
            self._print_pgf_to_fh(None, *args, **kwargs)
            return

        # figure out where the pgf is to be written to
        if is_string_like(fname_or_fh):
            with codecs.open(fname_or_fh, "w", encoding="utf-8") as fh:
                self._print_pgf_to_fh(fh, *args, **kwargs)
        elif is_writable_file_like(fname_or_fh):
            fh = codecs.getwriter("utf-8")(fname_or_fh)
            self._print_pgf_to_fh(fh, *args, **kwargs)
        else:
            raise ValueError("filename must be a path")

    def _print_image(self, filename, format, *args, **kwargs):
        if self.flags() & gtk.REALIZED == 0:
            # for self.window(for pixmap) and has a side effect of altering
            # figure width,height (via configure-event?)
            gtk.DrawingArea.realize(self)

        width, height = self.get_width_height()
        pixmap = gdk.Pixmap(self.window, width, height)
        self._renderer.set_pixmap(pixmap)
        self._render_figure(pixmap, width, height)

        # jpg colors don't match the display very well, png colors match
        # better
        pixbuf = gdk.Pixbuf(gdk.COLORSPACE_RGB, 0, 8, width, height)
        pixbuf.get_from_drawable(pixmap, pixmap.get_colormap(), 0, 0, 0, 0, width, height)

        # set the default quality, if we are writing a JPEG.
        # http://www.pygtk.org/docs/pygtk/class-gdkpixbuf.html#method-gdkpixbuf--save
        options = cbook.restrict_dict(kwargs, ["quality"])
        if format in ["jpg", "jpeg"]:
            if "quality" not in options:
                options["quality"] = rcParams["savefig.jpeg_quality"]

            options["quality"] = str(options["quality"])

        if is_string_like(filename):
            try:
                pixbuf.save(filename, format, options=options)
            except gobject.GError as exc:
                error_msg_gtk("Save figure failure:\n%s" % (exc,), parent=self)
        elif is_writable_file_like(filename):
            if hasattr(pixbuf, "save_to_callback"):

                def save_callback(buf, data=None):
                    data.write(buf)

                try:
                    pixbuf.save_to_callback(save_callback, format, user_data=filename, options=options)
                except gobject.GError as exc:
                    error_msg_gtk("Save figure failure:\n%s" % (exc,), parent=self)
            else:
                raise ValueError("Saving to a Python file-like object is only supported by PyGTK >= 2.8")
        else:
            raise ValueError("filename must be a path or a file-like object")

    def print_pgf(self, fname_or_fh, *args, **kwargs):
        """
        Output pgf commands for drawing the figure so it can be included and
        rendered in latex documents.
        """
        if kwargs.get("dryrun", False):
            self._print_pgf_to_fh(None, *args, **kwargs)
            return

        # figure out where the pgf is to be written to
        if is_string_like(fname_or_fh):
            with codecs.open(fname_or_fh, "w", encoding="utf-8") as fh:
                self._print_pgf_to_fh(fh, *args, **kwargs)
        elif is_writable_file_like(fname_or_fh):
            if not os.path.exists(fname_or_fh.name):
                warnings.warn("streamed pgf-code does not support raster "
                              "graphics, consider using the pgf-to-pdf option",
                              UserWarning)
            self._print_pgf_to_fh(fname_or_fh, *args, **kwargs)
        else:
            raise ValueError("filename must be a path")