Python3 and hmac . How to handle string not being binary

匿名 (未验证) 提交于 2019-12-03 01:47:02

问题:

I had a script in Python2 that was working great.

def _generate_signature(data):    return hmac.new('key', data, hashlib.sha256).hexdigest() 

Where data was the output of json.dumps.

Now, if I try to run the same kind of code in Python 3, I get the following:

Traceback (most recent call last):   File "", line 1, in    File "/usr/lib/python3.4/hmac.py", line 144, in new     return HMAC(key, msg, digestmod)   File "/usr/lib/python3.4/hmac.py", line 42, in __init__     raise TypeError("key: expected bytes or bytearray, but got %r" %type(key).__name__) TypeError: key: expected bytes or bytearray, but got 'str' 

If I try something like transforming the key to bytes like so:

bytes('key') 

I get

Traceback (most recent call last):   File "", line 1, in  TypeError: string argument without an encoding 

I'm still struggling to understand the encodings in Python 3.

回答1:

You can use bytes literal: b'key'

def _generate_signature(data):     return hmac.new(b'key', data, hashlib.sha256).hexdigest() 

In addition to that, make sure data is also bytes. For example, if it is read from file, you need to use binary mode (rb) when opening the file.



回答2:

Not to resurrect an old question but I did want to add something I feel is missing from this answer, to which I had trouble finding an appropriate explanation/example of anywhere else:

OP Aquiles Carattino was pretty close with his attempt at converting the string to bytes, but was missing the second argument, the encoding of the string to be converted to bytes.

If someone would like to convert a string to bytes through some other means than static assignment (such as reading from a config file or a DB), the following should work:

import hmac, hashlib  def _generate_signature(data):   key = 'key' # Defined as a simple string.   key_bytes= bytes(key , 'latin-1')   data_bytes = bytes(data, 'latin-1') # Assumes `data` is also a string.   return hmac.new(key_bytes, data_bytes , hashlib.sha256).hexdigest()  print(   _generate_signature('this is my string of data') ) 


易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!