Move the random hex generator to helpers.
This commit is contained in:
parent
2fceeedbbb
commit
0228fbebfd
2 changed files with 7 additions and 4 deletions
|
@ -217,6 +217,12 @@ def now(timestamp=True):
|
|||
return n.timestamp()
|
||||
return n
|
||||
|
||||
def random_hex(length=12):
|
||||
randbytes = os.urandom(math.ceil(length / 2))
|
||||
token = ''.join('{:02x}'.format(x) for x in randbytes)
|
||||
token = token[:length]
|
||||
return token
|
||||
|
||||
def read_filebytes(filepath, range_min, range_max, chunk_size=2 ** 20):
|
||||
'''
|
||||
Yield chunks of bytes from the file between the endpoints.
|
||||
|
|
|
@ -10,10 +10,7 @@ SESSION_MAX_AGE = 86400
|
|||
REQUEST_TYPES = (flask.Request, werkzeug.wrappers.Request, werkzeug.local.LocalProxy)
|
||||
|
||||
def _generate_token(length=32):
|
||||
randbytes = os.urandom(math.ceil(length / 2))
|
||||
token = ''.join('{:02x}'.format(x) for x in randbytes)
|
||||
token = token[:length]
|
||||
return token
|
||||
return etiquette.helpers.random_hex(length=length)
|
||||
|
||||
def _normalize_token(token):
|
||||
if isinstance(token, REQUEST_TYPES):
|
||||
|
|
Loading…
Reference in a new issue