darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-11 09:27:56 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix migrations

Browse Source
This commit is contained in:
jimmoffet
2022-10-04 17:42:04 -07:00
parent 434b7b2d08
commit 4feaa06f5d
4 changed files with 20 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
app/notifications/sns_cert_validator.py
View File

@@ -10,8 +10,7 @@ import six
from app import redis_store
from app.config import Config
USE_CACHE = True
VALIDATE_ARN = True
VALIDATE_SNS_TOPICS = Config.VALIDATE_SNS_TOPICS
VALID_SNS_TOPICS = Config.VALID_SNS_TOPICS
@@ -27,19 +26,16 @@ class ValidationError(Exception):
def get_certificate(url):
if USE_CACHE:
res = redis_store.get(url)
if res is not None:
return res
res = requests.get(url).text
redis_store.set(url, res, ex=60 * 60) # 60 minutes
res = redis_store.get(url)
if res is not None:
return res
else:
return requests.get(url).text
res = requests.get(url).text
redis_store.set(url, res, ex=60 * 60) # 60 minutes
return res
def validate_arn(sns_payload):
if VALIDATE_ARN:
if VALIDATE_SNS_TOPICS:
arn = sns_payload.get('TopicArn')
topic_name = arn.split(':')[5]
if topic_name not in VALID_SNS_TOPICS: