Source code

Revision control

Copy as Markdown

Other Tools

# Licensed under the Mozilla Public Licence 2.0.
import sys
import uuid
import base64
def encode(uuid_):
"""
Returns the given uuid.UUID object as a 22 character slug. This can be a
regular v4 slug or a "nice" slug.
"""
return _convert_bytes_to_slug(uuid_.bytes)
def _convert_bytes_to_slug(bytes_):
slug = base64.urlsafe_b64encode(bytes_)[:-2] # Drop '==' padding
if sys.version_info.major != 2 and isinstance(slug, bytes):
slug = slug.decode('utf-8')
return slug
def decode(slug):
"""
Returns the uuid.UUID object represented by the given v4 or "nice" slug
"""
if sys.version_info.major != 2 and isinstance(slug, bytes):
slug = slug.decode('ascii')
slug = slug + '==' # base64 padding
return uuid.UUID(bytes=base64.urlsafe_b64decode(slug))
def v4():
"""
Returns a randomly generated uuid v4 compliant slug
"""
return _convert_bytes_to_slug(uuid.uuid4().bytes)
def nice():
"""
Returns a randomly generated uuid v4 compliant slug which conforms to a set
of "nice" properties, at the cost of some entropy. Currently this means one
extra fixed bit (the first bit of the uuid is set to 0) which guarantees the
slug will begin with [A-Za-f]. For example such slugs don't require special
handling when used as command line parameters (whereas non-nice slugs may
start with `-` which can confuse command line tools).
Potentially other "nice" properties may be added in future to further
restrict the range of potential uuids that may be generated.
"""
rawBytes = bytearray(uuid.uuid4().bytes)
rawBytes[0] = rawBytes[0] & 0x7f # Ensure slug starts with [A-Za-f]
return _convert_bytes_to_slug(rawBytes)