no message

src/mail_order_bot/abcp_api/abcp_provider.py

@@ -1,7 +1,9 @@
 import os
 import hashlib
 import requests
+import logging
 
+logger = logging.getLogger(__name__)
 
 class AbcpProvider:
         HOST = "https://id23089.public.api.abcp.ru"
@@ -11,25 +13,34 @@ class AbcpProvider:
             "Content-Type": "application/x-www-form-urlencoded"
         }
 
-        def __init__(self):
+        def __init__(self, account="SYSTEM"):
             self.base_url = self.HOST
-            self.login = os.getenv("ABCP_LOGIN")
-            password = os.getenv("ABCP_PASSWORD")
-            self.password = hashlib.md5(password.encode("utf-8")).hexdigest()
 
-        def get_stock(self, order):
+        def get_stock(self, sku, manufacturer, partner="SYSTEM"):
             method = "GET"
             path = "/search/articles"
 
-            for position in order.positions:
-                params = {"number": position.sku, "brand": position.manufacturer, "withOutAnalogs": "1"}
-                position.stock = self._execute(path, method, params)
+            params = {"number": sku, "brand": manufacturer, "withOutAnalogs": "1"}
+            return self._execute(partner, path, method, params)
+
+        def _execute(self, partner, path, method="GET", params={}, data=None, ):
+            params["userlogin"] = os.getenv(f"ABCP_LOGIN_{partner}")
+            params["userpsw"] = hashlib.md5(os.getenv(f"ABCP_PASSWORD_{partner}").encode("utf-8")).hexdigest()
 
-        def _execute(self, path, method="GET", params={}, data=None):
-            params["userlogin"] = self.login
-            params["userpsw"] = self.password
             response = requests.request(method, self.HOST+path, data=data, headers=self.HEADERS, params=params)
-            if response.status_code != 200:
-                raise Exception(response.text)
-            return response.json()
+            payload = response.json()
+            if response.status_code == 200:
+                logger.debug(f"Получены данные об остатках на складе")
+                result = {
+                    "success": True,
+                    "data": payload
+                }
+            else:
+                logger.warning(f"ошибка получения данных об остатках на складе: {payload}")
+
+                result = {
+                    "success": False,
+                    "error": payload
+                }
+            return result
 

src/mail_order_bot/config.yml

@@ -1,8 +1,8 @@
 # Настройки обработки  =================================================================
 
 # Раздел с общими конфигурационными параметрами ===============================
-update_interval: 10
-work_interval: 30
+update_interval: 1
+work_interval: 60
 email_dir: "spareparts"
 
 # Логирование =================================================================
@@ -44,7 +44,7 @@ log:
   loggers:
     '':
       handlers: [console, file, telegram]
-      level: INFO
+      level: DEBUG
       propagate: False
 
     __main__:

src/mail_order_bot/configs/amtel.club.yml

@@ -0,0 +1,30 @@
+pipeline:
+  # Настраиваем парсинг экселя
+  - handler: BasicExcelParser
+    config:
+      sheet_name: 0
+      key_field: "Номер"
+      mapping:
+        article: "Номер"
+        manufacturer: "Фирма"
+        name: "Наименование"
+        price: "Цена"
+        quantity: "Кол-во"
+        total: "Сумма"
+
+  # Определяем логику обработки заказа (в данном случае все с локального склада)
+  - handler: DeliveryPeriodLocalStore
+
+  # Запрос остатков со склада
+  - handler: GetStock
+
+
+
+
+
+
+
+
+
+
+

src/mail_order_bot/configs/todx.ru.yml

@@ -1,6 +1,5 @@
 pipeline:
-  - handler: "ConfigurableExcelParser"
-    result_section: "positions"
+  - handler: BasicExcelParser
     config:
       sheet_name: 0
       key_field: "Код детали"
@@ -12,6 +11,9 @@ pipeline:
         quantity: "Кол-\nво"
         total: "Сумма"
 
+  - handler: GetStock
+
+
 
 
 

src/mail_order_bot/context.py

@@ -1,32 +1,78 @@
 import threading
 from typing import Any, Dict
+import logging
 
-class _SingletonMeta(type):
+logger = logging.getLogger()
+
+import threading
+from typing import Any
+
+class SingletonMeta(type):
     _instances = {}
-    _lock = threading.Lock()
 
     def __call__(cls, *args, **kwargs):
-        if cls not in cls._instances:
-            with cls._lock:
         if cls not in cls._instances:
             instance = super().__call__(*args, **kwargs)
             cls._instances[cls] = instance
         return cls._instances[cls]
 
-class Context(metaclass=_SingletonMeta):
+
+
+class Context2(metaclass=SingletonMeta):
     def __init__(self):
-        # будет вызван только при первом создании
-        self.context = {}
+        if not hasattr(self, 'initialized'):
+            self.data = {}
             self.email_client = None
+            self.initialized = True
+            logger.debug(f"Context создан {id}")  # опциональный лог
 
-    def clear_context(self):
+
+
+        # будет вызван только при первом создании
+
+
+    def clear(self):
         """Очищает self.context, устанавливая его в None или пустой словарь"""
-        with self._lock:  # потокобезопасная очистка
-            self.context = {}
-            print("Context очищен")  # опциональный лог
+        self.data = {}
+        logger.debug("Context очищен")  # опциональный лог
 
-    def set_context(self, new_context: Dict[str, Any]):
+    def set(self, new_context: Dict[str, Any]):
+        """Устанавливает новый контекст (бонусный метод)"""
+        self.data = new_context
+        logger.debug("Новый контекст установлен")
+
+
+class ThreadSafeSingletonMeta(type):
+    _instances = {}
+    _lock = threading.Lock()
+
+    def __call__(cls, *args: Any, **kwargs: Any) -> Any:
+        if cls not in cls._instances:
+            with cls._lock:
+                if cls not in cls._instances:
+                    # Инициализация ТУТ, не в __init__
+                    instance = super().__call__(*args, **kwargs)
+                    instance.data = {}
+                    instance.email_client = None
+                    instance._lock = threading.RLock()
+                    cls._instances[cls] = instance
+
+        return cls._instances[cls]
+
+
+class Context(metaclass=ThreadSafeSingletonMeta):
+    def __init__(self):
+        print(f"Context: {id(self)}, поток {threading.get_ident()}")
+        # будет вызван только при первом создании
+
+    def clear(self):
+        """Очищает self.context, устанавливая его в None или пустой словарь"""
+        with self._lock:
+            self.data = {}
+            logger.debug("Context очищен")  # опциональный лог
+
+    def set(self, new_context: Dict[str, Any]):
         """Устанавливает новый контекст (бонусный метод)"""
         with self._lock:
-            self.context = new_context
-            print("Новый контекст установлен")
+            self.data = new_context
+            logger.debug("Новый контекст установлен")

src/mail_order_bot/email_client/utils.py

@@ -1,3 +1,4 @@
+from email.header import decode_header, make_header
 import re
 from datetime import datetime
 from typing import List, Optional
@@ -11,33 +12,28 @@ from email.header import decode_header
 import imaplib
 import smtplib
 
+from .objects import EmailMessage, EmailAttachment
 
 
-class EmailMagic:
-    def __init__(self, email):
-        self.email = email
-
-    def _decode_header(self, header_value: str) -> str:
+class EmailUtils:
+    @staticmethod
+    def extract_header(msg, header_name) -> str:
         """Декодировать заголовок письма."""
-        if header_value is None:
+        header = msg.get(header_name, "")
+        if header is None:
             return ""
+        decoded = decode_header(header)
+        return str(make_header(decoded))
 
-        decoded_parts = []
-        for part, encoding in decode_header(header_value):
-            if isinstance(part, bytes):
-                if encoding:
-                    try:
-                        decoded_parts.append(part.decode(encoding))
-                    except:
-                        decoded_parts.append(part.decode('utf-8', errors='ignore'))
-                else:
-                    decoded_parts.append(part.decode('utf-8', errors='ignore'))
-            else:
-                decoded_parts.append(str(part))
+    @staticmethod
+    def extract_email(text) -> str:
+        match = re.search(r'<([^<>]+)>', text)
+        if match:
+            return match.group(1)
+        return None
 
-        return ''.join(decoded_parts)
-
-    def _extract_body(self, msg: email.message.Message) -> str:
+    @staticmethod
+    def extract_body(msg: email.message.Message) -> str:
         """Извлечь текст письма из любого типа содержимого, кроме вложений"""
         body = ""
         if msg.is_multipart():
@@ -65,13 +61,8 @@ class EmailMagic:
 
         return body
 
-    def __extract_email(self, text: str) -> str:
-        match = re.search(r'<([^<>]+)>', text)
-        if match:
-            return match.group(1)
-        return None
-
-    def _extract_first_sender(self, body: str):
+    @staticmethod
+    def extract_first_sender(body: str):
         """Извлекает адреса отправителей из пересылаемого сообщения. Нужно для отладки"""
         # Ищем email внутри скобок после строки "Пересылаемое сообщение"
         pattern = r"Пересылаемое сообщение.*?\((.*?)\)"
@@ -80,7 +71,8 @@ class EmailMagic:
             return match.group(1)
         return None
 
-    def _extract_attachments(self, msg: email.message.Message) -> List[EmailAttachment]:
+    @staticmethod
+    def extract_attachments(msg: email.message.Message) -> List[EmailAttachment]:
         """Извлечь вложения из письма."""
         attachments = []
 
@@ -91,9 +83,24 @@ class EmailMagic:
                 filename = part.get_filename()
                 if filename:
                     # Декодируем имя файла
-                    filename = self._decode_header(filename)
+                    filename = decode_header(filename)[0]
                     # Получаем содержимое
                     content = part.get_payload(decode=True)
                     if content:
-                        attachments.append(EmailAttachment(filename=filename, content=content))
+                        #attachments.append(EmailAttachment(filename=filename, content=content))
+                        attachments.append({"name": filename, "bytes": content})
+
         return attachments
+
+    @staticmethod
+    def extract_domain(email_message: str) -> str | None:
+        """Вернуть домен из email либо None, если формат странный."""
+        if "@" not in email_message:
+            return None
+        # убираем пробелы по краям и берём часть после '@'
+        return email_message.strip().split("@", 1)[1]
+
+
+
+
+

src/mail_order_bot/email_processor/handlers/__init__.py

@@ -1,9 +1,16 @@
-from .abcp_clients.check_stock import GetStock
-from .abcp_clients.create_order import InstantOrderTest
-
+from .attachment_handler.attachment_handler import AttachmentHandler
 from .excel_parcers.order_parcer_basic import BasicExcelParser
+from .destination_time.local_store import DeliveryPeriodLocalStore
+
+
+
+from .abcp.api_get_stock import GetStock
+from .abcp.api_create_order import InstantOrderTest
+
+
 
 from .notifications.test_notifier import TestNotifier
 
 
-from .validators.price_quantity_ckecker import CheckOrder
+
+

src/mail_order_bot/email_processor/handlers/abcp/api_get_stock.py

@@ -0,0 +1,26 @@
+import random
+import logging
+
+from mail_order_bot.email_processor.handlers.abstract_task import AbstractTask
+from mail_order_bot.abcp_api.abcp_provider import AbcpProvider
+
+logger = logging.getLogger(__name__)
+
+
+class APIGetStock(AbstractTask):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.abcp_provider = AbcpProvider()
+
+    def do(self) -> None:
+        attachments = self.context.data.get("attachments", [])
+        for attachment in attachments:
+            order = attachment["order"]
+            for position in order.positions:
+                stock = self.get_stock(position.sku, position.manufacturer)
+                position.update_stock(stock, order.delivery_period)
+                position.fill_from_stock()
+            logger.info(f"Получены позиции со склада для файла {attachment.get('name', "no name")}")
+
+    def get_stock(self, sku: str, manufacturer: str) -> int:
+        return self.abcp_provider.get_stock(sku, manufacturer)

src/mail_order_bot/email_processor/handlers/abcp_clients/check_stock.py

@@ -1,27 +0,0 @@
-import random
-import logging
-
-from mail_order_bot.email_processor.handlers.abstract_task import AbstractTask
-
-
-logger = logging.getLogger(__name__)
-
-
-def get_stock(brand, part_number):
-    return random.randint(0, 10)
-
-class GetStock(AbstractTask):
-
-    def do(self) -> None:
-        positions = self.order.positions
-        for position in positions:
-            self._update_stock(position)
-
-    def _update_stock(self, position):
-        # Эмулируем получение данных
-        max_stock = self.config.get('max_stock',10)
-        stock = random.randint(0, max_stock)
-        price = position.requested_price
-
-        position.stock_price = price
-        position.stock_quantity = stock

src/mail_order_bot/email_processor/handlers/abstract_task.py

@@ -4,13 +4,13 @@ from typing import Dict, Any
 from mail_order_bot.context import Context
 
 
-class AbstractTask(ABC, Context):
+class AbstractTask():
     RESULT_SECTION = "section"
     """
     Абстрактный базовый класс для всех хэндлеров.
     """
-    def __init__(self, config: Dict[str, Any]) -> None:
-        Context.__init__(self, {})
+    def __init__(self, config: Dict[str, Any]={}) -> None:
+        self.context = Context()
         self.config = config
 
     @abstractmethod

src/mail_order_bot/email_processor/handlers/attachment_handler/attachment_handler.py

@@ -0,0 +1,21 @@
+from mail_order_bot.email_processor.handlers.abstract_task import AbstractTask
+from mail_order_bot.email_client.utils import EmailUtils
+
+import logging
+
+logger = logging.getLogger(__name__)
+
+class AttachmentHandler(AbstractTask):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def do(self) -> None:
+        email = self.context.data["email"]
+        attachments = EmailUtils.extract_attachments(email)
+        self.context.data["attachments"] = attachments
+        logger.debug(f"AttachmentHandler отработал, извлек вложений: {len(attachments)} ")
+
+
+
+
+

src/mail_order_bot/email_processor/handlers/destination_time/local_store.py

@@ -0,0 +1,21 @@
+from mail_order_bot.email_processor.handlers.abstract_task import AbstractTask
+from mail_order_bot.email_client.utils import EmailUtils
+
+import logging
+
+logger = logging.getLogger(__name__)
+
+class DeliveryPeriodLocalStore(AbstractTask):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def do(self) -> None:
+        attachments = self.context.data["attachments"]
+        for attachment in attachments:
+            order = attachment["order"]
+            order.set_delivery_period(0)
+        logger.info(f"Доставка только с локального склада, срок 1 день.")
+
+
+
+

src/mail_order_bot/email_processor/handlers/excel_parcers/order_parcer_basic.py

@@ -7,46 +7,46 @@ from io import BytesIO
 from mail_order_bot.email_processor.handlers.abstract_task import AbstractTask
 
 from ...order.auto_part_position import AutoPartPosition
+from ...order.auto_part_order import AutoPartOrder
 
 logger = logging.getLogger(__name__)
 
 
 class BasicExcelParser(AbstractTask):
-    RESULT_SECTION = "positions"
     """
     Универсальный парсер, настраиваемый через конфигурацию.
     Подходит для большинства стандартных случаев.
     """
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
 
     def do(self) -> None:
 
         # todo сделать проверку на наличие файла и его тип
-        file_bytes = BytesIO(self.context.get("attachment").content) # self.context.get("attachment") #
+
+        attachments = self.context.data.get("attachments", [])
+        for attachment in attachments:
+            file_bytes = BytesIO(attachment['bytes']) # self.context.get("attachment") #
             try:
                 df = self._make_dataframe(file_bytes)
-            # Получаем маппинг колонок из конфигурации
                 mapping = self.config['mapping']
+                order = AutoPartOrder()
 
                 # Парсим строки
                 positions = []
                 for idx, row in df.iterrows():
-                try:
                     position = self._parse_row(row, mapping)
                     if position:
-                        positions.append(position)
-                        self.order.add_position(position)
-                except Exception as e:
-                    logger.error(f"Ошибка парсинга строки {idx}: {e}, {row}")
-                    continue
-
-            logger.info(f"Успешно обработано {len(positions)} позиций из {len(df)} строк")
-
-            self.context[self.RESULT_SECTION] = positions
+                        order.add_position(position)
 
+                logger.info(f"Успешно обработано {len(order)} позиций из {len(df)} строк")
 
             except Exception as e:
                 logger.error(f"Ошибка при обработке файла: {e}")
-            raise Exception from e
+            else:
+                attachment["order"] = order
+
+
 
     def _parse_row(self, row: pd.Series, mapping: Dict[str, str]) -> Optional[AutoPartPosition]:
         """Парсит одну строку Excel в OrderPosition"""

src/mail_order_bot/email_processor/handlers/logic/order_acceptance_criteria.py

@@ -0,0 +1,24 @@
+import random
+import logging
+from mail_order_bot.email_processor.handlers.abstract_task import AbstractTask
+from mail_order_bot.email_processor.order.auto_part_order import OrderStatus
+from mail_order_bot.email_processor.order.auto_part_position import AutoPartPosition, PositionStatus
+from decimal import Decimal
+import random
+logger = logging.getLogger(__name__)
+
+
+class LocalStoreOrder(AbstractTask):
+    """Сейчас логика такая
+     - ищем на складе наш сапплиер код, берем самую дешевую позицию и делаем заказ из нее
+
+     Другие чуть более дорогие не рассматриваем
+
+    """
+      # это код нашего склада
+
+    def do(self) -> None:
+        attachments = self.context.data["attachments"]
+        for attachment in attachments:
+            order = attachment["order"]
+

src/mail_order_bot/email_processor/handlers/validators/price_quantity_ckecker.py

@@ -1,50 +0,0 @@
-import random
-import logging
-from mail_order_bot.email_processor.handlers.abstract_task import AbstractTask
-from mail_order_bot.email_processor.order.auto_part_order import OrderStatus
-from decimal import Decimal
-import random
-logger = logging.getLogger(__name__)
-
-
-class CheckOrder(AbstractTask):
-
-    def do(self) -> None:
-        refused = 0
-        positions = self.order.positions
-        for position in positions:
-            self._set_order_price(position)
-            self._set_order_quantity(position)
-
-            if position.order_price == 0 or position.order_quantity == 0:
-                refused += 1
-
-        self._check_refusal_threshold(refused)
-
-
-    def _set_order_price(self, position):
-        # Эмулируем получение данных
-        acceptable_price_reduction = self.config.get("acceptable_price_reduction")
-        acceptable_price = position.stock_price* Decimal(str((1-acceptable_price_reduction/100)))
-
-        if  position.requested_price < acceptable_price:
-            position.order_price = 0
-        else:
-            position.order_price = position.requested_price
-
-    def _set_order_quantity(self, position):
-        max_stock = self.config.get("max_stock", 100)
-        min_stock = self.config.get("min_stock", 0)
-
-        stock_quantity = random.randint(min_stock, max_stock)
-
-        position.order_quantity = max(0, min(position.stock_quantity, stock_quantity))
-
-    def _check_refusal_threshold(self, refused):
-        refusal_threshold_limit = self.config.get("refusal_threshold", 1)
-        refusal_level = refused/len(self.order.positions)
-
-        if refusal_level > refusal_threshold_limit:
-            self.order.status = OrderStatus.OPERATOR_REQUIRED
-            self.order.reason = "Превышен порог отказов"
-

src/mail_order_bot/email_processor/order/__init__.py

@@ -0,0 +1,2 @@
+from .auto_part_order import AutoPartOrder, OrderStatus
+from .auto_part_position import AutoPartPosition, PositionStatus

src/mail_order_bot/email_processor/order/auto_part_order.py

@@ -1,5 +1,6 @@
 from typing import List, Optional
-from .auto_part_position import AutoPartPosition
+from .auto_part_position import AutoPartPosition, PositionStatus
+
 from enum import Enum
 
 class OrderStatus(Enum):
@@ -15,12 +16,12 @@ class AutoPartOrder:
     def __init__(self):
         self.positions: List[AutoPartPosition] = []
         self.status = OrderStatus.NEW
+        self.delivery_period = 0
         self.reason = ""
+        self.errors = []
 
     def add_position(self, position: AutoPartPosition) -> None:
         self.positions.append(position)
-        if self.status == OrderStatus.NEW:
-            self.status = OrderStatus.IN_PROGRESS
 
     def find_positions(self, brand: Optional[str] = None, sku: Optional[str] = None) -> List[AutoPartPosition]:
         results = self.positions
@@ -30,5 +31,29 @@ class AutoPartOrder:
             results = [p for p in results if p.sku == sku]
         return results
 
+    def set_delivery_period(self, delivery_period: int) -> None:
+        self.delivery_period = delivery_period
+
+    def fill_from_local_supplier(self) -> None:
+        for position in self.positions:
+            errors = position.fill_from_stock()
+            self.errors += errors
+
+
+
+    def check_order(self, config) -> None:
+        """ Проверяет заказ на возможность исполнения"""
+        # 1. Проверка общего количества отказов
+        order_refusal_threshold = config.get("order_refusal_threshold", 1)
+        refusal_positions_count = len([position for position in self.positions if str(position.status) in
+                                       [PositionStatus.REFUSED, PositionStatus.STOCK_FAILED]])
+
+        order_refusal_rate = refusal_positions_count / len(self.positions)
+        if order_refusal_rate > order_refusal_threshold:
+            self.errors.append(f"Превышен порог отказов в заказе - {order_refusal_rate:.0%} "
+                               f"({refusal_positions_count} из {len(self.positions)})")
+            self.status = OrderStatus.OPERATOR_REQUIRED
+
+
     def __len__(self):
         return len(self.positions)

src/mail_order_bot/email_processor/order/auto_part_position.py

@@ -2,25 +2,45 @@ from typing import List, Optional
 from dataclasses import dataclass, field
 from typing import Dict, Any
 from decimal import Decimal
+from enum import Enum
+
+class PositionStatus(Enum):
+    NEW = "new"  # Новая позиция
+    STOCK_RECIEVED = "stock_received"  # Получен остаток
+    STOCK_FAILED = "stock_failed" # Остаток не получен
+    NO_AVAILABLE_STOCK = "no_available_stock" #Нет доступных складов
+    READY = "ready"
+    READY_PARTIAL = "ready_partial"
+    ORDERED = "ordered"  # Заказано
+    REFUSED = "refused"  # Отказано
 
 
 @dataclass
 class AutoPartPosition:
+    DISTRIBUTOR_ID = "1577730"
     """
     Унифицированная модель позиции для заказа.
     Все контрагенты приводятся к этой структуре.
     """
     sku: str  # Артикул товара
     manufacturer: str  # Производитель
+
+    requested_price: Decimal  # Цена за единицу
     requested_quantity: int  # Количество
+
     total: Decimal = 0  # Общая сумма
     name: str = ""  # Наименование
-    requested_price: Decimal = 0  # Цена за единицу
+
     order_quantity: int = 0  # Количество для заказа
     order_price: Decimal = Decimal('0.0')  # Цена в заказе
+    order_item: Dict[str, Any] = field(default_factory=dict)
+
     stock: List[Dict[str, Any]] = None
     additional_attrs: Dict[str, Any] = field(default_factory=dict)
 
+    status: PositionStatus = PositionStatus.NEW
+    desc: str = ""
+
     def __post_init__(self):
         """Валидация после инициализации"""
         if self.requested_quantity < 0:
@@ -28,50 +48,71 @@ class AutoPartPosition:
         if self.requested_price < 0:
             raise ValueError(f"Цена не может быть отрицательной: {self.requested_price}")
 
+    def update_stock(self, stock: Dict[str, Any], delivery_period: int = 0) -> None:
+        if stock["success"]:
+            available_distributors = stock["data"]
+
+            # Для доставки только с локального склада сперва убираем все остальные склады
+            if delivery_period == 0:
+                available_distributors = self._filter_only_local_storage(available_distributors)
+
+            #Отбираем склады по сроку доставки
+            available_distributors = self._filter_proper_delivery_time(available_distributors, delivery_period)
+
+            # Убираем дорогие склады с ценой выше запрошенной
+            available_distributors = self._filter_proper_price(available_distributors)
+
+            # Убираем отрицательные остатки
+            available_distributors = self._filter_proper_availability(available_distributors)
+
+            # Сортируем по цене
+            available_distributors.sort(key=lambda item: Decimal(item["price"]), reverse=False)
+
+            self.stock = available_distributors
+            if len (self.stock):
+                self.status = PositionStatus.STOCK_RECIEVED
+            else:
+                self.status = PositionStatus.NO_AVAILABLE_STOCK
+        else:
+            self.status = PositionStatus.STOCK_FAILED
+
+
+    def fill_from_stock(self):
+        if self.status == PositionStatus.STOCK_RECIEVED:
+
+            for distributor in self.stock:
+                distributor["profit"] = int(distributor["availability"]) * self.requested_price - int(distributor["availability"]) * Decimal(distributor["price"])
+
+            self.stock.sort(key=lambda item: item["profit"], reverse=True)
+
+            self.order_quantity = self.stock[0]["availability"]
+            self.order_price = self.requested_price
+            self.order_item = self.stock[0]
+
+            self.status = PositionStatus.READY
+
+
+    def _filter_only_local_storage(self, distributors):
+        return [item for item in distributors if str(item["distributorId"]) == self.DISTRIBUTOR_ID]
+
+    def _filter_proper_delivery_time(self, distributors, delivery_period):
+        return [item for item in distributors if item["deliveryPeriod"] <= delivery_period]
+
+    def _filter_proper_price(self, distributors):
+        return [item for item in distributors if Decimal(item["price"]) <= self.requested_price]
+
+    def _filter_proper_availability(self, distributors):
+        return [item for item in distributors if Decimal(item["availability"]) > 0]
+
+
+
 
 
 
-class AutoPartPosition2:
-    brand: str
-    sku: str
-    name: str
-    customer_price: float
-    customer_quantity: int
-    supplier_price: float
-    stock_remaining: int
 
-    def __init__(self, brand: str, sku: str, name: str,
-                 customer_price: float, customer_quantity: int,
-                 supplier_price: float, stock_remaining: int):
-        self.brand = brand
-        self.sku = sku
-        self.name = name
-        self.customer_price = customer_price
-        self.customer_quantity = customer_quantity
-        self.supplier_price = supplier_price
-        self.stock_remaining = stock_remaining
 
-    def customer_cost(self) -> float:
-        return self.customer_price * self.customer_quantity
 
-    def supplier_cost(self) -> float:
-        return self.supplier_price * self.customer_quantity
 
-    def is_available(self) -> bool:
-        return self.stock_remaining >= self.customer_quantity
 
-    def restock(self, amount: int) -> None:
-        if amount < 0:
-            raise ValueError("Restock amount must be non-negative")
-        self.stock_remaining += amount
 
-    def __post_init__(self):
-        if self.customer_price < 0:
-            raise ValueError("Customer price cannot be negative")
-        if self.customer_quantity < 0:
-            raise ValueError("Customer quantity cannot be negative")
-        if self.supplier_price < 0:
-            raise ValueError("Supplier price cannot be negative")
-        if self.stock_remaining < 0:
-            raise ValueError("Stock remaining cannot be negative")
 

src/mail_order_bot/email_processor/processor.py

@@ -3,12 +3,17 @@ import yaml
 import logging
 from typing import Dict, Any
 from pathlib import Path
+import threading
+from mail_order_bot.context import Context
+from mail_order_bot.email_client.utils import EmailUtils
+from enum import Enum
+
+from mail_order_bot.email_processor.handlers import *
+
+from mail_order_bot.email_processor.handlers import AttachmentHandler
 
 logger = logging.getLogger(__name__)
 
-from mail_order_bot.context import Context
-from enum import Enum
-
 
 class RequestStatus(Enum):
     NEW = "new"
@@ -19,9 +24,10 @@ class RequestStatus(Enum):
     INVALID = "invalid"
 
 
-class EmailProcessor(Context):
-    def __init__(self, configs_path: Path):
+class EmailProcessor:
+    def __init__(self, configs_path: str):
         super().__init__()
+        self.context = Context()
         self.configs_path = configs_path
         self.status = RequestStatus.NEW
 
@@ -30,23 +36,42 @@ class EmailProcessor(Context):
         self.context.clear()
 
         # Сохранить письмо в контекст
-        self.context["email"] = email
+        self.context.data["email"] = email
 
         # Определить клиента
+        email_body = EmailUtils.extract_body(email)
+        email_from = EmailUtils.extract_first_sender(email_body)
+        client = EmailUtils.extract_domain(email_from)
 
-
+        try:
             # Определить конфиг для пайплайна
-        config = {}
+            config = self._load_config(client)
+            self.context.data["config"] = config
+
+            # Обработка вложений
+            attachments_handler_task = AttachmentHandler()
+            attachments_handler_task.do()
 
             # Запустить обработку пайплайна
             for stage in config["pipeline"]:
                 handler_name = stage["handler"]
                 logger.info(f"Processing handler: {handler_name}")
-            task = globals()[handler_name](stage.get("config", None), self.context)
+                task = globals()[handler_name](stage.get("config", None))
                 task.do()
 
+
+        except FileNotFoundError:
+            logger.error(f"Конфиг для клиента {client} не найден")
+
+        for attachment in self.context.data["attachments"]:
+            print(attachment["order"].__dict__)
+        #except Exception as e:
+        #    logger.error(f"Произошла другая ошибка: {e}")
+
+
     def _load_config(self, client) -> Dict[str, Any]:
         """Загружает конфигурацию из YAML или JSON"""
+
         path = os.path.join(self.configs_path, client + '.yml')
         with open(path, 'r', encoding='utf-8') as f:
             return yaml.safe_load(f)

src/mail_order_bot/main.py

@@ -1,4 +1,4 @@
-
+import threading
 from config_manager import ConfigManager
 from dotenv import load_dotenv
 import asyncio
@@ -9,7 +9,7 @@ from dotenv import load_dotenv
 from email_client import EmailClient
 from email_processor import EmailProcessor
 
-from context import Context
+from mail_order_bot.context import Context
 
 
 logger = logging.getLogger()
@@ -34,11 +34,11 @@ class MailOrderBot(ConfigManager):
         self.context.email_client = self.email_client
 
         # Обработчик писем
-        self.email_processor = EmailProcessor()
+        self.email_processor = EmailProcessor("./configs")
+        logger.warning("MailOrderBot инициализирован")
+
 
     def execute(self):
-        logger.debug(f"Check emails for new orders")
-
         # Получить список айдишников письма
         unread_email_ids = self.email_client.get_emails_id(folder="spareparts")
 
@@ -46,10 +46,12 @@ class MailOrderBot(ConfigManager):
 
         # Обработать каждое письмо по идентификатору
         for email_id in unread_email_ids:
+            logger.debug(f"==================================================")
             logger.debug(f"Обработка письма с идентификатором {email_id}")
             # Получить письмо по идентификатору и запустить его обработку
             email = self.email_client.get_email(email_id)
             self.email_processor.process_email(email)
+            pass
 
 
 logger = logging.getLogger()