Adjust __all__ imports

saritasa-nest · Jul 4, 2024 · 6da68c9 · 6da68c9
1 parent 5ad1e41
commit 6da68c9
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 0 deletions.
diff --git a/sns_sqs_communicator/clients/__init__.py b/sns_sqs_communicator/clients/__init__.py
@@ -1,2 +1,9 @@
 from .sns import SNSClient, get_boto3_sns_client
 from .sqs import SQSClient, get_boto3_sqs_client
+
+__all__ = (
+    "SQSClient",
+    "get_boto3_sqs_client",
+    "SNSClient",
+    "get_boto3_sns_client",
+)
diff --git a/sns_sqs_communicator/parsers/__init__.py b/sns_sqs_communicator/parsers/__init__.py
@@ -1,3 +1,9 @@
 from .protocol import ParserProtocol
 from .sns import SNSParser
 from .sqs import SQSParser
+
+__all__ = (
+    "ParserProtocol",
+    "SNSParser",
+    "SQSParser",
+)
diff --git a/sns_sqs_communicator/processing/__init__.py b/sns_sqs_communicator/processing/__init__.py
@@ -6,3 +6,12 @@
     Processor,
     ProcessorT,
 )
+
+__all__ = (
+    "CancelProcessingError",
+    "ProcessingResult",
+    "ProcessingResultReturnT",
+    "ProcessingResultStatus",
+    "Processor",
+    "ProcessorT",
+)
diff --git a/sns_sqs_communicator/sentry/__init__.py b/sns_sqs_communicator/sentry/__init__.py
@@ -1 +1,3 @@
 from .processing import SentryProcessor
+
+__all__ = ("SentryProcessor",)
diff --git a/sns_sqs_communicator/testing/__init__.py b/sns_sqs_communicator/testing/__init__.py
@@ -7,3 +7,13 @@
     topic_factory,
 )
 from .worker import TestWorker
+
+__all__ = (
+    "check_messages_in_queue",
+    "push_and_pull_canceled_result",
+    "push_and_pull_failed_result",
+    "push_and_pull_successful_result",
+    "queue_factory",
+    "topic_factory",
+    "TestWorker",
+)
Original file line number	Diff line number	Diff line change
		@@ -1 +1,3 @@
		from .processing import SentryProcessor

		__all__ = ("SentryProcessor",)