You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.7 KiB
65 lines
1.7 KiB
"""
|
|
Common functions for providing cross-python version compatibility.
|
|
"""
|
|
import sys
|
|
import re
|
|
from six import integer_types
|
|
|
|
|
|
def str_idx_as_int(string, index):
|
|
"""Take index'th byte from string, return as integer"""
|
|
val = string[index]
|
|
if isinstance(val, integer_types):
|
|
return val
|
|
return ord(val)
|
|
|
|
|
|
if sys.version_info < (3, 0): # pragma: no branch
|
|
|
|
def normalise_bytes(buffer_object):
|
|
"""Cast the input into array of bytes."""
|
|
# flake8 runs on py3 where `buffer` indeed doesn't exist...
|
|
return buffer(buffer_object) # noqa: F821
|
|
|
|
def hmac_compat(ret):
|
|
return ret
|
|
|
|
if sys.version_info < (2, 7) or sys.version_info < ( # pragma: no branch
|
|
2,
|
|
7,
|
|
4,
|
|
):
|
|
|
|
def remove_whitespace(text):
|
|
"""Removes all whitespace from passed in string"""
|
|
return re.sub(r"\s+", "", text)
|
|
|
|
else:
|
|
|
|
def remove_whitespace(text):
|
|
"""Removes all whitespace from passed in string"""
|
|
return re.sub(r"\s+", "", text, flags=re.UNICODE)
|
|
|
|
|
|
else:
|
|
if sys.version_info < (3, 4): # pragma: no branch
|
|
# on python 3.3 hmac.hmac.update() accepts only bytes, on newer
|
|
# versions it does accept memoryview() also
|
|
def hmac_compat(data):
|
|
if not isinstance(data, bytes): # pragma: no branch
|
|
return bytes(data)
|
|
return data
|
|
|
|
else:
|
|
|
|
def hmac_compat(data):
|
|
return data
|
|
|
|
def normalise_bytes(buffer_object):
|
|
"""Cast the input into array of bytes."""
|
|
return memoryview(buffer_object).cast("B")
|
|
|
|
def remove_whitespace(text):
|
|
"""Removes all whitespace from passed in string"""
|
|
return re.sub(r"\s+", "", text, flags=re.UNICODE)
|