aboutsummaryrefslogtreecommitdiffstats
path: root/mediagoblin/_compat.py
blob: 65fb5140aa12e827e55748d10b5f09b3598e9cef (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from six import PY3, iteritems

from mediagoblin import mg_globals

if PY3:
    from email.mime.text import MIMEText
    from urllib import parse as urlparse
    # TODO(berker): Rename to gettext and ungettext instead?
    ugettext = mg_globals.thread_scope.translations.gettext
    ungettext = mg_globals.thread_scope.translations.ngettext
else:
    from email.MIMEText import MIMEText
    import urlparse
    ugettext = mg_globals.thread_scope.translations.ugettext
    ungettext = mg_globals.thread_scope.translations.ungettext


# taken from
# https://github.com/django/django/blob/master/django/utils/encoding.py
def py2_unicode(klass):
    if not PY3:
        if '__str__' not in klass.__dict__:
            raise ValueError("@py2_unicode cannot be applied "
                             "to %s because it doesn't define __str__()." %
                             klass.__name__)
        klass.__unicode__ = klass.__str__
        klass.__str__ = lambda self: self.__unicode__().encode('utf-8')
    return klass