Fixing unit tests for python3 for custom staff filters

This commit is contained in:
Garret Wassermann
2017-10-30 05:07:44 -04:00
parent 27ee01449e
commit 6e2a16e5a6
5 changed files with 48 additions and 44 deletions

View File

@ -11,18 +11,14 @@ import mimetypes
import os
from smtplib import SMTPException
try:
# Python 2 support
from base64 import urlsafe_b64encode as b64encode
except ImportError:
# Python 3 support
import six
if six.PY3:
from base64 import encodebytes as b64encode
try:
# Python 2 support
from base64 import urlsafe_b64decode as b64decode
except ImportError:
# Python 3 support
from base64 import decodebytes as b64decode
else:
from base64 import urlsafe_b64encode as b64encode
from base64 import urlsafe_b64decode as b64decode
from django.conf import settings
from django.db.models import Q