python/samba/compat.py

   1 # module which helps with porting to Python 3
   2 #
   3 # Copyright (C) Lumir Balhar <lbalhar@redhat.com> 2017
   4 #
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 3 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 """module which helps with porting to Python 3"""
  19
  20 import sys
  21
  22 PY3 = sys.version_info[0] == 3
  23
  24 if PY3:
  25     # compat functions
  26     from  urllib.parse import quote as urllib_quote
  27     from urllib.request import urlopen as urllib_urlopen
  28
  29     # compat types
  30     integer_types = int,
  31     string_types = str
  32     text_type = str
  33     binary_type = bytes
  34
  35     # alias
  36     import io
  37     StringIO = io.StringIO
  38 else:
  39     # compat functions
  40     from urllib import quote as urllib_quote
  41     from urllib import urlopen as urllib_urlopen
  42
  43     # compat types
  44     integer_types = (int, long)
  45     string_types = basestring
  46     text_type = unicode
  47     binary_type = str
  48
  49     # alias
  50     import StringIO
  51     StringIO = StringIO.StringIO