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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
"""
HappyBase utility module.
These functions are not part of the public API.
"""
import re
import six
from six.moves import range
CAPITALS = re.compile('([A-Z])')
try:
# Python 2.7
from collections import OrderedDict
except ImportError:
try:
# External package for Python 2.6
from ordereddict import OrderedDict
except ImportError as exc:
# Stub to throw errors at run-time (not import time)
def OrderedDict(*args, **kwargs):
raise RuntimeError(
"No OrderedDict implementation available; please "
"install the 'ordereddict' Package from PyPI.")
def camel_case_to_pep8(name):
"""Convert a camel cased name to PEP8 style."""
converted = CAPITALS.sub(lambda m: '_' + m.groups()[0].lower(), name)
if converted[0] == '_':
return converted[1:]
else:
return converted
def pep8_to_camel_case(name, initial=False):
"""Convert a PEP8 style name to camel case."""
chunks = name.split('_')
converted = [s[0].upper() + s[1:].lower() for s in chunks]
if initial:
return ''.join(converted)
else:
return chunks[0].lower() + ''.join(converted[1:])
def thrift_attrs(obj_or_cls):
"""Obtain Thrift data type attribute names for an instance or class."""
return [v[1] for v in obj_or_cls.thrift_spec.values()]
def thrift_type_to_dict(obj):
"""Convert a Thrift data type to a regular dictionary."""
return dict((camel_case_to_pep8(attr), getattr(obj, attr))
for attr in thrift_attrs(obj))
def ensure_bytes(str_or_bytes, binary_type=six.binary_type,
text_type=six.text_type):
"""Convert text into bytes, and leaves bytes as-is."""
if isinstance(str_or_bytes, binary_type):
return str_or_bytes
if isinstance(str_or_bytes, text_type):
return str_or_bytes.encode('utf-8')
raise TypeError(
"input must be a text or byte string, got {}"
.format(type(str_or_bytes).__name__))
def str_increment(s):
"""Increment and truncate a byte string (for sorting purposes)
This functions returns the shortest string that sorts after the given
string when compared using regular string comparison semantics.
This function increments the last byte that is smaller than ``0xFF``, and
drops everything after it. If the string only contains ``0xFF`` bytes,
`None` is returned.
"""
for i in range(len(s) - 1, -1, -1):
if s[i] != '\xff':
return s[:i] + chr(ord(s[i]) + 1)
return None
|
