zulip/zerver/lib/str_utils.py

"""
String Utilities:

This module helps in converting strings from one type to another.

Currently we have strings of 3 semantic types:

1.  text strings: These strings are used to represent all textual data,
    like people's names, stream names, content of messages, etc.
    These strings can contain non-ASCII characters, so its type should be
    six.text_type (which is `str` in python 3 and `unicode` in python 2).

2.  binary strings: These strings are used to represent binary data.
    This should be of type six.binary_type (which is `bytes` in python 3
    and `str` in python 2).

3.  native strings: These strings are for internal use only.  Strings of
    this type are not meant to be stored in database, displayed to end
    users, etc.  Things like exception names, parameter names, attribute
    names, etc should be native strings.  These strings should only
    contain ASCII characters and they should have type `str`.

There are 3 utility functions provided for converting strings from one type
to another - force_text, force_bytes, force_str

Interconversion between text strings and binary strings can be done by
using encode and decode appropriately or by using the utility functions
force_text and force_bytes.

It is recommended to use the utility functions for other string conversions.
"""

import six
from six import text_type, binary_type
from typing import Any, Mapping, Union, TypeVar

NonBinaryStr = TypeVar('NonBinaryStr', str, text_type)
# This is used to represent text or native strings

def force_text(s, encoding='utf-8'):
    # type: (Union[text_type, binary_type], str) -> text_type
    """converts a string to a text string"""
    if isinstance(s, text_type):
        return s
    elif isinstance(s, binary_type):
        return s.decode(encoding)
    else:
        raise TypeError("force_text expects a string type")

def force_bytes(s, encoding='utf-8'):
    # type: (Union[text_type, binary_type], str) -> binary_type
    """converts a string to binary string"""
    if isinstance(s, binary_type):
        return s
    elif isinstance(s, text_type):
        return s.encode(encoding)
    else:
        raise TypeError("force_bytes expects a string type")

def force_str(s, encoding='utf-8'):
    # type: (Union[text_type, binary_type], str) -> str
    """converts a string to a native string"""
    if isinstance(s, str):
        return s
    elif isinstance(s, text_type):
        return s.encode(encoding)
    elif isinstance(s, binary_type):
        return s.decode(encoding)
    else:
        raise TypeError("force_str expects a string type")

def dict_with_str_keys(dct, encoding='utf-8'):
    # type: (Mapping[NonBinaryStr, Any], str) -> Dict[str, Any]
    """applies force_str on the keys of a dict (non-recursively)"""
    return {force_str(key, encoding): value for key, value in six.iteritems(dct)}

class ModelReprMixin(object):
    """
    This mixin provides a python 2 and 3 compatible way of handling string representation of a model.
    When declaring a model, inherit this mixin before django.db.models.Model.
    Define __unicode__ on your model which returns a six.text_type object.
    This mixin will automatically define __str__ and __repr__.
    """
    def __unicode__(self):
        # type: () -> text_type
        # Originally raised an exception, but Django (e.g. the ./manage.py shell)
        # was catching the exception and not displaying any sort of error
        return u"Implement __unicode__ in your subclass of ModelReprMixin!"

    def __str__(self):
        # type: () -> str
        return force_str(self.__unicode__())

    def __repr__(self):
        # type: () -> str
        return force_str(self.__unicode__())