Centralize compatibility imports in repoze.who._compat.

Tres Seaver

2012-03-18 adef053f4114bd614780340403f35e8092495ead

Centralize compatibility imports in repoze.who._compat.

 repoze/who/_auth_tkt.py

@@ -36,25 +36,14 @@
makes it possible to use the same authentication process with
non-Python code run under Apache.
"""

try:
    from Cookie import SimpleCookie
except ImportError: #pragma NO COVER Python >= 3.0
    from http.cookies import SimpleCookie
from hashlib import md5
import time as time_mod
try:
    from urllib import quote as url_quote
    from urllib import unquote as url_unquote
except ImportError: #pragma NO COVER Python >= 3.0
    from urllib.parse import quote as url_quote
    from urllib.parse import unquote as url_unquote
try:
    from hashlib import md5
except ImportError:
    from md5 import md5

from repoze.who._compat import get_cookies
from repoze.who._compat import SimpleCookie
from repoze.who._compat import STRING_TYPES
from repoze.who._compat import url_quote
from repoze.who._compat import url_unquote


class AuthTicket(object):

 repoze/who/_compat.py

@@ -1,12 +1,3 @@
import wsgiref.util
import wsgiref.headers
try:
    from Cookie import SimpleCookie
    from Cookie import CookieError
except ImportError: #pragma NO COVER Python >= 3.0
    from http.cookies import SimpleCookie
    from http.cookies import CookieError

try:
    STRING_TYPES = (str, unicode)
except NameError:
@@ -24,6 +15,40 @@
    from base64 import decodestring as decodebytes
    from base64 import encodestring as encodebytes

try:
    from ConfigParser import ConfigParser
    from ConfigParser import ParsingError
except ImportError: #pragma NO COVER Python >= 3.0
    from configparser import ConfigParser
    from configparser import ParsingError

try:
    from Cookie import SimpleCookie
    from Cookie import CookieError
except ImportError: #pragma NO COVER Python >= 3.0
    from http.cookies import SimpleCookie
    from http.cookies import CookieError

try:
    from itertools import izip_longest
except ImportError: #pragma NO COVER Python >= 3.0
    from itertools import zip_longest as izip_longest

try:
    from StringIO import StringIO
except ImportError: #pragma NO COVER Python >= 3.0
    from io import StringIO

try:
    from urllib import quote as url_quote
    from urllib import unquote as url_unquote
except ImportError: #pragma NO COVER Python >= 3.0
    from urllib.parse import quote as url_quote
    from urllib.parse import unquote as url_unquote

import wsgiref.util
import wsgiref.headers

def REQUEST_METHOD(environ):
    return environ['REQUEST_METHOD']


 repoze/who/config.py

@@ -1,15 +1,5 @@
""" Configuration parser
"""
try:
    from ConfigParser import ConfigParser
    from ConfigParser import ParsingError
except ImportError: #pragma NO COVER Python >= 3.0
    from configparser import ConfigParser
    from configparser import ParsingError
try:
    from StringIO import StringIO
except ImportError: #pragma NO COVER Python >= 3.0
    from io import StringIO
import logging
from pkg_resources import EntryPoint
import sys
@@ -24,6 +14,9 @@
from repoze.who.interfaces import IPlugin
from repoze.who.interfaces import IRequestClassifier
from repoze.who.middleware import PluggableAuthenticationMiddleware
from repoze.who._compat import StringIO
from repoze.who._compat import ConfigParser
from repoze.who._compat import ParsingError

def _resolve(name):
    if name:

 repoze/who/middleware.py

@@ -1,12 +1,9 @@
import logging
try:
    from StringIO import StringIO
except ImportError: #pragma NO COVER Python >= 3.0
    from io import StringIO
import sys

from repoze.who.api import APIFactory
from repoze.who.interfaces import IChallenger
from repoze.who._compat import StringIO

_STARTED = '-- repoze.who request started (%s) --'
_ENDED = '-- repoze.who request ended (%s) --'

 repoze/who/tests/test_config.py

@@ -39,7 +39,7 @@
        self.assertEqual(len(config.mdproviders), 0)

    def test_parse_empty_file(self):
        from StringIO import StringIO
        from repoze.who._compat import StringIO
        config = self._makeOne()
        config.parse(StringIO())
        self.assertEqual(config.request_classifier, None)

 repoze/who/tests/test_middleware.py

@@ -487,7 +487,7 @@
        self.failUnless(wrapper.buffer)

    def test_finish_response(self):
        from StringIO import StringIO
        from repoze.who._compat import StringIO
        statuses = []
        headerses = []
        datases = []