cleanup on six.string_types

encode · Apr 3, 2016 · 0e39cb6 · 0e39cb6
1 parent 36353eb
commit 0e39cb6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/rest_framework/authentication.py b/rest_framework/authentication.py
@@ -3,7 +3,8 @@
 """
 from __future__ import unicode_literals
 
-import base64, six
+import base64
+from six import string_types
 
 from django.contrib.auth import authenticate, get_user_model
 from django.middleware.csrf import CsrfViewMiddleware
@@ -19,7 +20,7 @@ def get_authorization_header(request):
     Hide some test client ickyness where the header can be unicode.
     """
     auth = request.META.get('HTTP_AUTHORIZATION', b'')
-    if isinstance(auth, six.string_types):
+    if isinstance(auth, string_types):
         # Work around django test client oddness
         auth = auth.encode(HTTP_HEADER_ENCODING)
     return auth