Перенёс схемы в модели

src/fastapi_app/lib/api/schemas/__init__.py

@@ -1,6 +0,0 @@
-from .broker_message import *
-
-__all__ = [
-    "BrokerMessagePayload",
-    "BrokerMessage",
-]

src/fastapi_app/lib/db/brokers/base_broker.py

@@ -1,8 +1,7 @@
 import abc
-import contextlib
 import typing
 
-import lib.api.schemas as api_schemas
+import lib.models.broker_message as lib_models_broker_message
 
 T = typing.TypeVar("T", bound="BasePublisher")
 
@@ -17,5 +16,5 @@ class BasePublisher(abc.ABC):
         pass
 
     @abc.abstractmethod
-    async def publish_message(self, message_body: api_schemas.broker_message.BrokerMessage, routing_key: str) -> None:
+    async def publish_message(self, message_body: lib_models_broker_message.BrokerMessage, routing_key: str) -> None:
         pass

src/fastapi_app/lib/db/brokers/broker.py

@@ -1,7 +1,6 @@
-import contextlib
 import typing
 
-import lib.api.schemas as api_schemas
+import lib.models.broker_message as lib_models_broker_message
 
 
 class BrokerPublisher:
@@ -14,5 +13,5 @@ class BrokerPublisher:
     async def dispose(self):
         await self.broker.dispose()
 
-    async def publish_message(self, message_body: api_schemas.BrokerMessage, routing_key: str):
+    async def publish_message(self, message_body: lib_models_broker_message.BrokerMessage, routing_key: str):
         await self.broker.publish_message(message_body, routing_key)

src/fastapi_app/lib/db/brokers/rabbitmq.py

@@ -5,9 +5,9 @@ import logging
 
 import aio_pika
 
-import lib.api.schemas as api_schemas
 import lib.app.split_settings as app_split_settings
 import lib.db.brokers as db_brokers
+import lib.models.broker_message as models_broker_message
 
 
 class RabbitMQPublisher(db_brokers.base_broker.BasePublisher):
@@ -33,6 +33,7 @@ class RabbitMQPublisher(db_brokers.base_broker.BasePublisher):
     async def dispose(self):
         try:
             while not self.pool.empty():
+                print("Closing channel")
                 channel = await self.pool.get()
                 await channel.close()
 
@@ -57,7 +58,7 @@ class RabbitMQPublisher(db_brokers.base_broker.BasePublisher):
         finally:
             await self.pool.put(channel)
 
-    async def publish_message(self, message_body: api_schemas.BrokerMessage, routing_key: str):
+    async def publish_message(self, message_body: models_broker_message.BrokerMessage, routing_key: str):
         try:
             async with self.__get_channel() as channel:
                 message = aio_pika.Message(content_type="application/json", body=json.dumps(message_body).encode())

src/fastapi_app/lib/models/__init__.py

@@ -1,3 +1,8 @@
 from .base_sqlalchemy import Base
+from .broker_message import *
 
-__all__ = ["Base"]
+__all__ = [
+    "Base",
+    "BrokerMessagePayload",
+    "BrokerMessage",
+]