diff --git a/alb-s3-sqs-es/alb_s3/ingester/ingester.py b/alb-s3-sqs-es/alb_s3/ingester/ingester.py index ff648ef..680274d 100644 --- a/alb-s3-sqs-es/alb_s3/ingester/ingester.py +++ b/alb-s3-sqs-es/alb_s3/ingester/ingester.py @@ -2,8 +2,8 @@ import os from time import sleep -from ingester.sink.elasticsearch import ElasticSink -from ingester.source import S3Source +from alb_s3.ingester.sink.elasticsearch import ElasticSink +from alb_s3.ingester.source import S3Source class Ingester: diff --git a/alb-s3-sqs-es/alb_s3/run.py b/alb-s3-sqs-es/alb_s3/run.py index 4efa2f7..3a4544d 100755 --- a/alb-s3-sqs-es/alb_s3/run.py +++ b/alb-s3-sqs-es/alb_s3/run.py @@ -2,7 +2,7 @@ import sentry_sdk from sentry_sdk.integrations.aws_lambda import AwsLambdaIntegration -from .ingester.ingester import Ingester as i +from alb_s3.ingester.ingester import Ingester sentry_sdk.init(integrations=[AwsLambdaIntegration()], traces_sample_rate=1.0) @@ -18,8 +18,8 @@ def lambda_handler(event, context): bucket_name = record_event["s3"]["bucket"]["name"] key = record_event["s3"]["object"]["key"] - (sinks) = i.from_environment() - ingester = i(sinks=sinks, bucket_name=bucket_name, key=key) + (sinks) = Ingester.from_environment() + ingester = Ingester(sinks=sinks, bucket_name=bucket_name, key=key) ingester.run() return "complete"