在python中用类定义变量时，如何使用self

Question

我正在学习使用函数，现在学习python中的类。如果函数不在类中，我创建的函数可以工作，但现在我的目标是为它们创建一个类。看起来我不明白self是如何工作的。我已经将它添加为每个函数的参数，但是在调用函数时遇到了NameError: name 'variable name' is not defined消息。我错过了什么

以下是正在进行的脚本myTokClassTst.py代码：

import time
import json
import jwt
import requests
from lib.jwksutils import rsa_pem_from_jwk

class TokenMgr():
    """An attempt to model a Token Manager"""

    def __init__(self):
        """Initialize token attributes as needed"""
        pass

    def get_kid(self, token):
        headers = jwt.get_unverified_header(token)
        if not headers:
            raise InvalidAuthorizationToken('missing headers')
        try:
            return headers['kid']
        except KeyError:
            raise InvalidAuthorizationToken('missing kid')


    def get_jwk(self, kid):
        with open('testkeys/jwks-keys', 'r') as az:
            jwks = json.load(az)
        for jwk in jwks.get('keys'):
            if jwk.get('kid') == kid:
                print ('This is the jwk:', jwk)
                return jwk
        raise InvalidAuthorizationToken('kid not recognized')


    def get_public_key(self, token):
        print ('Response from get_public_token:', rsa_pem_from_jwk(get_jwk(get_kid(token))) )
        return rsa_pem_from_jwk(get_jwk(get_kid(token)))


    def validate_jwt(self, jwt_to_validate):
        public_key = get_public_key(jwt_to_validate)
        #Hard coding these for now
        valid_audiences = ['abc123aa86-1f3a-4774-8bae-2704bff9f797'] 
        issuer = 'https://login.microsoftonline.com/99c77bbe-8598-4b85-9e51-1ca753fa50f2/v2.0'
        decoded = jwt.decode(jwt_to_validate,
                            public_key,
                            verify=True,
                            algorithms=['RS256'],
                            audience=valid_audiences,
                            issuer=issuer)
        print('This is decoded:', decoded)

我在python shell中这样调用函数：

$ python
Python 3.7.2 (tags/v3.7.2:9a3ffc0492, Dec 23 2018, 23:09:28) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import myTokClassTst as mytok
>>> idval = mytok.TokenMgr()
>>> idval.validate_jwt('eyJ0eXAiOiJKV1QiLCJhbGciOiJSUA')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "C:\myTokClassTst.py", line 97, in validate_jwt
    public_key = get_public_key(jwt_to_validate)
NameError: name 'get_public_key' is not defined

Answer 1

使用self.get_public_key，因为您试图调用相同的类方法。 与get_jwk和get_kid相同

最终代码：

import time
import json
import jwt
import requests
from lib.jwksutils import rsa_pem_from_jwk

class TokenMgr():
    """An attempt to model a Token Manager"""

    def __init__(self):
        """Initialize token attributes as needed"""
        pass

    def get_kid(self, token):
        headers = jwt.get_unverified_header(token)
        if not headers:
            raise InvalidAuthorizationToken('missing headers')
        try:
            return headers['kid']
        except KeyError:
            raise InvalidAuthorizationToken('missing kid')


    def get_jwk(self, kid):
        with open('testkeys/jwks-keys', 'r') as az:
            jwks = json.load(az)
        for jwk in jwks.get('keys'):
            if jwk.get('kid') == kid:
                print ('This is the jwk:', jwk)
                return jwk
        raise InvalidAuthorizationToken('kid not recognized')


    def get_public_key(self, token):
        print ('Response from get_public_token:', rsa_pem_from_jwk(self.get_jwk(self.get_kid(token))) )
        return rsa_pem_from_jwk(self.get_jwk(self.get_kid(token)))


    def validate_jwt(self, jwt_to_validate):
        public_key = self.get_public_key(jwt_to_validate)
        #Hard coding these for now
        valid_audiences = ['abc123aa86-1f3a-4774-8bae-2704bff9f797'] 
        issuer = 'https://login.microsoftonline.com/99c77bbe-8598-4b85-9e51-1ca753fa50f2/v2.0'
        decoded = jwt.decode(jwt_to_validate,
                            public_key,
                            verify=True,
                            algorithms=['RS256'],
                            audience=valid_audiences,
                            issuer=issuer)
        print('This is decoded:', decoded)

Answer 2

get_public_key是一个实例方法。所以你需要这样称呼它：

self.get_public_key

其中self指的是类的实例（如该关键字）