writer.py

   1 # -*- encoding: utf-8 -*-
   2 #
   3 #
   4 # Copyright (C) 2013 Jörg Lehmann <joergl@users.sourceforge.net>
   5 # Copyright (C) 2013 André Wobst <wobsta@users.sourceforge.net>
   6 #
   7 # This file is part of PyX (http://pyx.sourceforge.net/).
   8 #
   9 # PyX is free software; you can redistribute it and/or modify
  10 # it under the terms of the GNU General Public License as published by
  11 # the Free Software Foundation; either version 2 of the License, or
  12 # (at your option) any later version.
  13 #
  14 # PyX is distributed in the hope that it will be useful,
  15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17 # GNU General Public License for more details.
  18 #
  19 # You should have received a copy of the GNU General Public License
  20 # along with PyX; if not, write to the Free Software
  21 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
  22
  23 class writer:
  24
  25     def __init__(self, file, encoding="ascii", errors="surrogateescape"):
  26         self.file = file
  27         self.encoding = encoding
  28         self.errors = errors
  29
  30     def write(self, s):
  31         self.file.write(s.encode(self.encoding, errors=self.errors))
  32
  33     def write_bytes(self, b):
  34         self.file.write(b)
  35
  36     def tell(self):
  37         return self.file.tell()
  38
  39     def __enter__(self):
  40         return self
  41
  42     def __exit__(self, exc_type, exc_value, traceback):
  43         return self.file.__exit__(exc_type, exc_value, traceback)