invoices, order handling, db updates

main.py

@@ -11,6 +11,7 @@ from middlewares.di import DIMiddleware
 from middlewares.repository import RepositoryMiddleware
 from middlewares.session import DBSessionMiddleware
 from repositories.categories import CategoriesRepository
+from repositories.invoices import InvoiceRepository
 from repositories.order_items import OrderItemRepository
 from repositories.orders import OrderRepository
 from repositories.products import ProductRepository
@@ -38,26 +39,18 @@ async def main():
 
     product_repository = ProductRepository()
 
-    # DI Middleware Init
-    dp.message.middleware(DIMiddleware(order_service))
-    dp.message.middleware(DBSessionMiddleware(async_session))
-    dp.message.middleware(
-        RepositoryMiddleware(
-            order_repository,
-            category_repository,
-            product_repository,
-            order_items_repository,
-        )
-    )
+    invoice_repository = InvoiceRepository()
 
-    dp.callback_query.middleware(DIMiddleware(order_service))
-    dp.callback_query.middleware(DBSessionMiddleware(async_session))
-    dp.callback_query.middleware(
+    # DI Middleware Init
+    dp.update.middleware(DIMiddleware(order_service))
+    dp.update.middleware(DBSessionMiddleware(async_session))
+    dp.update.middleware(
         RepositoryMiddleware(
             order_repository,
             category_repository,
             product_repository,
             order_items_repository,
+            invoice_repository,
         )
     )