fix accepting all Vapid key instances (#137) (#139)

confirmed source change
This commit is contained in:
Martin Plicka 2021-07-27 20:37:12 +02:00 committed by GitHub
parent a2dbc6f0f1
commit 248cd434ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -19,7 +19,7 @@ import requests
from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.asymmetric import ec
from cryptography.hazmat.primitives import serialization
from py_vapid import Vapid
from py_vapid import Vapid, Vapid01
class WebPushException(Exception):
@ -458,7 +458,7 @@ def webpush(subscription_info,
vapid_claims['exp']))
if not vapid_private_key:
raise WebPushException("VAPID dict missing 'private_key'")
if isinstance(vapid_private_key, Vapid):
if isinstance(vapid_private_key, Vapid01):
vv = vapid_private_key
elif os.path.isfile(vapid_private_key):
# Presume that key from file is handled correctly by