Question

嗨，stackoverflow中有一个与此链接相同的队列： How to generate temporary URLs in Django

但是接受的答案代码是针对python 2的，我把它转换为python3 vertion：

import hashlib, zlib
import pickle as pickle
import urllib.request, urllib.parse, urllib.error

my_secret = "michnorts"

def encode_data(data):
    """Turn `data` into a hash and an encoded string, suitable for use with `decode_data`."""
    text = zlib.compress(pickle.dumps(data, 0)).encode('base64').replace('\n', '')
    m = hashlib.md5(my_secret + text).hexdigest()[:12]
    return m, text

def decode_data(hash, enc):
    """The inverse of `encode_data`."""
    text = urllib.parse.unquote(enc)
    m = hashlib.md5(my_secret + text).hexdigest()[:12]
    if m != hash:
        raise Exception("Bad hash!")
    data = pickle.loads(zlib.decompress(text.decode('base64')))
    return data

hash, enc = encode_data(['Hello', 'Goodbye'])
print(hash, enc)
print(decode_data(hash, enc))

但它有错误：

    text = zlib.compress(pickle.dumps(data, 0)).encode('base64').replace('\n', '')
AttributeError: 'bytes' object has no attribute 'encode'

我该如何解决这个问题？

Answer 1

尝试使您的代码适应Python 3，我提出了这个：

import hashlib, zlib
import pickle as pickle
import urllib.request, urllib.parse, urllib.error
import base64

my_secret = "michnorts"

def encode_data(data):
    """Turn `data` into a hash and an encoded string, suitable for use with `decode_data`."""
    compressed_text = zlib.compress(pickle.dumps(data, 0))
    text = base64.b64encode(compressed_text).decode().replace('\n', '')
    m = hashlib.md5(str.encode('{}{}'.format(my_secret, text))).hexdigest()[:12]
    return m, text

def decode_data(hash, enc):
    """The inverse of `encode_data`."""
    text = urllib.parse.unquote(enc)
    m = hashlib.md5(str.encode('{}{}'.format(my_secret, text))).hexdigest()[:12]
    if m != hash:
        raise Exception("Bad hash!")
    data = pickle.loads(zlib.decompress(base64.b64decode(text)))
    return data

hash, enc = encode_data(['Hello', 'Goodbye'])
print(hash, enc)
print(decode_data(hash, enc))

我需要考虑一些事项：

在Python 3中，编码/解码为base64的方法是使用base64 module
将bytes对象转换为字符串，我使用了bytes.decode method
将字符串对象强制转换为bytes对象，我使用了str.encode function
hashlib.md5函数接受bytes对象（字符串需要先编码）
我改变了将字符串（即str1 + str2）与更加pythonic结构连接起来的方式：'{}{}'.format(str1, str2)

我希望这会有所帮助;）

Answer 2

我建议使用内置的秘密模块，尤其是secrets.token_urlsafe。

在Django 2 python 3中生成临时URL

2 个答案: