将Python脚本TBA SHA1更改为SHA256

3

我最近被一家使用NetSuite的大公司雇用为初级开发人员。一位老开发人员编写了一个处理由设计师制作的图片的Python脚本,当这些图片上传到特定文件夹时,它会将它们上传到NetSuite。

由于该脚本使用的是SHA1,因此我需要更改TBA以使用SHA256,因为NetSuite不再支持SHA1。

我很难理解老开发人员的代码,并找到有关如何将TBA从SHA1更改为SHA256的文档。

以下是代码片段:

import datetime
import requests
import os
import oauth2 as oauth
import json
import time
import base64
import sys
import hashlib
import hmac
    url = "https://xxxxx=1"
token = oauth.Token(key="xxxxxxxxxxx",secret="xxxxxxxxxx")
consumer = oauth.Consumer(key="xxxxxxxxxxxxxxxx",secret="xxxxxxxxxxxxxxxx")
realm="xxxxxxxxxxxxxx"
signature_method = oauth.SignatureMethod_HMAC_SHA1()

在这部分中,我理解他初始化了方法oauth.SignatureMethod_HMAC_SHA1()

然后当我去查看oauth文件时,我发现了这个:

class SignatureMethod_HMAC_SHA1(SignatureMethod):
    name = 'HMAC-SHA1'

    def signing_base(self, request, consumer, token):
        if (not hasattr(request, 'normalized_url') or request.normalized_url is None):
            raise ValueError("Base URL for request is not set.")

        sig = (
            escape(request.method),
            escape(request.normalized_url),
            escape(request.get_normalized_parameters()),
        )

        key = '%s&' % escape(consumer.secret)
        if token:
            key += escape(token.secret)
        raw = '&'.join(sig)
        return key.encode('ascii'), raw.encode('ascii')

    def sign(self, request, consumer, token):
        """Builds the base signature string."""
        key, raw = self.signing_base(request, consumer, token)

        hashed = hmac.new(key, raw, sha1)

        # Calculate the digest base 64.
        return binascii.b2a_base64(hashed.digest())[:-1]

我仔细查看了这个文件,里面没有包含任何使用SHA256的方法,只有SHA1和PLAINTEXT。

我尝试将值更改为SHA256,但当然没有成功。 我试图查找关于oAuth2的文档,但只找到了很少量的信息,似乎它只包含SHA1和PLAINTEXT。

那么如何更改脚本以使用SHA256而不是SHA1呢?

编辑以回答评论 Hashlib 包含了这个:

    class _Hash(object):
    digest_size: int
    block_size: int

    # [Python documentation note] Changed in version 3.4: The name attribute has
    # been present in CPython since its inception, but until Python 3.4 was not
    # formally specified, so may not exist on some platforms
    name: str

    def __init__(self, data: _DataType = ...) -> None: ...

    def copy(self) -> _Hash: ...
    def digest(self) -> bytes: ...
    def hexdigest(self) -> str: ...
    def update(self, arg: _DataType) -> None: ...

def md5(arg: _DataType = ...) -> _Hash: ...
def sha1(arg: _DataType = ...) -> _Hash: ...
def sha224(arg: _DataType = ...) -> _Hash: ...
def sha256(arg: _DataType = ...) -> _Hash: ...
def sha384(arg: _DataType = ...) -> _Hash: ...
def sha512(arg: _DataType = ...) -> _Hash: ...

def new(name: str, data: _DataType = ...) -> _Hash: ...

algorithms_guaranteed: AbstractSet[str]
algorithms_available: AbstractSet[str]

def pbkdf2_hmac(hash_name: str, password: _DataType, salt: _DataType, iterations: int, dklen: Optional[int] = ...) -> bytes: ...

if sys.version_info >= (3, 6):
    class _VarLenHash(object):
        digest_size: int
        block_size: int
        name: str

        def __init__(self, data: _DataType = ...) -> None: ...

        def copy(self) -> _VarLenHash: ...
        def digest(self, length: int) -> bytes: ...
        def hexdigest(self, length: int) -> str: ...
        def update(self, arg: _DataType) -> None: ...

    sha3_224 = _Hash
    sha3_256 = _Hash
    sha3_384 = _Hash
    sha3_512 = _Hash
    shake_128 = _VarLenHash
    shake_256 = _VarLenHash

    def scrypt(password: _DataType, *, salt: _DataType, n: int, r: int, p: int, maxmem: int = ..., dklen: int = ...) -> bytes: ...

    class _BlakeHash(_Hash):
        MAX_DIGEST_SIZE: int
        MAX_KEY_SIZE: int
        PERSON_SIZE: int
        SALT_SIZE: int

        def __init__(self, data: _DataType = ..., digest_size: int = ..., key: _DataType = ..., salt: _DataType = ..., person: _DataType = ..., fanout: int = ..., depth: int = ..., leaf_size: int = ..., node_offset: int = ..., node_depth: int = ..., inner_size: int = ..., last_node: bool = ...) -> None: ...

    blake2b = _BlakeHash
    blake2s = _BlakeHash

在这行代码 hashed = hmac.new(key, raw, sha1) 中,sha1 是从哪里来的? - Rolv Apneseth
它来自包含以下代码行的HashLib库等:def md5(arg: _DataType = ...) -> _Hash: ... def sha1(arg: _DataType = ...) -> _Hash: ... def sha224(arg: _DataType = ...) -> _Hash: ... def sha256(arg: _DataType = ...) -> _Hash: ... def sha384(arg: _DataType = ...) -> _Hash: ... def sha512(arg: _DataType = ...) -> _Hash: ...我在帖子中添加了一个编辑,这样您就可以看到Hashlib包含什么内容 :) - Christian Aaby Høgh
1
我已经点赞了,希望有更多知识的人能够帮助你。还有一个需要澄清的问题,当您说您尝试用sha256替换值时它没有起作用,您是用什么替换,它是以什么方式失败的?如果它给出了一个错误,请提供回溯信息。 - Rolv Apneseth
我刚刚将名为SHA1的变量替换为SHA256,我并没有期望它能够工作,但我还是尝试了一下,因为你永远不知道...结果我得到了一个400错误。 - Christian Aaby Høgh
1个回答

1
已经有Haslib文件中的sha256()函数了,因此你可以尝试将类SignatureMethod_HMAC_SHA256添加到oauth文件中,它可以类似于SHA1。

只需像这样更改hmac.new()函数的参数:

hashed = hmac.new(key, raw, sha256)

整个类可以像这样:

...

class SignatureMethod_HMAC_SHA256(SignatureMethod):
    name = 'HMAC-SHA256'

    def signing_base(self, request, consumer, token):
        if (not hasattr(request, 'normalized_url') or request.normalized_url is None):
            raise ValueError("Base URL for request is not set.")

        sig = (
            escape(request.method),
            escape(request.normalized_url),
            escape(request.get_normalized_parameters()),
        )

        key = '%s&' % escape(consumer.secret)
        if token:
            key += escape(token.secret)
        raw = '&'.join(sig)
        return key.encode('ascii'), raw.encode('ascii')

    def sign(self, request, consumer, token):
        """Builds the base signature string."""
        key, raw = self.signing_base(request, consumer, token)

        hashed = hmac.new(key, raw, sha256)

        # Calculate the digest base 64.
        return binascii.b2a_base64(hashed.digest())[:-1]

然后您可以在脚本中调用新的SHA256方法,而不是已弃用的SHA1方法:
signature_method = oauth.SignatureMethod_HMAC_SHA256()

我按照你提到的所有事情都做了,但无法获得所需的输出。出现以下错误 - NameError: name 'SignatureMethod_HMAC_SHA256' is not defined - Sayeesh
可能您没有正确导入 OAuth,或者您将该类粘贴到了错误的文件中,而这个文件并没有正确连接。 - vhaska

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接