diff --git a/gcn_email/cli.py b/gcn_email/cli.py index 5b489c8..a0f826a 100644 --- a/gcn_email/cli.py +++ b/gcn_email/cli.py @@ -11,7 +11,7 @@ import click import prometheus_client -from .core import connect_as_consumer, subscribe_to_topics, recieve_alerts +from .core import connect_as_consumer, recieve_alerts, subscribe_to_topics log = logging.getLogger(__name__) diff --git a/gcn_email/core.py b/gcn_email/core.py index 141152e..2d6aa25 100644 --- a/gcn_email/core.py +++ b/gcn_email/core.py @@ -5,19 +5,19 @@ # # SPDX-License-Identifier: Apache-2.0 # -from email.message import EmailMessage +import json import logging import os -import json +from email.message import EmailMessage +import backoff import boto3 from boto3.dynamodb.conditions import Key from gcn_kafka import Consumer, config_from_env -from ratelimit import limits, RateLimitException -import backoff +from ratelimit import RateLimitException, limits -from .helpers import periodic_task from . import metrics +from .helpers import periodic_task SESV2 = boto3.client("sesv2") SENDER = f'GCN Notices <{os.environ["EMAIL_SENDER"]}>' diff --git a/gcn_email/metrics.py b/gcn_email/metrics.py index 07e94ac..b42b971 100644 --- a/gcn_email/metrics.py +++ b/gcn_email/metrics.py @@ -6,6 +6,7 @@ # SPDX-License-Identifier: Apache-2.0 # """Prometheus metrics.""" + import prometheus_client received = prometheus_client.Counter(