Перевел таск процессор на обработку только файлов

example.env

@@ -10,3 +10,7 @@ IMAP_PORT=993
 SMTP_HOST=smtp.gmail.com
 SMTP_PORT=587
 
+# Telegram Bot settings
+TELEGRAM_BOT_TOKEN=your_bot_token_here
+TELEGRAM_CHAT_ID=your_chat_id_here
+

src/mail_order_bot/__init__.py

@@ -0,0 +1 @@
+from .main import MailOrderBotException

src/mail_order_bot/abcp_api/abcp_provider.py

@@ -29,10 +29,24 @@ class AbcpProvider:
             path = "/search/articles"
 
             params = {"number": sku, "brand": manufacturer, "withOutAnalogs": "1"}
-            return self._execute(path, method, params)
 
-        def save_order(self, order):
-            pass
+            status_code, payload = self._execute(path, method, params)
+
+            if status_code == 200:
+                response = {"success": True, "data": payload}
+                logger.debug(f"Получены данные об остатках на складе")
+                return response
+
+            elif status_code == 301:
+                response = {"success": True, "data": []}
+                logger.debug(f"Не найдены позиции по запрошенным параметрам")
+                return response
+
+            else:
+                response = {"success": False, "data": payload}
+                logger.debug(f"Ошибка при получении остатков со склада")
+                return response
+
 
         def _execute(self, path, method="GET", params={}, data=None):
             params["userlogin"] = self.login
@@ -40,18 +54,4 @@ class AbcpProvider:
 
             response = requests.request(method, self.HOST+path, data=data, headers=self.HEADERS, params=params)
             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
-
+            return response.status_code, payload

src/mail_order_bot/config.yml

@@ -1,10 +1,46 @@
-# Настройки обработки  =================================================================
+# Настройки обработки =================================================================
+folder: "spareparts"
+
+clients:
+  lesha.spb@gmail.com:
+    enabled: true
+    client_id: 6148154    # Сейчас стоит айдишник Димы для тестовых заказов
+
+    pipeline:
+      - ExcelExtractor
+      - DeliveryPeriodFromConfig
+      - OrderExtractor
+      - StockSelector
+      - UpdateExcelFile
+      - SaveOrderToTelegram
+      - EmailReplyTask
+
+    excel:
+      sheet_name: 0
+      key_field: "Номер"
+      mapping:
+        article: "Номер"
+        manufacturer: "Фирма"
+        name: "Наименование"
+        price: "Цена"
+        quantity: "Кол-во"
+        total: "Сумма"
+      updatable_fields:
+        ordered_quantity: "Кол-во Поставщика"
+        ordered_price: "Цена Поставщика"
+
+    # Значение для хендлера DeliveryPeriodFromConfig
+    delivery_period: 100  # в часах
+
+  amtel.ru:
+    enabled: false
+
+
 
 # Раздел с общими конфигурационными параметрами ===============================
 update_interval: 1
 work_interval: 60
 email_dir: "spareparts"
-
 # Логирование =================================================================
 log:
   version: 1
@@ -18,7 +54,7 @@ log:
 
   handlers:
     console:
-      level: DEBUG
+      level: WARNING
       formatter: standard
       class: logging.StreamHandler
       stream: ext://sys.stdout  # Default is stderr
@@ -36,23 +72,22 @@ log:
       level: CRITICAL
       formatter: telegram
       class: logging_telegram_handler.TelegramHandler
-      chat_id: 211945135
+      chat_id: -1002960678041 #-1002960678041 #211945135
       alias: "Mail order bot"
 
-
   # Логгеры
   loggers:
     '':
       handlers: [console, file, telegram]
-      level: DEBUG
+      level: WARNING
       propagate: False
 
     __main__:
       handlers: [console, file, telegram]
-      level: INFO
+      level: WARNING
       propagate: False
 
     config_manager:
       handlers: [console, file]
-      level: DEBUG
+      level: ERROR
     

src/mail_order_bot/configs/amtel.club.yml

@@ -1,32 +1,32 @@
 #=========================================
-config:
-
-
-pipeline:
- - 
+client: amtel.club
+enabled: true
+client_id: 6148154    # Сейчас стоит айдишник Димы, фактический id  у amtel.club - 156799563
 
+delivery_period: 100  # в часах
+excel:
+  sheet_name: 0
+  key_field: "Номер"
+  mapping:
+    article: "Номер"
+    manufacturer: "Фирма"
+    name: "Наименование"
+    price: "Цена"
+    quantity: "Кол-во"
+    total: "Сумма"
 
+  updatable_fields:
+    ordered_quantity: "Кол-во Поставщика"
+    ordered_price: "Цена Поставщика"
 
 #=========================================
 pipeline:
-  # Настраиваем парсинг экселя
-  - handler: BasicExcelParser
-    config:
-      sheet_name: 0
-      key_field: "Номер"
-      mapping:
-        article: "Номер"
-        manufacturer: "Фирма"
-        name: "Наименование"
-        price: "Цена"
-        quantity: "Кол-во"
-        total: "Сумма"
-
-  # Определяем логику обработки заказа (в данном случае все с локального склада)
-  - handler: DeliveryPeriodLocalStore
-
-  # Запрос остатков со склада
-  - handler: APIGetStock
+ - ExcelExtractor
+ - OrderExtractor
+ - DeliveryPeriodFromConfig
+ - StockSelector
+ - UpdateExcelFile
+ - SaveOrderToTelegram
 
 
 

src/mail_order_bot/configs/gmail.com.yml

@@ -0,0 +1,39 @@
+#=========================================
+client: gmail.com
+enabled: true
+client_id: 6148154    # Сейчас стоит айдишник Димы, фактический id  у amtel.club - 156799563
+
+delivery_period: 100  # в часах
+excel:
+  sheet_name: 0
+  key_field: "Номер"
+  mapping:
+    article: "Номер"
+    manufacturer: "Фирма"
+    name: "Наименование"
+    price: "Цена"
+    quantity: "Кол-во"
+    total: "Сумма"
+
+  updatable_fields:
+    ordered_quantity: "Кол-во Поставщика"
+    ordered_price: "Цена Поставщика"
+
+#=========================================
+pipeline:
+ - ExcelExtractor
+ - OrderExtractor
+ - DeliveryPeriodFromConfig
+ - StockSelector
+ - UpdateExcelFile
+ - SaveOrderToTelegram
+
+
+
+
+
+
+
+
+
+

src/mail_order_bot/context.py

@@ -1,3 +1,27 @@
+"""
+Структура контекста
+
+  email        : Электронное письмо в формате EmailObject
+  config       : Конфиг для текущего клиента
+  attachments  : [   Список вложений
+    {
+      name    : Имя файла
+      isOrder : Признак что файл является валидным файлом заказа
+      bytes   : содержимое файла в формате BytesIO
+      deliveryPeriod: Int
+      sheet   : Распарсенный лист в формате ExcelFileParcer
+      order   : Файл заказа в формате AutopartOrder
+      log     : Лог сообщений по обработке файла в формате LogMessage
+    }
+  ]
+  status:
+
+
+
+"""
+
+
+
 import threading
 from typing import Any, Dict
 import logging

src/mail_order_bot/credential_provider.py

@@ -75,8 +75,8 @@ class CredentialProvider:
         Raises:
             ValueError: Если учетные данные системной учетной записи не найдены
         """
-        login_key = f"{self.prefix}_LOGIN_{self.SYSTEM_ACCOUNT}"
-        password_key = f"{self.prefix}_PASSWORD_{self.SYSTEM_ACCOUNT}"
+        login_key = f"{self.prefix}_LOGIN"
+        password_key = f"{self.prefix}_PASSWORD"
         
         login = os.getenv(login_key)
         password = os.getenv(password_key)

src/mail_order_bot/deprecated/old_excel_parcer/__init__.py

@@ -1 +0,0 @@
-from .processor import ExcelProcessor

src/mail_order_bot/deprecated/old_excel_parcer/configurable_parser.py

@@ -1,113 +0,0 @@
-import logging
-import pandas as pd
-from typing import Dict, Any, Optional, List
-from decimal import Decimal
-
-from .excel_parser import ExcelParser
-from .order_position import OrderPosition
-
-
-logger = logging.getLogger(__name__)
-
-
-class ConfigurableExcelParser(ExcelParser):
-    """
-    Универсальный парсер, настраиваемый через конфигурацию.
-    Подходит для большинства стандартных случаев.
-    """
-    
-    def parse(self, file_bytes: str) -> List[OrderPosition]:
-        try:
-            # Читаем Excel
-            df = self._make_dataframe(file_bytes)
-            
-            # Получаем маппинг колонок из конфигурации
-            mapping = self.config['mapping']
-  
-            # Парсим строки
-            positions = []
-            for idx, row in df.iterrows():
-                try:
-                    position = self._parse_row(row, mapping)
-                    if position:
-                        positions.append(position)
-                except Exception as e:
-                    logger.error(f"Ошибка парсинга строки {idx}: {e}, {row}")
-                    continue
-            
-            logger.info(f"Успешно обработано {len(positions)} позиций из {len(df)} строк")
-            return positions
-            
-        except Exception as e:
-            logger.error(f"Ошибка при обработке файла: {e}")
-            raise Exception from e
-    
-    def _parse_row(self, row: pd.Series, mapping: Dict[str, str]) -> Optional[OrderPosition]:
-        """Парсит одну строку Excel в OrderPosition"""
-
-        # Проверяем обязательные поля
-        required_fields = ['article', 'price', 'quantity']
-
-        for field in required_fields:
-            if pd.isna(row.get(mapping[field])):
-                logger.warning(f"Позиция не создана - не заполнено поле {mapping[field]}")
-                return None
-
-        price = Decimal(str(row[mapping['price']]).replace(",", ".").strip())
-        quantity = int(row[mapping['quantity']])
-
-        if "total" in mapping.keys():
-            total = Decimal(str(row[mapping['total']]).replace(",", ".").strip())
-        else:
-            total = price * quantity
-
-        if mapping.get('name',"") in mapping.keys():
-            name = str(row[mapping.get('name', "")]).strip()
-        else:
-            name = ""
-        
-        # Создаем объект позиции
-        position = OrderPosition(
-            article=str(row[mapping['article']]).strip(),
-            manufacturer=str(row[mapping.get('manufacturer',"")]).strip(),
-            name=name,
-            price=price,
-            quantity=quantity,
-            total=total,
-            additional_attrs=self._extract_additional_attrs(row, mapping)
-        )
-        return position
-    
-    def _extract_additional_attrs(self, row: pd.Series, mapping: Dict[str, str]) -> Dict[str, Any]:
-        """Извлекает дополнительные атрибуты, не входящие в основную модель"""
-        additional = {}
-        mapped_columns = set(mapping.values())
-        
-        for col in row.index:
-            if col not in mapped_columns and not pd.isna(row[col]):
-                additional[col] = row[col]
-        
-        return additional
-
-
-    def _make_dataframe(self, bio) -> pd.DataFrame:
-        # Получаем все данные из файла
-        sheet_name = self.config.get("sheet_name", 0)
-        df_full = pd.read_excel(bio, sheet_name=sheet_name, header=None)
-
-        # Находим индекс строки с заголовком
-        key_field = self.config.get("key_field")
-        header_row_idx = df_full[
-            df_full.apply(lambda row: row.astype(str).str.contains(key_field, case=False, na=False).any(),
-                          axis=1)].index[0]
-
-        # Считываем таблицу с правильным заголовком
-        df = pd.read_excel(bio, header=header_row_idx, sheet_name=sheet_name, engine='calamine') #openpyxl  calamine
-
-        # Находим индекс первой строки с пустым 'Артикул'
-        first_empty_index = df[df[key_field].isna()].index.min()
-
-        # Обрезаем DataFrame до первой пустой строки (не включая её)
-        df_trimmed = df.loc[:first_empty_index - 1]
-
-        return df_trimmed

src/mail_order_bot/deprecated/old_excel_parcer/custom_parser_autoeuro.py

@@ -1,105 +0,0 @@
-import logging
-import pandas as pd
-from typing import Dict, Any, Optional, List
-from decimal import Decimal
-import xlrd
-from io import BytesIO
-
-from .excel_parser import ExcelParser
-from .order_position import OrderPosition
-
-logger = logging.getLogger(__name__)
-
-
-class CustomExcelParserAutoeuro(ExcelParser):
-    """
-    Универсальный парсер, настраиваемый через конфигурацию.
-    Подходит для большинства стандартных случаев.
-    """
-
-    def parse(self, file_bytes: BytesIO) -> List[OrderPosition]:
-        try:
-            # Читаем Excel
-            df = self._make_dataframe(file_bytes)
-
-            # Получаем маппинг колонок из конфигурации
-            mapping = self.config['mapping']
-
-            # Парсим строки
-            positions = []
-            for idx, row in df.iterrows():
-                try:
-                    position = self._parse_row(row, mapping)
-                    if position:
-                        positions.append(position)
-                except Exception as e:
-                    logger.error(f"Ошибка парсинга строки {idx}: {e}, {row}")
-                    continue
-
-            logger.info(f"Успешно обработано {len(positions)} позиций из {len(df)} строк")
-            return positions
-
-        except Exception as e:
-            logger.error(f"Ошибка при обработке файла: {e}")
-            raise Exception from e
-
-    def _parse_row(self, row: pd.Series, mapping: Dict[str, str]) -> Optional[OrderPosition]:
-        """Парсит одну строку Excel в OrderPosition"""
-
-        # Проверяем обязательные поля
-        required_fields = ['article', 'price', 'quantity']
-
-        for field in required_fields:
-            if pd.isna(row.get(mapping[field])):
-                logger.warning(f"Позиция не создана - не заполнено поле {mapping[field]}")
-                return None
-
-        price = Decimal(str(row[mapping['price']]).replace(",", ".").strip())
-        quantity = int(row[mapping['quantity']])
-
-        if "total" in mapping.keys():
-            total = Decimal(str(row[mapping['total']]).replace(",", ".").strip())
-        else:
-            total = price * quantity
-
-        # Создаем объект позиции
-        position = OrderPosition(
-            article=str(row[mapping['article']]).strip(),
-            manufacturer=str(row[mapping.get('manufacturer', "")]).strip(),
-            name="", #str(row[mapping.get('name', "name")]).strip(),
-            price=price,
-            quantity=quantity,
-            total=total,
-            additional_attrs=self._extract_additional_attrs(row, mapping)
-        )
-        return position
-
-    def _extract_additional_attrs(self, row: pd.Series, mapping: Dict[str, str]) -> Dict[str, Any]:
-        """Извлекает дополнительные атрибуты, не входящие в основную модель"""
-        additional = {}
-        mapped_columns = set(mapping.values())
-
-        for col in row.index:
-            if col not in mapped_columns and not pd.isna(row[col]):
-                additional[col] = row[col]
-
-        return additional
-
-    def _make_dataframe(self, bio) -> pd.DataFrame:
-
-        file_bytes = bio.read()
-        book = xlrd.open_workbook(file_contents=file_bytes, encoding_override='cp1251')
-        sheet = book.sheet_by_index(self.config.get("sheet_index", 0))
-        data = [sheet.row_values(row) for row in range(sheet.nrows)]
-
-        df_full = pd.DataFrame(data)
-
-        key_field = self.config.get("key_field")
-        header_row_idx = df_full[
-            df_full.apply(lambda row: row.astype(str).str.contains(key_field, case=False, na=False).any(),
-                          axis=1)].index[0]
-
-        df = df_full[header_row_idx:]
-        df.columns = df.iloc[0]  # первая строка становится заголовком
-        df = df.reset_index(drop=True).drop(0).reset_index(drop=True)  # удаляем первую строку и сбрасываем индекс
-        return df

src/mail_order_bot/deprecated/old_excel_parcer/excel_parser.py

@@ -1,28 +0,0 @@
-import logging
-import pandas as pd
-from abc import ABC, abstractmethod
-from typing import Dict, Any, List
-from io import BytesIO
-
-
-from .order_position import OrderPosition
-
-
-logger = logging.getLogger(__name__)
-
-class ExcelParser(ABC):
-    """
-    Абстрактный базовый класс для всех парсеров Excel.
-    Реализует Strategy Pattern - каждый контрагент = своя стратегия.
-    """
-    
-    def __init__(self, config: Dict[str, Any]):
-        self.config = config
-    
-    @abstractmethod
-    def parse(self, file: BytesIO) -> List[OrderPosition]:
-        """
-        Парсит Excel файл и возвращает список позиций.
-        Должен быть реализован в каждом конкретном парсере.
-        """
-        pass

src/mail_order_bot/deprecated/old_excel_parcer/order_position.py

@@ -1,24 +0,0 @@
-from dataclasses import dataclass, field
-from typing import Dict, Any
-from decimal import Decimal
-
-@dataclass
-class OrderPosition:
-    """
-    Унифицированная модель позиции для заказа.
-    Все контрагенты приводятся к этой структуре.
-    """
-    article: str                    # Артикул товара
-    manufacturer: str               # Производитель
-    name: str                       # Наименование
-    price: Decimal                  # Цена за единицу
-    quantity: int                   # Количество
-    total: Decimal                  # Общая сумма
-    additional_attrs: Dict[str, Any] = field(default_factory=dict)
-    
-    def __post_init__(self):
-        """Валидация после инициализации"""
-        if self.quantity < 0:
-            raise ValueError(f"Количество не может быть отрицательным: {self.quantity}")
-        if self.price < 0:
-            raise ValueError(f"Цена не может быть отрицательной: {self.price}")

src/mail_order_bot/deprecated/old_excel_parcer/parser_factory.py

@@ -1,54 +0,0 @@
-import yaml
-import json
-import logging
-from pathlib import Path
-from typing import Dict, Any, List
-
-from .excel_parser import ExcelParser
-from .configurable_parser import ConfigurableExcelParser
-from .custom_parser_autoeuro import CustomExcelParserAutoeuro
-
-
-logger = logging.getLogger(__name__)
-
-class ParserFactory:
-    """
-    Фабрика парсеров (Factory Pattern).
-    Создает нужный парсер на основе названия контрагента.
-    """
-    
-    # Реестр кастомных парсеров
-    CUSTOM_PARSERS = {
-        'autoeuro.ru': CustomExcelParserAutoeuro,
-        # Добавляйте сюда специализированные парсеры
-    }
-    
-    def __init__(self, config: Dict[str, Any]):
-        self.config = config
-
-    def get_parser(self, supplier_name: str) -> ExcelParser:
-        """
-        Возвращает парсер для указанного контрагента.
-        Использует кастомный парсер если есть, иначе конфигурируемый.
-        """
-        if supplier_name not in self.config['suppliers']:
-            raise ValueError(
-                f"Контрагент '{supplier_name}' не найден в конфигурации. "
-                f"Доступные: {list(self.config['suppliers'].keys())}"
-            )
-        
-        config = self.config['suppliers'][supplier_name]
-        
-        # Проверяем, есть ли кастомный парсер
-        if supplier_name in self.CUSTOM_PARSERS:
-            parser_class = self.CUSTOM_PARSERS[supplier_name]
-            logger.debug(f"Используется кастомный парсер для {supplier_name}")
-        else:
-            parser_class = ConfigurableExcelParser
-            logger.debug(f"Используется конфигурируемый парсер для {supplier_name}")
-        
-        return parser_class(config)
-    
-    def list_suppliers(self) -> List[str]:
-        """Возвращает список всех доступных контрагентов"""
-        return list(self.config['suppliers'].keys())

src/mail_order_bot/deprecated/old_excel_parcer/processor.py

@@ -1,110 +0,0 @@
-import logging
-from pathlib import Path
-from decimal import Decimal
-from io import BytesIO
-from typing import Dict, Any, List
-import yaml
-import json
-
-
-from .parser_factory import ParserFactory
-from .order_position import OrderPosition
-
-
-logger = logging.getLogger(__name__)
-
-class ExcelProcessor:
-    """
-    Главный класс-фасад для обработки Excel файлов.
-    Упрощает использование системы.
-    """
-    
-    def __init__(self, config_path: str = 'config/suppliers.yaml', ):
-        self.config_path = Path(config_path)
-        self.config = self._load_config()
-        self.factory = ParserFactory(self.config)
-
-    def process(self, file_bytes: BytesIO, file_name: str, supplier_name: str, validate: bool = False) -> List[OrderPosition]:
-        """
-        Обрабатывает Excel файл от контрагента.
-
-        Args:
-            file_bytes: Байты файла
-            file_name: Имя файла
-            supplier_name: Название контрагента (из конфигурации)
-            validate: Выполнять ли дополнительную валидацию
-
-        Returns:
-            Список объектов OrderPosition
-
-        Raises:
-            ValueError: Если контрагент не найден
-        """
-        logger.info(f"Обработка файла: {file_name} для {supplier_name}")
-
-        parser = self.factory.get_parser(supplier_name)
-        positions = parser.parse(file_bytes)
-
-        # Дополнительная валидация если нужна
-        if validate:
-            positions = self._validate_positions(positions)
-
-        logger.debug(f"Обработка завершена: получено {len(positions)} позиций")
-        return positions
-
-    def process_file(self, file_path: str, supplier_name: str, validate: bool = False) -> List[OrderPosition]:
-        # Проверка существования файла
-        logger.debug(f"Чтение файла: {file_path}")
-        if not Path(file_path).exists():
-            raise FileNotFoundError(f"Файл не найден: {file_path}")
-
-        with open(file_path, 'rb') as file:  # бинарный режим
-            raw_data = file.read()
-        bio = BytesIO(raw_data)
-
-        positions = self.process(bio, file_path, supplier_name, validate=validate)
-
-        return positions
-
-    
-    def _validate_positions(self, positions: List[OrderPosition]) -> List[OrderPosition]:
-        """Дополнительная валидация позиций"""
-        valid_positions = []
-        
-        for pos in positions:
-            try:
-                # Проверка на непустые обязательные поля
-                if not pos.article or not pos.name:
-                    logger.warning(f"Пропущена позиция с пустыми полями: {pos}")
-                    continue
-                
-                # Проверка корректности суммы
-                expected_total = pos.price * pos.quantity
-                if abs(pos.total - expected_total) > Decimal('0.01'):
-                    logger.warning(
-                        f"Несоответствие суммы для {pos.article}: "
-                        f"ожидается {expected_total}, получено {pos.total}"
-                    )
-                
-                valid_positions.append(pos)
-                
-            except Exception as e:
-                logger.error(f"Ошибка валидации позиции: {e}")
-                continue
-        
-        return valid_positions
-    
-    def get_available_suppliers(self) -> List[str]:
-        """Возвращает список доступных контрагентов"""
-        return self.factory.list_suppliers()
-
-    def _load_config(self) -> Dict[str, Any]:
-        """Загружает конфигурацию из YAML или JSON"""
-        if self.config_path.suffix in ['.yaml', '.yml']:
-            with open(self.config_path, 'r', encoding='utf-8') as f:
-                return yaml.safe_load(f)
-        elif self.config_path.suffix == '.json':
-            with open(self.config_path, 'r', encoding='utf-8') as f:
-                return json.load(f)
-        else:
-            raise ValueError(f"Неподдерживаемый формат конфига: {self.config_path.suffix}")

src/mail_order_bot/email_client/client.py

@@ -1,6 +1,6 @@
 import re
 from datetime import datetime
-from typing import List, Optional
+from typing import List, Optional, Union
 from dataclasses import dataclass
 import email
 from email import encoders
@@ -8,17 +8,22 @@ from email.mime.text import MIMEText
 from email.mime.multipart import MIMEMultipart
 from email.mime.base import MIMEBase
 from email.header import decode_header
+from email.message import Message
 import imaplib
 import smtplib
 
 import logging
 
+
+
+
 logger = logging.getLogger(__name__)
 
 
 # from .objects import EmailMessage, EmailAttachment
 
 
+
 class EmailClient:
     def __init__(self, imap_host: str, smtp_host: str, email: str, password: str,
                  imap_port: int = 993, smtp_port: int = 587):
@@ -31,7 +36,7 @@ class EmailClient:
         self.imap_conn = None
 
     def connect(self):
-        """Установkение IMAP соединения"""
+        """Установление IMAP соединения"""
         if self.imap_conn is None:
             self.imap_conn = imaplib.IMAP4_SSL(self.imap_host, self.imap_port)
             self.imap_conn.login(self.email, self.password)
@@ -57,18 +62,27 @@ class EmailClient:
 
     def get_emails_id(self, folder: str = "INBOX", only_unseen: bool = True) -> List[int]:
         """Получить список новых электронных писем."""
-        self.connect()
-        self.imap_conn.select(folder, readonly=False)
-        # Ищем письма
-        search_criteria = "(UNSEEN)" if only_unseen else "ALL"
-        status, messages = self.imap_conn.search(None, search_criteria)
+        try:
+            self.connect()
+            self.imap_conn.select(folder, readonly=False)
+            # Ищем письма
+            search_criteria = "(UNSEEN)" if only_unseen else "ALL"
+            status, messages = self.imap_conn.search(None, search_criteria)
+
+            # ToDo сделать обработку ошибок, подумать нужна ли она!
+            if status != "OK":
+                return []
+
+            email_ids = messages[0].split()
+
+        except Exception as e:
+            logger.error(e)
+            raise
+        else:
+            return email_ids
+
 
-        # ToDo сделать обработку ошибок, подумать нужна ли она!
-        if status != "OK":
-            return []
 
-        email_ids = messages[0].split()
-        return email_ids
 
     def get_email(self, email_id, mark_as_read: bool = True):
         """Получить список новых электронных писем."""
@@ -105,4 +119,62 @@ class EmailClient:
             else:
                 decoded_parts.append(str(part))
 
-        return ''.join(decoded_parts)
+        return ''.join(decoded_parts)
+
+    def send_email(self, message: Union[MIMEMultipart, MIMEText, Message]):
+        """
+        Отправить подготовленное письмо через SMTP.
+        
+        Args:
+            message: Подготовленное письмо (MIMEMultipart, MIMEText или email.message.Message)
+                    Должно содержать заголовки From, To и Subject
+        """
+        try:
+            # Извлекаем получателей из письма
+            recipients = []
+            
+            # Основной получатель
+            to_header = message.get("To", "")
+            if to_header:
+                # Обрабатываем несколько адресов, разделенных запятыми
+                to_addresses = [addr.strip() for addr in to_header.split(",")]
+                recipients.extend(to_addresses)
+            
+            # Копия
+            cc_header = message.get("Cc", "")
+            if cc_header:
+                cc_addresses = [addr.strip() for addr in cc_header.split(",")]
+                recipients.extend(cc_addresses)
+            
+            # Скрытая копия
+            bcc_header = message.get("Bcc", "")
+            if bcc_header:
+                bcc_addresses = [addr.strip() for addr in bcc_header.split(",")]
+                recipients.extend(bcc_addresses)
+            
+            if not recipients:
+                raise ValueError("Не указаны получатели письма (To, Cc или Bcc)")
+            
+            # Извлекаем отправителя из письма или используем email из настроек
+            from_email = message.get("From", self.email)
+            
+            # Подключаемся к SMTP серверу
+            with smtplib.SMTP(self.smtp_host, self.smtp_port) as server:
+                server.starttls()
+                server.login(self.email, self.password)
+                
+                # Отправляем письмо
+                server.sendmail(
+                    from_email,
+                    recipients,
+                    message.as_string()
+                )
+                
+                logger.info(f"Письмо успешно отправлено получателям: {', '.join(recipients)}")
+                
+        except smtplib.SMTPException as e:
+            logger.error(f"Ошибка SMTP при отправке письма: {str(e)}")
+            raise Exception(f"Ошибка SMTP: {str(e)}")
+        except Exception as e:
+            logger.error(f"Ошибка при отправке письма: {str(e)}")
+            raise Exception(f"Ошибка при отправке письма: {str(e)}")

src/mail_order_bot/email_client/utils.py

@@ -78,7 +78,7 @@ class EmailUtils:
                 filename = part.get_filename()
                 if filename:
                     # Декодируем имя файла
-                    filename = decode_header(filename)[0]
+                    filename = decode_header(filename)[0][0]
                     # Получаем содержимое
                     content = part.get_payload(decode=True)
                     if content:

src/mail_order_bot/main.py

@@ -7,13 +7,16 @@ import os
 from dotenv import load_dotenv
 
 from email_client import EmailClient
-from email_processor import EmailProcessor
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+from task_processor import TaskProcessor
 
 from mail_order_bot.context import Context
 
 
 logger = logging.getLogger()
 
+class MailOrderBotException(Exception):
+    pass
 
 class MailOrderBot(ConfigManager):
     def __init__(self, *agrs, **kwargs):
@@ -34,29 +37,44 @@ class MailOrderBot(ConfigManager):
         self.context.email_client = self.email_client
 
         # Обработчик писем
-        self.email_processor = EmailProcessor("./configs")
+        #self.email_processor = TaskProcessor("./configs")
+        
+        config = self.config.get("clients")
+        self.email_processor = TaskProcessor(config)
         logger.warning("MailOrderBot инициализирован")
 
 
     def execute(self):
         # Получить список айдишников письма
-        unread_email_ids = self.email_client.get_emails_id(folder="spareparts")
 
-        logger.info(f"Новых писем - {len(unread_email_ids)}")
+        folder = self.config.get("folder")
+        try:
+            unread_email_ids = self.email_client.get_emails_id(folder=folder)
+            logger.info(f"Новых писем - {len(unread_email_ids)}")
+
+            # Обработать каждое письмо по идентификатору
+            for email_id in unread_email_ids:
+                try:
+                    logger.info(f"Обработка письма с идентификатором {email_id}")
+                    # Получить письмо по идентификатору и запустить его обработку
+                    email = self.email_client.get_email(email_id, mark_as_read=False)
+                    self.email_processor.process_email(email)
+
+                except MailOrderBotException as e:
+                    logger.error(f"Произошла ошибка {e}")
+
+        except MailOrderBotException as e:
+            logger.error(f"Произошла ошибка {e}")
+
+        except Exception as e:
+            logger.error(f"Произошла непредвиденная ошибка {e}")
 
-        # Обработать каждое письмо по идентификатору
-        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()
 
 async def main():
+    logger.critical("Запуск приложения")
     app = MailOrderBot("config.yml")
     await app.start()
 
@@ -65,9 +83,12 @@ async def main():
 
 if __name__ == "__main__":
     print(os.getcwd())
+
     if os.environ.get("APP_ENV") != "PRODUCTION":
         logger.warning("Non production environment")
         load_dotenv()
 
     asyncio.run(main())
+
+
     

src/mail_order_bot/order/auto_part_order.py

@@ -1,8 +1,6 @@
 from typing import List, Optional
 from .auto_part_position import AutoPartPosition, PositionStatus
 
-from mail_order_bot.task_processor.handlers.abcp.stock_selector import StockSelector
-
 from enum import Enum
 
 
@@ -38,17 +36,6 @@ class AutoPartOrder:
     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:
-            selector = StockSelector(position, self.delivery_period)
-            selector.select_optimal_supplier()
-
     def check_order(self, config) -> None:
         """ Проверяет заказ на возможность исполнения"""
         # 1. Проверка общего количества отказов

src/mail_order_bot/order/auto_part_position.py

@@ -21,23 +21,22 @@ class AutoPartPosition:
     Унифицированная модель позиции для заказа.
     Все контрагенты приводятся к этой структуре.
     """
-
-    DISTRIBUTOR_ID = 1577730  # ID локального склада
-
     sku: str  # Артикул товара
     manufacturer: str  # Производитель
 
-    requested_price: Decimal  # Цена за единицу
-    requested_quantity: int  # Количество
+    asking_price: Decimal  # Цена за единицу
+    asking_quantity: int  # Количество
 
     total: Decimal = 0  # Общая сумма
     name: str = ""  # Наименование
-    order_delivery_period: int = 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
+    order_item: Dict[str, Any] = field(default_factory=dict)
+    order_price: Decimal = Decimal('0.0')  # Цена в заказе
+    order_quantity: int = 0  # Количество для заказа
+    order_delivery_period: int = 0
+
+    profit: Decimal = Decimal('0.0')
+
     additional_attrs: Dict[str, Any] = field(default_factory=dict)
 
     status: PositionStatus = PositionStatus.NEW
@@ -45,65 +44,30 @@ class AutoPartPosition:
 
     def __post_init__(self):
         """Валидация после инициализации"""
-        if self.requested_quantity < 0:
-            raise ValueError(f"Количество не может быть отрицательным: {self.requested_quantity}")
-        if self.requested_price < 0:
-            raise ValueError(f"Цена не может быть отрицательной: {self.requested_price}")
+        if self.asking_quantity < 0:
+            raise ValueError(f"Количество не может быть отрицательным: {self.asking_quantity}")
+        if self.asking_price < 0:
+            raise ValueError(f"Цена не может быть отрицательной: {self.asking_price}")
 
-    def set_stock(self, stock):
-        if stock.get("success"):
-            self.stock = stock["data"]
-            if len(self.stock):
-                self.status = PositionStatus.STOCK_RECIEVED
-            else:
-                self.status = PositionStatus.NO_AVAILABLE_STOCK
-        else:
-            self.status = PositionStatus.STOCK_FAILED
+    def set_order_item(self, order_item):
+        # Запоминаем всю позицию
+        self.order_item = order_item
 
-    def set_order_item(self):
-        """Выбирает позицию для заказа"""
-        if self.status == PositionStatus.STOCK_RECIEVED:
-            available_distributors = self.stock
+        # ---===Устанавливаем конкретные значения по параметрам заказа===---
+        # Берем максимально доступное значение со склада, но не больше чем в заказе.
+        self.order_quantity = min(self.order_item.get("availability"), self.asking_quantity)
 
-            # BR-1. Отсекаем склады для заказов из наличия (только локальный склад)
-            if self.order_delivery_period == 0:
-                available_distributors = self._filter_only_local_storage(available_distributors)
+        # Продаем по цене, которая была заказана
+        self.order_price = self.asking_price
 
-            # BR-2. Цена не должна превышать цену из заказа
-            available_distributors = self._filter_proper_price(available_distributors)
+        # Устанавливаем актуальный срок доставки
+        self.order_delivery_period = self.order_item.get("deliveryPeriod")
 
-            # BR-3. Срок доставки не должен превышать ожидаемый
-            available_distributors = self._filter_proper_delivery_time(available_distributors)
+        # ФИксируем профит. Для инфо/отчетности
+        self.profit = (self.asking_price - Decimal(self.order_item.get("price"))) * self.order_quantity
 
-            # BR-4. Без отрицательных остатков
-            available_distributors = self._filter_proper_availability(available_distributors)
+        # Устанавливаем статус
+        self.status = PositionStatus.READY
 
-            # Приоритет на склады с полным стоком
 
-            # BR-5. Сначала оборачиваем локальный склад, потом удаленные
 
-            # BR-6. Выбираем цену максимально близкую к цене из заказа (максимальная)
-            available_distributors.sort(key=lambda item: Decimal(item["price"]), reverse=True)
-
-            if len(available_distributors):
-                self.order_item = self.stock[0]
-                self.status = PositionStatus.READY
-            else:
-                self.status = PositionStatus.NO_AVAILABLE_STOCK
-
-    def _filter_only_local_storage(self, distributors: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
-        """Фильтрует только локальные склады"""
-
-        return [item for item in distributors if item["distributorId"] == self.DISTRIBUTOR_ID]
-
-    def _filter_proper_delivery_time(self, distributors: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
-        """Фильтрует склады по сроку доставки"""
-        return [item for item in distributors if item["deliveryPeriod"] <= self.order_delivery_period]
-
-    def _filter_proper_price(self, distributors: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
-        """Фильтрует склады по цене (убирает дорогие)"""
-        return [item for item in distributors if Decimal(item["price"]) <= self.requested_price]
-
-    def _filter_proper_availability(self, distributors: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
-        """Фильтрует склады с положительным остатком"""
-        return [item for item in distributors if Decimal(item["availability"]) > 0]

src/mail_order_bot/parsers/excel_parcer.py

@@ -18,7 +18,6 @@ class ExcelFileParcer:
             df = pd.read_excel(file_bytes, sheet_name=self.sheet_name, header=None)
         except Exception as e:
             df = None
-            logger.warning("Не удалось распарсить значение файла")
         return df
 
     def set_value(self, sku, manufacturer, column, value):

src/mail_order_bot/parsers/order_parcer.py

@@ -5,6 +5,7 @@ from decimal import Decimal
 from io import BytesIO
 
 from mail_order_bot.order import AutoPartPosition
+from mail_order_bot.order import AutoPartOrder
 
 logger = logging.getLogger(__name__)
 
@@ -15,11 +16,13 @@ class OrderParser:
     Подходит для большинства стандартных случаев.
     """
 
-    def __init__(self, mapping, delivery_period):
+    def __init__(self, mapping, delivery_period, client_id):
         self.mapping = mapping
         self.delivery_period = delivery_period
+        self.client_id = client_id
 
-    def parse(self, order, df):
+    def parse(self, df):
+        order = AutoPartOrder(self.client_id)
         # Парсим строки
         positions = []
         for idx, row in df.iterrows():
@@ -63,8 +66,8 @@ class OrderParser:
             sku=str(row[mapping['article']]).strip(),
             manufacturer=str(row[mapping.get('manufacturer', "")]).strip(),
             name=name,
-            requested_price=price,
-            requested_quantity=quantity,
+            asking_price=price,
+            asking_quantity=quantity,
             total=total,
             order_delivery_period=self.delivery_period,
             additional_attrs=self._extract_additional_attrs(row, mapping)

src/mail_order_bot/task_processor/__init__.py

@@ -1 +1,2 @@
-from .processor import TaskProcessor
+from .processor import TaskProcessor
+from .message import LogMessage, LogMessageLevel, LogMessageStorage

src/mail_order_bot/task_processor/abstract_task.py

@@ -9,12 +9,13 @@ class AbstractTask():
     """
     Абстрактный базовый класс для всех хэндлеров.
     """
-    def __init__(self, config: Dict[str, Any]={}) -> None:
+    def __init__(self) -> None:
         self.context = Context()
-        self.config = config
+        #self.config = config
+        self.config = self.context.data.get("config", {})
 
     @abstractmethod
-    def do(self) -> None:
+    def do(self, attachment) -> None:
         """
         Выполняет работу над заданием
         Входные и выходные данные - в self.context
@@ -22,3 +23,6 @@ class AbstractTask():
         """
         raise NotImplementedError
 
+    def get_name(self) -> str:
+        pass
+

src/mail_order_bot/task_processor/handlers/__init__.py

@@ -1,10 +1,16 @@
 
-
-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 APIGetStock
+from .delivery_time.local_store import DeliveryPeriodLocalStore
+from .delivery_time.from_config import DeliveryPeriodFromConfig
 from .notifications.test_notifier import TestNotifier
-
-
+from .excel_parcers.excel_extractor import ExcelExtractor
+from .excel_parcers.order_extractor import OrderExtractor
+from .abcp.api_save_order import SaveOrderToTelegram
+
+from .stock_selectors.stock_selector import StockSelector
+
+from .excel_parcers.update_excel_file import UpdateExcelFile
+
+from .email.email_reply_task import EmailReplyTask
 
 

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

@@ -1,8 +1,15 @@
+"""
+Перебирает аттачменты
+Для каждого ордера в аттачменте перебирает позиции
+Для каждой позиции запрашивает остатки и запускает процедуру выбора оптмальной позиции со склада/
+Возможно логику выбора позиции надо вынести из позиции, но пока так
+"""
 import logging
 
 from mail_order_bot.task_processor.abstract_task import AbstractTask
 from mail_order_bot.abcp_api.abcp_provider import AbcpProvider
 from mail_order_bot.credential_provider import CredentialProvider
+from mail_order_bot.order.auto_part_order import OrderStatus
 
 logger = logging.getLogger(__name__)
 
@@ -16,18 +23,17 @@ class APIGetStock(AbstractTask):
         client_login, client_password = credential_provider.get_client_credentials()
         self.client_provider = AbcpProvider(login=client_login, password=client_password)
 
-    def do(self) -> None:
-        attachments = self.context.data.get("attachments", [])
-        for attachment in attachments:
-            order = attachment["order"]
-            for position in order.positions:
-                # Получаем остатки из-под учетной записи клиента
-                client_stock = self.client_provider.get_stock(position.sku, position.manufacturer)
-            
-                position.set_stock(client_stock)
-                position.set_order_item()
-                
-            logger.info(f"Получены позиции со склада для файла {attachment.get('name', "no name")}")
+    def do(self, attachment) -> None:
+
+        order = attachment.get("order", None)
+        for position in order.positions:
+            # Получаем остатки из-под учетной записи клиента
+            client_stock = self.client_provider.get_stock(position.sku, position.manufacturer)
+            position.set_order_item(client_stock)
+            #position.set_order_item()
+
+        logger.info(f"Получены позиции со склада для файла {attachment.get('name', "no name")}")
+
 
     def get_stock(self, sku: str, manufacturer: str) -> int:
-        return self.client_provider.get_stock(sku, manufacturer)
+        return self.client_provider.get_stock(sku, manufacturer)

src/mail_order_bot/task_processor/handlers/abcp/api_save_order.py

@@ -0,0 +1,62 @@
+"""
+Перебирает аттачменты
+Для каждого ордера в аттачменте перебирает позиции
+Для каждой позиции запрашивает остатки и запускает процедуру выбора оптмальной позиции со склада/
+Возможно логику выбора позиции надо вынести из позиции, но пока так
+"""
+import logging
+
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+from mail_order_bot.abcp_api.abcp_provider import AbcpProvider
+from mail_order_bot.credential_provider import CredentialProvider
+
+from mail_order_bot.telegram.client import TelegramClient
+
+logger = logging.getLogger(__name__)
+
+class SaveOrderToTelegram(AbstractTask):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def do(self, attachment) -> None:
+        client = TelegramClient()
+
+        try:
+            order = attachment["order"]
+            positions = order.positions
+            message = "\nОбработка заказа {указать название контрагента}\n"
+            message += f"\nПолучено {len(positions)} позиций от {order.client_id}\n"
+            message += "===============================\n"
+            for position in positions:
+                message += f"{position.sku} - {position.manufacturer} - {position.name} \n"
+                message += f"{position.asking_quantity} x {position.asking_price} = {position.total} \n"
+
+                rejected = position.asking_quantity - position.order_quantity
+                if position.order_quantity == 0:
+                    message += f"Отказ\n"
+                elif rejected:
+                    message += (f"Отказ: {rejected}, запрошено, {position.asking_quantity}, "
+                                f"отгружено {position.order_quantity}, профит {position.profit}\n")
+                else:
+                    message += f"Позиция отгружена полностью, профит {position.profit}\n"
+                message += "-------------------------------\n"
+
+            result = client.send_message(message)
+
+            # Отправка экселя в телеграм
+            excel = attachment["excel"]
+            file = excel.get_file_bytes()
+
+            client.send_document(
+                document=file,
+                filename="document.xlsx"
+            )
+        except Exception as e:
+            logger.error("Ошибка при отправке инфо по заказу в телеграм")
+        else:
+            logger.warning("Инфо по заказу отправлено в телеграм")
+
+    #===============================
+
+
+            

src/mail_order_bot/task_processor/handlers/abcp/stock_selector.py

@@ -1,95 +0,0 @@
-import sys
-from turtle import position
-from typing import Dict, Any, List, Optional
-from decimal import Decimal
-from mail_order_bot.email_processor.order.auto_part_position import AutoPartPosition, PositionStatus
-
-logger = __import__('logging').getLogger(__name__)
-
-
-"""
-1. Получить 2 вида складских остатков
-2. Добавляем цену закупки
-3. Применяем правила фильтрации
-  - можно указать какие правила применены
-4. Выбираем цену по профиту
- - можно указать ограничения
-
-
-
-"""
-class StockSelector:
-    """
-    Класс для выбора оптимального поставщика для позиции заказа.
-    Выполняет фильтрацию складов и выбор наилучшего варианта.
-    """
-    
-    DISTRIBUTOR_ID = "1577730"  # ID локального склада
-    
-    def __init__(self, position: AutoPartPosition, delivery_period: int = 0):
-        """
-        Инициализация StockSelector.
-        
-        Args:
-            position: Позиция заказа
-            delivery_period: Период доставки в днях
-        """
-        self.position = position
-        self.delivery_period = delivery_period
-        self.client_stock = None
-        self.system_stock = None
-
-    
-    def filter_stock(self, client_stock) -> None:
-        """
-        Обрабатывает результат запроса остатков из-под учетной записи клиента и обновляет позицию.
-        
-        Args:
-            client_stock: Результат запроса остатков от API из-под учетной записи клиента
-            system_stock: Результат запроса остатков от API из-под системной учетной записи
-        """
-        self.client_stock = client_stock
-
-        if client_stock["success"]:
-            available_distributors = client_stock["data"]
-            
-            # Для доставки только с локального склада сперва убираем все остальные склады
-            if self.delivery_period == 0:
-                available_distributors = self._filter_only_local_storage(available_distributors)
-            
-            # Отбираем склады по сроку доставки
-            available_distributors = self._filter_proper_delivery_time(available_distributors, self.delivery_period)
-            
-            # Убираем дорогие склады с ценой выше запрошенной
-            available_distributors = self._filter_proper_price(available_distributors)
-            
-            # Убираем отрицательные остатки
-            available_distributors = self._filter_proper_availability(available_distributors)
-        
-            # Добавляем данные о закупочных ценах
-            available_distributors = self._set_system_price(available_distributors)
-
-            # Сортируем по цене
-            available_distributors.sort(key=lambda item: Decimal(item["price"]), reverse=True)
-            
-            return available_distributors
-
-        else:
-            return None
-        
-    def _filter_only_local_storage(self, distributors: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
-        """Фильтрует только локальные склады"""
-        return [item for item in distributors if str(item["distributorId"]) == self.DISTRIBUTOR_ID]
-    
-    def _filter_proper_delivery_time(self, distributors: List[Dict[str, Any]], delivery_period: int) -> List[Dict[str, Any]]:
-        """Фильтрует склады по сроку доставки"""
-        return [item for item in distributors if item["deliveryPeriod"] <= delivery_period]
-    
-    def _filter_proper_price(self, distributors: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
-        """Фильтрует склады по цене (убирает дорогие)"""
-        return [item for item in distributors if Decimal(item["price"]) <= self.position.requested_price]
-    
-    def _filter_proper_availability(self, distributors: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
-        """Фильтрует склады с положительным остатком"""
-        return [item for item in distributors if Decimal(item["availability"]) > 0]
-

src/mail_order_bot/task_processor/handlers/attachment_handler/attachment_handler.py

@@ -1,21 +0,0 @@
-from mail_order_bot.task_processor.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/task_processor/handlers/delivery_time/from_config.py

@@ -0,0 +1,29 @@
+"""
+Устанавливает хардкодом период доставки 0, что означает использование локального склада.
+Для заказчиков, которые должны всегда получать заказ только из наличия
+"""
+
+import logging
+
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+
+logger = logging.getLogger(__name__)
+
+class DeliveryPeriodFromConfig(AbstractTask):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def do(self, attachment) -> None:
+        try:
+            delivery_period = self.config.get("delivery_period")
+
+        except Exception as e:
+            logger.error(f"Ошибка при получении срока доставки из конфига: {e}")
+
+        else:
+            attachment["delivery_period"] = delivery_period
+            logger.warning(f"Срок доставки установлен из конфига - {delivery_period} (ч.)")
+
+
+
+

src/mail_order_bot/task_processor/handlers/delivery_time/from_subject.py

@@ -1,3 +1,7 @@
+"""
+Парсер срока доставки из темы письма
+"""
+
 from mail_order_bot.task_processor.abstract_task import AbstractTask
 
 import logging
@@ -22,25 +26,30 @@ class DeliveryPeriodFromSubject(AbstractTask):
         - Срок переводится в часы (умножается на 24)
         """
         # Получаем тему письма
-        email_subj = self.context.data.get("email_subj", "")
-        if not email_subj:
-            logger.warning("Тема письма не найдена в контексте")
-            email_subj = ""
-        
-        # Парсим срок доставки
-        delivery_days = self._parse_delivery_period(email_subj)
-        
-        # Переводим в часы
-        delivery_time = delivery_days * 24
-        
-        logger.info(f"Извлечен срок доставки из темы: {delivery_days} дней ({delivery_time} часов)")
-        
-        # Сохраняем в каждый элемент attachments
-        attachments = self.context.data.get("attachments", [])
-        for attachment in attachments:
-            attachment["delivery_time"] = delivery_time
-        
-        logger.debug(f"Срок доставки сохранен в {len(attachments)} вложений")
+
+        try:
+            email_subj = self.context.data.get("email_subj", "")
+            if not email_subj:
+                logger.warning("Тема письма не найдена в контексте")
+                email_subj = ""
+
+            # Парсим срок доставки
+            delivery_days = self._parse_delivery_period(email_subj)
+
+            # Переводим в часы
+            delivery_time = delivery_days * 24
+
+            logger.info(f"Извлечен срок доставки из темы: {delivery_days} дней ({delivery_time} часов)")
+
+            # Сохраняем в каждый элемент attachments
+            attachments = self.context.data.get("attachments", [])
+            for attachment in attachments:
+                attachment["delivery_time"] = delivery_time
+                logger.debug(f"Срок доставки для файла {attachment["name"]} установлен как {delivery_time}")
+
+        except Exception as e:
+            logger.error(e)
+            
     
     def _parse_delivery_period(self, subject: str) -> int:
         """

src/mail_order_bot/task_processor/handlers/delivery_time/local_store.py

@@ -0,0 +1,20 @@
+"""
+Устанавливает хардкодом период доставки 0, что означает использование локального склада.
+Для заказчиков, которые должны всегда получать заказ только из наличия
+"""
+
+import logging
+
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+
+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:
+            attachment["delivery_period"] = 0
+            logger.info(f"Срок доставки для файла {attachment["name"]} - только из наличия")

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

@@ -1,19 +0,0 @@
-from mail_order_bot.task_processor.abstract_task import AbstractTask
-
-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:
-            attachment["delivery_period"] = 0
-        logger.info(f"Доставка только с локального склада, срок 1 день.")
-
-
-
-

src/mail_order_bot/task_processor/handlers/email/email_parcer.py

@@ -0,0 +1,62 @@
+"""
+Обрабатывает письмо
+
+"""
+import logging
+
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+from mail_order_bot.email_client.utils import EmailUtils
+
+
+logger = logging.getLogger(__name__)
+
+class EmailParcerException(Exception):
+    pass
+
+
+class EmailParcer(AbstractTask):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def do(self) -> None:
+        # Определить клиента
+        try:
+            email = self.context.data.get("email", None)
+            if email is not None:
+                email_body = EmailUtils.extract_body(email)
+                self.context.data["email_body"] = email_body
+
+                # todo при переводе на основной ящик переделать на другую функцию
+                header_from = EmailUtils.extract_header(email, "From")
+                email_from = EmailUtils.extract_email(header_from)
+                #email_from = EmailUtils.extract_first_sender(email_body)
+                self.context.data["email_from"] = email_from
+
+                email_from_domain = EmailUtils.extract_domain(email_from)
+                self.context.data["email_from_domain"] = email_from_domain
+
+                email_subj = EmailUtils.extract_header(email, "subj")
+                self.context.data["email_subj"] = email_subj
+
+                client = EmailUtils.extract_domain(email_from)
+                self.context.data["client"] = client
+
+                attachments = EmailUtils.extract_attachments(email)
+
+                self.context.data["attachments"] = attachments
+                logger.info(f"Извлечено вложений: {len(attachments)} ")
+
+        except Exception as e:
+            logger.error(e)
+
+            self.context.data["error"].add(
+               "переделать ошибку на нормальную"
+            )
+
+            #raise EmailParcerException(f"Ошибка при парсинге письма {e}") from e
+
+
+
+
+
+

src/mail_order_bot/task_processor/handlers/email/email_reply_task.py

@@ -0,0 +1,82 @@
+import logging
+
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+from email.mime.base import MIMEBase
+from email.utils import formatdate
+from email import encoders
+
+
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+
+logger = logging.getLogger(__name__)
+
+class EmailReplyTaskException(Exception):
+    pass
+
+
+class EmailReplyTask(AbstractTask):
+    """Формирует ответ на входящее письмо с запросом на заказ°"""
+    EMAIl = "zosimovaa@yandex.ru" #"noreply@zapchastiya.ru"
+
+    def do(self, attachment):
+
+        try:
+            email = self.context.data.get("email")
+
+            if not email:
+                raise ValueError("В контексте нет входящего сообщения")
+
+            email_from = self.context.data.get("email_from")
+            if not email_from:
+                raise ValueError("В контексте не определен адрес отправителя")
+
+
+            reply_message = MIMEMultipart()
+
+            email_subj = self.context.data.get("email_subj")
+
+            reply_message["From"] = self.EMAIl
+            reply_message["To"] = email_from
+            #reply_message["Cc"] = self.config.get("reply_to", "")
+            reply_message["Subject"] = f"Re: {email_subj}"
+            reply_message["Date"] = formatdate(localtime=True)
+
+            body = "Автоматический ответ на создание заказа"
+            reply_message.attach(MIMEText(body, "plain", "utf-8"))
+
+
+            self._attach_file(reply_message, attachment)
+
+            self.context.email_client.send_email(reply_message)
+        except Exception as e:
+            logger.error(f"Ошибка при отправке ответа по заказу на email \n{e}")
+        else:
+            logger.warning(f"Сформирован ответ на заказ на email")
+
+    def _attach_file(self, reply_message, attachment):
+        """
+        Args:
+            message: MIMEMultipart
+            file_path:
+        """
+        try:
+
+            part = MIMEBase("application", "octet-stream")
+
+            excel_file = attachment["excel"]
+            excel_file_bytes = excel_file.get_file_bytes()
+            part.set_payload(excel_file_bytes.read())
+
+            encoders.encode_base64(part)
+
+            file_name = attachment["name"]
+            part.add_header(
+                "Content-Disposition",
+                f"attachment; filename= {file_name}"
+            )
+
+            reply_message.attach(part)
+
+        except Exception as e:
+            raise Exception(f"Ошибка при аттаче файла: {str(e)}")

src/mail_order_bot/task_processor/handlers/email/send_email.py

@@ -1,141 +0,0 @@
-
-import smtplib
-from email.mime.multipart import MIMEMultipart
-from email.mime.text import MIMEText
-from email.mime.base import MIMEBase
-from email.utils import formatdate
-from email import encoders
-from abc import ABC, abstractmethod
-import os
-
-
-class AbstractTask(ABC):
-    """Базовый класс для задач"""
-
-    def __init__(self, context, config):
-        self.context = context
-        self.config = config
-
-    @abstractmethod
-    def do(self):
-        """Метод для реализации в подклассах"""
-        pass
-
-
-class EmailReplyTask(AbstractTask):
-    """Класс для ответа на электронные письма"""
-
-    def do(self):
-        """
-        Отправляет ответ на входящее письмо
-
-        Ожидает в self.context:
-            - message: email.message.Message объект входящего письма
-            - attachment: путь к файлу для вложения
-
-        Ожидает в self.config:
-            - reply_to: адрес электронной почты для копии
-            - smtp_host: хост SMTP сервера
-            - smtp_port: порт SMTP сервера
-            - smtp_user: пользователь SMTP
-            - smtp_password: пароль SMTP
-            - from_email: адрес отправителя
-        """
-
-        incoming_message = self.context.get("message")
-        attachment_path = self.context.get("attacnment")
-
-        if not incoming_message:
-            raise ValueError("В context не найдено письмо (message)")
-
-        # Получаем адрес отправителя входящего письма
-        from_addr = incoming_message.get("From")
-        if not from_addr:
-            raise ValueError("Входящее письмо не содержит адреса отправителя")
-
-        # Создаем ответное письмо
-        reply_message = MIMEMultipart()
-
-        # Заголовки ответного письма
-        reply_message["From"] = self.config.get("from_email", "noreply@example.com")
-        reply_message["To"] = from_addr
-        reply_message["Cc"] = self.config.get("reply_to", "")
-        reply_message["Subject"] = f"Re: {incoming_message.get('Subject', '')}"
-        reply_message["Date"] = formatdate(localtime=True)
-
-        # Тело письма
-        body = "Ваш заказ создан"
-        reply_message.attach(MIMEText(body, "plain", "utf-8"))
-
-        # Добавляем вложение если указан путь к файлу
-        if attachment_path and os.path.isfile(attachment_path):
-            self._attach_file(reply_message, attachment_path)
-
-        # Отправляем письмо
-        self._send_email(reply_message, from_addr)
-
-    def _attach_file(self, message, file_path):
-        """
-        Добавляет файл в качестве вложения к письму
-
-        Args:
-            message: MIMEMultipart объект
-            file_path: путь к файлу для вложения
-        """
-        try:
-            with open(file_path, "rb") as attachment:
-                part = MIMEBase("application", "octet-stream")
-                part.set_payload(attachment.read())
-
-            encoders.encode_base64(part)
-
-            file_name = os.path.basename(file_path)
-            part.add_header(
-                "Content-Disposition",
-                f"attachment; filename= {file_name}"
-            )
-
-            message.attach(part)
-        except FileNotFoundError:
-            raise FileNotFoundError(f"Файл не найден: {file_path}")
-        except Exception as e:
-            raise Exception(f"Ошибка при добавлении вложения: {str(e)}")
-
-    def _send_email(self, message, recipient):
-        """
-        Отправляет письмо через SMTP
-
-        Args:
-            message: MIMEMultipart объект письма
-            recipient: адрес получателя
-        """
-        try:
-            smtp_host = self.config.get("smtp_host")
-            smtp_port = self.config.get("smtp_port", 587)
-            smtp_user = self.config.get("smtp_user")
-            smtp_password = self.config.get("smtp_password")
-
-            if not all([smtp_host, smtp_user, smtp_password]):
-                raise ValueError("Не указаны параметры SMTP в config")
-
-            with smtplib.SMTP(smtp_host, smtp_port) as server:
-                server.starttls()
-                server.login(smtp_user, smtp_password)
-
-                # Получаем адреса получателей (основной + копия)
-                recipients = [recipient]
-                reply_to = self.config.get("reply_to")
-                if reply_to:
-                    recipients.append(reply_to)
-
-                server.sendmail(
-                    self.config.get("from_email"),
-                    recipients,
-                    message.as_string()
-                )
-        except smtplib.SMTPException as e:
-            raise Exception(f"Ошибка SMTP: {str(e)}")
-        except Exception as e:
-            raise Exception(f"Ошибка при отправке письма: {str(e)}")
-
-

src/mail_order_bot/task_processor/handlers/excel_parcers/deprecated/order_parser.py

@@ -1,122 +0,0 @@
-import logging
-import pandas as pd
-from typing import Dict, Any, Optional
-from decimal import Decimal
-from io import BytesIO
-#from mail_order_bot.task_processor.handlers.order_position import OrderPosition
-from mail_order_bot.task_processor.abstract_task import AbstractTask
-
-from mail_order_bot.task_processor.order.auto_part_position import AutoPartPosition
-from mail_order_bot.task_processor.order.auto_part_order import AutoPartOrder
-
-logger = logging.getLogger(__name__)
-
-
-class OrderParser(AbstractTask):
-    """
-    Универсальный парсер, настраиваемый через конфигурацию.
-    Подходит для большинства стандартных случаев.
-    """
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-
-    def do(self) -> None:
-
-        # todo сделать проверку на наличие файла и его тип
-
-        attachments = self.context.data.get("attachments", [])
-        for attachment in attachments:
-            file_bytes = BytesIO(attachment['bytes']) # self.context.get("attachment") #
-            delivery_period = attachment.get("delivery_period", 0)
-            #try:
-            df = self._make_dataframe(file_bytes)
-            mapping = self.config["mapping"]
-            client_id = self.config["client_id"]
-            order = AutoPartOrder()
-            attachment["order"] = order
-
-            # Парсим строки
-            positions = []
-            for idx, row in df.iterrows():
-                position = self._parse_row(row, mapping)
-                if position:
-                    position.order_delivery_period = delivery_period
-                    order.add_position(position)
-
-            logger.info(f"Успешно обработано {len(order)} позиций из {len(df)} строк")
-
-            #except Exception as e:
-             #   logger.error(f"Ошибка при обработке файла: {e}")
-            #else:
-            attachment["order"] = order
-
-
-
-    def _parse_row(self, row: pd.Series, mapping: Dict[str, str]) -> Optional[AutoPartPosition]:
-        """Парсит одну строку Excel в OrderPosition"""
-
-        # Проверяем обязательные поля
-        required_fields = ['article', 'price', 'quantity']
-
-        for field in required_fields:
-            if pd.isna(row.get(mapping[field])):
-                logger.warning(f"Позиция не создана - не заполнено поле {mapping[field]}")
-                return None
-
-        price = Decimal(str(row[mapping['price']]).replace(",", ".").strip())
-        quantity = int(row[mapping['quantity']])
-
-        if "total" in mapping.keys():
-            total = Decimal(str(row[mapping['total']]).replace(",", ".").strip())
-        else:
-            total = price * quantity
-
-        if "name" in mapping:
-            name = str(row[mapping.get('name', "")]).strip()
-        else:
-            name = ""
-
-        # Создаем объект позиции
-        position = AutoPartPosition(
-            sku=str(row[mapping['article']]).strip(),
-            manufacturer=str(row[mapping.get('manufacturer', "")]).strip(),
-            name=name,
-            requested_price=price,
-            requested_quantity=quantity,
-            total=total,
-            additional_attrs=self._extract_additional_attrs(row, mapping)
-        )
-        return position
-
-    def _extract_additional_attrs(self, row: pd.Series, mapping: Dict[str, str]) -> Dict[str, Any]:
-        """Извлекает дополнительные атрибуты, не входящие в основную модель"""
-        additional = {}
-        mapped_columns = set(mapping.values())
-
-        for col in row.index:
-            if col not in mapped_columns and not pd.isna(row[col]):
-                additional[col] = row[col]
-
-        return additional
-
-    def _make_dataframe(self, bio) -> pd.DataFrame:
-        # Получаем все данные из файла
-        sheet_name = self.config.get("sheet_name", 0)
-        df_full = pd.read_excel(bio, sheet_name=sheet_name, header=None)
-
-        # Находим индекс строки с заголовком
-        key_field = self.config.get("key_field")
-        header_row_idx = df_full[
-            df_full.apply(lambda row: row.astype(str).str.contains(key_field, case=False, na=False).any(),
-                          axis=1)].index[0]
-
-        # Считываем таблицу с правильным заголовком
-        df = pd.read_excel(bio, header=header_row_idx, sheet_name=sheet_name, engine='calamine')  # openpyxl  calamine
-
-        # Находим индекс первой строки с пустым 'Артикул'
-        first_empty_index = df[df[key_field].isna()].index.min()
-
-        # Обрезаем DataFrame до первой пустой строки (не включая её)
-        df_trimmed = df.loc[:first_empty_index - 1]
-
-        return df_trimmed

src/mail_order_bot/task_processor/handlers/excel_parcers/deprecated/sheet_parcer.py

@@ -1,118 +0,0 @@
-import logging
-import pandas as pd
-from typing import Dict, Any, Optional
-from decimal import Decimal
-from io import BytesIO
-#from mail_order_bot.task_processor.handlers.order_position import OrderPosition
-from mail_order_bot.email_processor.handlers.abstract_task import AbstractTask
-
-from ...order.auto_part_position import AutoPartPosition
-
-logger = logging.getLogger(__name__)
-
-
-class BasicExcelParser(AbstractTask):
-    RESULT_SECTION = "positions"
-    """
-    Универсальный парсер, настраиваемый через конфигурацию.
-    Подходит для большинства стандартных случаев.
-    """
-
-    def do(self) -> None:
-
-        # todo сделать проверку на наличие файла и его тип
-        file_bytes = BytesIO(self.context.get("attachment").content) # self.context.get("attachment") #
-        try:
-            df = self._make_dataframe(file_bytes)
-            # Получаем маппинг колонок из конфигурации
-            mapping = self.config['mapping']
-
-            # Парсим строки
-            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
-
-
-        except Exception as e:
-            logger.error(f"Ошибка при обработке файла: {e}")
-            raise Exception from e
-
-    def _parse_row(self, row: pd.Series, mapping: Dict[str, str]) -> Optional[AutoPartPosition]:
-        """Парсит одну строку Excel в OrderPosition"""
-
-        # Проверяем обязательные поля
-        required_fields = ['article', 'price', 'quantity']
-
-        for field in required_fields:
-            if pd.isna(row.get(mapping[field])):
-                logger.warning(f"Позиция не создана - не заполнено поле {mapping[field]}")
-                return None
-
-        price = Decimal(str(row[mapping['price']]).replace(",", ".").strip())
-        quantity = int(row[mapping['quantity']])
-
-        if "total" in mapping.keys():
-            total = Decimal(str(row[mapping['total']]).replace(",", ".").strip())
-        else:
-            total = price * quantity
-
-        if mapping.get('name', "") in mapping.keys():
-            name = str(row[mapping.get('name', "")]).strip()
-        else:
-            name = ""
-
-        # Создаем объект позиции
-        position = AutoPartPosition(
-            sku=str(row[mapping['article']]).strip(),
-            manufacturer=str(row[mapping.get('manufacturer', "")]).strip(),
-            name=name,
-            requested_price=price,
-            requested_quantity=quantity,
-            total=total,
-            additional_attrs=self._extract_additional_attrs(row, mapping)
-        )
-        return position
-
-    def _extract_additional_attrs(self, row: pd.Series, mapping: Dict[str, str]) -> Dict[str, Any]:
-        """Извлекает дополнительные атрибуты, не входящие в основную модель"""
-        additional = {}
-        mapped_columns = set(mapping.values())
-
-        for col in row.index:
-            if col not in mapped_columns and not pd.isna(row[col]):
-                additional[col] = row[col]
-
-        return additional
-
-    def _make_dataframe(self, bio) -> pd.DataFrame:
-        # Получаем все данные из файла
-        sheet_name = self.config.get("sheet_name", 0)
-        df_full = pd.read_excel(bio, sheet_name=sheet_name, header=None)
-
-        # Находим индекс строки с заголовком
-        key_field = self.config.get("key_field")
-        header_row_idx = df_full[
-            df_full.apply(lambda row: row.astype(str).str.contains(key_field, case=False, na=False).any(),
-                          axis=1)].index[0]
-
-        # Считываем таблицу с правильным заголовком
-        df = pd.read_excel(bio, header=header_row_idx, sheet_name=sheet_name, engine='calamine')  # openpyxl  calamine
-
-        # Находим индекс первой строки с пустым 'Артикул'
-        first_empty_index = df[df[key_field].isna()].index.min()
-
-        # Обрезаем DataFrame до первой пустой строки (не включая её)
-        df_trimmed = df.loc[:first_empty_index - 1]
-
-        return df_trimmed

src/mail_order_bot/task_processor/handlers/excel_parcers/excel_extractor.py

@@ -1,9 +1,13 @@
 import logging
 import pandas as pd
 from io import BytesIO
+
+from mail_order_bot.email_client import EmailUtils
 #from mail_order_bot.task_processor.handlers.order_position import OrderPosition
 from mail_order_bot.task_processor.abstract_task import AbstractTask
 
+from mail_order_bot.parsers.excel_parcer import ExcelFileParcer
+
 logger = logging.getLogger(__name__)
 
 
@@ -13,22 +17,22 @@ class ExcelExtractor(AbstractTask):
     """
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
+        self.excel_config = self.config.get("excel", {})
 
-    def do(self) -> None:
-
-        # todo сделать проверку на наличие файла и его тип
-
-        attachments = self.context.data.get("attachments", [])
-        for attachment in attachments:
+    def do(self, attachment) -> None:
+        try:
             file_bytes = BytesIO(attachment['bytes'])
+            excel_file = ExcelFileParcer(file_bytes, self.excel_config)
+
+        except Exception as e:
+            logger.error(f"Не удалось распарсить файл: \n{e}")
+            attachment["excel"] = None
+
+        else:
+            attachment["excel"] = excel_file
+            logger.warning(f"Произведен успешный парсинг файла")
+            
             
-             # Получаем все данные из файла
-            sheet_name = self.config.get("sheet_name", 0)
-            try:
-                attachment["sheet"] = pd.read_excel(file_bytes, sheet_name=sheet_name, header=None)
-            except Exception as e:
-                attachment["sheet"] = None
-                logger.warning("Не удалось распарсить значение файла")
 
 
 

src/mail_order_bot/task_processor/handlers/excel_parcers/order_extractor.py

@@ -0,0 +1,44 @@
+import logging
+import pandas as pd
+from io import BytesIO
+from mail_order_bot.parsers.order_parcer import OrderParser
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+
+from mail_order_bot.parsers.excel_parcer import ExcelFileParcer
+
+logger = logging.getLogger(__name__)
+
+
+class OrderExtractor(AbstractTask):
+    """
+    Хендлер для каждого вложения считывает эксель файл и сохраняет его контекст
+    """
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.excel_config = self.config.get("excel", {})
+
+
+    def do(self, attachment) -> None:
+        # todo сделать проверку на наличие файла и его тип
+
+        try:
+            delivery_period = attachment.get("delivery_period", 0)
+            mapping = self.excel_config.get("mapping")
+
+            excel_file = attachment.get("excel")
+            client_id = self.config.get("client_id")
+
+            order_parcer = OrderParser(mapping, delivery_period, client_id)
+
+            order_dataframe = excel_file.get_order_rows()
+            order = order_parcer.parse(order_dataframe)
+
+        except Exception as e:
+            logger.error(f"Ошибка при парсинге заказа файла: \n{e}")
+
+        else:
+            attachment["order"] = order
+            logger.warning(f"Обработан файл с заказом, извлечено позиций, {len(order.positions)}")
+
+
+

src/mail_order_bot/task_processor/handlers/excel_parcers/update_excel_file.py

@@ -0,0 +1,54 @@
+import logging
+import pandas as pd
+from io import BytesIO
+# from mail_order_bot.task_processor.handlers.order_position import OrderPosition
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+
+
+from mail_order_bot.order.auto_part_position import PositionStatus
+from mail_order_bot.parsers.excel_parcer import ExcelFileParcer
+
+logger = logging.getLogger(__name__)
+
+
+class UpdateExcelFile(AbstractTask):
+    """
+    Хендлер для каждого вложения считывает эксель файл и сохраняет его контекст
+    """
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.excel_config = self.config.get("excel", {})
+
+    def do(self, attachment) -> None:
+        # todo сделать проверку на наличие файла и его тип
+
+        try:
+            excel_file = attachment.get("excel")
+            order = attachment.get("order")
+            config = self.context.data.get("config", {})
+            excel_config = config.get("excel", {})
+            updatable_fields = excel_config.get("updatable_fields", {})
+
+            for position in order.positions:
+
+                sku = position.sku
+                manufacturer = position.manufacturer
+
+                for key, value in updatable_fields.items():
+
+                    if key == "ordered_quantity":
+                        column = value
+                        value = position.order_quantity
+                        excel_file.set_value(sku, manufacturer, column, value)
+
+                    if key == "ordered_price":
+                        column = value
+                        value = position.order_price
+                        excel_file.set_value(sku, manufacturer, column, value)
+        except Exception as e:
+            logger.error(f"Ошибка при правке excel файла: \n{e}")
+
+        else:
+            logger.warning(f"Файл excel успешно обновлен")
+

src/mail_order_bot/task_processor/handlers/notifications/test_notifier.py

@@ -1,6 +1,6 @@
 import logging
 
-from mail_order_bot.email_processor.handlers.abstract_task import AbstractTask
+from mail_order_bot.task_processor.abstract_task import AbstractTask
 
 logger = logging.getLogger(__name__)
 
@@ -12,4 +12,4 @@ class TestNotifier(AbstractTask):
         print(f"\nПолучено {len(positions)} позиций от {self.context["client"]}:")
         for pos in positions:  # Первые 5
             print(f"  - {pos.sku}: {pos.name} "
-                  f"({pos.requested_quantity} x {pos.requested_price} = {pos.total})")
+                  f"({pos.asking_quantity} x {pos.asking_price} = {pos.total})")

src/mail_order_bot/task_processor/handlers/stock_selectors/stock_selector.py

@@ -0,0 +1,126 @@
+import logging
+import pandas as pd
+from io import BytesIO
+
+from dotenv.parser import Position
+
+from mail_order_bot.parsers.order_parcer import OrderParser
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+from mail_order_bot.order.auto_part_position import AutoPartPosition, PositionStatus
+from mail_order_bot.parsers.excel_parcer import ExcelFileParcer
+from decimal import Decimal
+
+from mail_order_bot.task_processor.abstract_task import AbstractTask
+from mail_order_bot.abcp_api.abcp_provider import AbcpProvider
+from mail_order_bot.credential_provider import CredentialProvider
+from mail_order_bot.order.auto_part_order import OrderStatus
+
+
+from typing import Dict, Any
+from typing import List, Optional
+
+logger = logging.getLogger(__name__)
+
+
+class StockSelector(AbstractTask):
+    DISTRIBUTOR_ID = 1577730  # ID локального склада
+    """
+    Выбирает подходящие позиции со склада
+    """
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        credential_provider = CredentialProvider(context=self.context)
+        # Создаем провайдер для учетной записи клиента
+        client_login, client_password = credential_provider.get_system_credentials()
+        self.client_provider = AbcpProvider(login=client_login, password=client_password)
+
+    def do(self, attachment) -> None:
+        # todo сделать проверку на наличие файла и его тип
+
+        try:
+            order = attachment.get("order", None)
+            delivery_period = attachment.get("delivery_period")
+            for position in order.positions:
+
+                #1. Получаем остатки со складов
+                stock_data = self.client_provider.get_stock(position.sku, position.manufacturer)
+
+                #2. Из данных остатков выбираем оптимальное значение по стратегии
+                if stock_data["success"]:
+                    stock_list = stock_data.get("data", [])
+                    asking_price = position.asking_price
+                    asking_quantity = position.asking_quantity
+
+
+                    optimal_stock_positions = self.get_optimal_stock(stock_list, asking_price, asking_quantity, delivery_period)
+
+                    # 3. Устанавливаем выбранное значение в позицию
+                    if len(optimal_stock_positions):
+                        position.set_order_item(optimal_stock_positions[0])
+                    else:
+                        position.status = PositionStatus.NO_AVAILABLE_STOCK
+                # Мне не очень нравится управление статусами в этом месте, кажется что лучше это делать внутри AutoPartPosition
+                else:
+                    position.status = PositionStatus.STOCK_FAILED
+        except Exception as e:
+            logger.error(f"Ошибка при выборе позиции со складов: {e}")
+
+        else:
+            logger.warning("Определены оптимальные позиции со складов")
+
+
+
+    def get_optimal_stock(self, stock_list, asking_price, asking_quantity, delivery_period):
+        """Выбирает позицию для заказа"""
+
+        # BR-1. Отсекаем склады для заказов из наличия (только локальный склад)
+        stock_list = self._br1_only_local_stock(stock_list)
+
+        # BR-2. Цена не должна превышать цену из заказа
+        #stock_list = self._br2_price_below_asked_price(stock_list, asking_price)
+
+        # BR-3. Срок доставки не должен превышать ожидаемый
+        stock_list = self._br3_delivery_time_shorted_asked_time(stock_list, delivery_period)
+
+        # BR-4. Без отрицательных остатков
+        stock_list = self._br4_only_positive_stock(stock_list)
+
+        # BR-5 Выбираем склад с максимальным профитом
+        stock_list = self._br5_max_profit(stock_list, asking_price, asking_quantity)
+
+        # пока не реализовано
+        # BR-7 Приоритет на склады с полным стоком
+        # BR-8. Сначала оборачиваем локальный склад, потом удаленные
+        # BR-9. Даем немного уйти в минус при заказе из наличия
+
+        return stock_list
+
+    def _br1_only_local_stock(self, stocks):
+        return [item for item in stocks if item["distributorId"] == self.DISTRIBUTOR_ID]
+
+    def _br2_price_below_asked_price(self, distributors: List[Dict[str, Any]], asking_price) -> List[Dict[str, Any]]:
+        """Фильтрует склады по цене (убирает дорогие)"""
+        return [item for item in distributors if Decimal(item["price"]) <= asking_price]
+
+    def _br3_delivery_time_shorted_asked_time(self, distributors: List[Dict[str, Any]], delivery_period) -> List[Dict[str, Any]]:
+        """Фильтрует склады по сроку доставки"""
+        # Вопрос - надо ли ориентироваться на deliveryPeriodMax
+        return [item for item in distributors if item["deliveryPeriod"] <= delivery_period]
+
+    def _br4_only_positive_stock(self, distributors: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
+        """Фильтрует склады с положительным остатком"""
+        return [item for item in distributors if Decimal(item["availability"]) > 0]
+
+    def _br5_max_profit(self, distributors: List[Dict[str, Any]], asking_price, asking_quantity) -> List[Dict[str, Any]]:
+        """Фильтрует склады с положительным остатком"""
+        for item in distributors:
+            item["profit"] = (asking_price - Decimal(item["price"])) * min(asking_quantity, item["availability"])
+
+        distributors.sort(key=lambda item: Decimal(item["profit"]), reverse=False)
+
+        return distributors
+
+
+
+

src/mail_order_bot/task_processor/message.py

@@ -0,0 +1,44 @@
+
+from enum import Enum
+
+
+class LogMessageLevel(Enum):
+    SUCCESS = "SUCCESS"
+    WARNING = "WARNING"
+    ERROR = "ERROR"
+
+class LogMessage:
+    def __init__(self, handler=None, level=None, message=None, error_data=None):
+        self.handler = handler
+        self.level = level
+        self.message = message
+        self.error_data = error_data
+
+    def __str__(self):
+        return self.message
+
+class LogMessageStorage:
+    def __init__(self, filename=None):
+        self.filename = filename
+        self.messages = []
+
+    def append(self, message):
+        self.messages.append(message)
+
+    def check_errors(self) -> bool:
+        fatal_statuses = [message.level == LogMessageLevel.ERROR for message in self.messages]
+        return bool(sum(fatal_statuses))
+
+    def get_messages_log(self) -> str:
+        response = ""
+
+        if self.filename is not None:
+            response += f" Лог обработки файла: {self.filename}"
+
+        for message in self.messages:
+            if len(response):
+                response += "\n"
+            response += f"{message.handler} [{message.level}]: {message.message}"
+            if message.error_data is not None:
+                response += f"\n{message.error_data}"
+

src/mail_order_bot/task_processor/processor.py

@@ -1,7 +1,39 @@
+"""
+Общая логика обработки писем следующая
+
+1. Общая часть
+   - скачиваем письмо
+   - складываем в контекст
+   - обработчик и парсим данные - тело, тема, отправитель
+
+2. Запускаем паплайн
+    - прогоняем обработчик для каждого вложения
+    - каждый обработчик для вложения докидывает результат своей работы
+    - каждый обработчик анализирует общий лог на наличие фатальных ошибок. Если есть - пропускаем шаг.
+    Последний обработчик направляет лог ошибок на администратора
+
+Ограничения:
+- каждое вложение воспринимается как "отдельное письмо", т.е. если клиент в одном письме направит несколько вложений,
+то они будут обрабатываться как отдельные письма, и на каждое будет дан ответ (если он требуется).
+
+
+Исключительные ситуации:
+ - При невозможности создать заказ - пересылаем письмо на администратора с логом обработки вложения
+ - Вложения, которые не являются файлами заказа игнорируем.
+
+
+todo
+ [ ] Нужен класс, который будет хранить сообщения от обработчиков
+    - метод для добавления сообщения
+    - метод для проверки фатальных ошибок
+    - метод для извлечения лога
+
+"""
+
 import os
 import yaml
 import logging
-from typing import Dict, Any
+from typing import Dict, Any, List
 from pathlib import Path
 import threading
 from mail_order_bot.context import Context
@@ -9,8 +41,10 @@ from mail_order_bot.email_client.utils import EmailUtils
 from enum import Enum
 
 from mail_order_bot.task_processor.handlers import *
+from mail_order_bot.task_processor.handlers.email.email_parcer import EmailParcer
+
+from mail_order_bot.task_processor.message import LogMessage, LogMessageLevel, LogMessageStorage
 
-from mail_order_bot.task_processor.handlers import AttachmentHandler
 
 logger = logging.getLogger(__name__)
 
@@ -25,62 +59,56 @@ class RequestStatus(Enum):
 
 
 class TaskProcessor:
-    def __init__(self, configs_path: str):
+    #def __init__(self, configs_path: str):
+    def __init__(self, config: Dict[str, Any]):
         super().__init__()
         self.context = Context()
-        self.configs_path = configs_path
+        #self.configs_path = configs_path
+        self.config = config
         self.status = RequestStatus.NEW
 
     def process_email(self, email):
-        # Очистить контекст
+        # Очистить контекст и запушить туда письмо
         self.context.clear()
-
-        # Сохранить письмо в контекст
         self.context.data["email"] = email
 
-        # Определить клиента
-        email_body = EmailUtils.extract_body(email)
-        self.context.data["email_body"] = email_body
-
-        email_from = EmailUtils.extract_first_sender(email_body)
-        self.context.data["email_from"] = email_from
-
-        email_subj = EmailUtils.extract_header(email, "subj")
-        self.context.data["email_subj"] = email_subj
-
-        client = EmailUtils.extract_domain(email_from)
-        self.context.data["client"] = client
-
-        
         try:
+            # Парсинг письма
+            email_parcer = EmailParcer()
+            email_parcer.do()
+
+            email_sender = self.context.data.get("email_from")
+
             # Определить конфиг для пайплайна
-            config = self._load_config(client)
+            config = self._load_config(email_sender)
             self.context.data["config"] = config
 
-            # Обработка вложений
-            attachments_handler_task = AttachmentHandler()
-            attachments_handler_task.do()
+            pipeline = config["pipeline"]
 
-            # Запустить обработку пайплайна
-            for stage in config["pipeline"]:
-                handler_name = stage["handler"]
-                logger.info(f"Processing handler: {handler_name}")
-                task = globals()[handler_name](stage.get("config", None))
-                task.do()
+            attachments = self.context.data.get("attachments", [])
+            for attachment in attachments:
+                file_name = attachment["name"]
+                logger.warning(f"Начата обработка файла: {file_name} =>")
+
+                attachment["log_messages"] = LogMessageStorage(file_name)
+
+                # Запустить обработку пайплайна
+                for handler_name in pipeline:
+                    logger.info(f"Processing handler: {handler_name}")
+                    task = globals()[handler_name]()
+                    task.do(attachment)
+
+        except Exception as e:
+            logger.error(f"Произошла ошибка: {e}")
 
 
-        except FileNotFoundError:
-            logger.error(f"Конфиг для клиента {client} не найден")
+    def _load_config(self, email_from) -> Dict[str, Any]:
+        if email_from in self.config:
+            return self.config[email_from]
 
-        for attachment in self.context.data["attachments"]:
-            print(attachment["order"].__dict__)
-        #except Exception as e:
-        #    logger.error(f"Произошла другая ошибка: {e}")
+        email_from_domain = EmailUtils.extract_domain(email_from)
 
+        if email_from_domain in self.config:
+            return self.config[email_from_domain]
 
-    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)
+        raise FileNotFoundError

src/mail_order_bot/telegram/__init__.py

@@ -0,0 +1,4 @@
+from mail_order_bot.telegram.client import TelegramClient
+
+__all__ = ['TelegramClient']
+

src/mail_order_bot/telegram/client.py

@@ -0,0 +1,176 @@
+import os
+import logging
+import requests
+from typing import Optional
+from io import BytesIO
+
+logger = logging.getLogger(__name__)
+
+
+class TelegramClient:
+    """
+    Класс для отправки сообщений через Telegram Bot API.
+    
+    Поддерживает отправку:
+    - Текстовых сообщений
+    - Сообщений с вложением (Excel файл в формате BytesIO)
+    """
+    
+    BASE_URL = "https://api.telegram.org/bot"
+    
+    def __init__(self, bot_token: Optional[str] = None, chat_id: Optional[str] = None):
+        """
+        Инициализация TelegramClient.
+        
+        Args:
+            bot_token: Токен бота Telegram. Если не указан, берется из TELEGRAM_BOT_TOKEN
+            chat_id: ID чата для отправки сообщений. Если не указан, берется из TELEGRAM_CHAT_ID
+        """
+        self.bot_token = bot_token or os.getenv('TELEGRAM_BOT_TOKEN')
+        self.chat_id = chat_id or os.getenv('TELEGRAM_CHAT_ID')
+        
+        if not self.bot_token:
+            raise ValueError("Telegram bot token is required. Set TELEGRAM_BOT_TOKEN environment variable or pass bot_token parameter.")
+        
+        if not self.chat_id:
+            raise ValueError("Telegram chat ID is required. Set TELEGRAM_CHAT_ID environment variable or pass chat_id parameter.")
+        
+        self.api_url = f"{self.BASE_URL}{self.bot_token}"
+    
+    def send_message(self, text: str, parse_mode: Optional[str] = None) -> dict:
+        """
+        Отправляет текстовое сообщение в Telegram.
+        
+        Args:
+            text: Текст сообщения для отправки
+            parse_mode: Режим парсинга (HTML, Markdown, MarkdownV2). По умолчанию None
+        
+        Returns:
+            dict: Результат запроса с полями success (bool) и data/error
+        """
+        url = f"{self.api_url}/sendMessage"
+        
+        payload = {
+            "chat_id": self.chat_id,
+            "text": text
+        }
+        
+        if parse_mode:
+            payload["parse_mode"] = parse_mode
+        
+        try:
+            response = requests.post(url, json=payload)
+            response.raise_for_status()
+            
+            result = response.json()
+            if result.get("ok"):
+                logger.debug(f"Сообщение успешно отправлено в Telegram")
+                return {
+                    "success": True,
+                    "data": result.get("result")
+                }
+            else:
+                error_description = result.get("description", "Unknown error")
+                logger.warning(f"Ошибка отправки сообщения в Telegram: {error_description}")
+                return {
+                    "success": False,
+                    "error": error_description
+                }
+        
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Ошибка при отправке сообщения в Telegram: {e}")
+            return {
+                "success": False,
+                "error": str(e)
+            }
+    
+    def send_document(
+        self, 
+        document: BytesIO, 
+        filename: str = "document.xlsx",
+        caption: Optional[str] = None,
+        parse_mode: Optional[str] = None
+    ) -> dict:
+        """
+        Отправляет документ (Excel файл) в Telegram.
+        
+        Args:
+            document: BytesIO объект с содержимым файла
+            filename: Имя файла для отправки (по умолчанию "document.xlsx")
+            caption: Подпись к документу (опционально)
+            parse_mode: Режим парсинга для подписи (HTML, Markdown, MarkdownV2). По умолчанию None
+        
+        Returns:
+            dict: Результат запроса с полями success (bool) и data/error
+        """
+        url = f"{self.api_url}/sendDocument"
+        
+        # Убедимся, что указатель файла находится в начале
+        document.seek(0)
+        
+        files = {
+            "document": (filename, document, "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
+        }
+        
+        data = {
+            "chat_id": self.chat_id
+        }
+        
+        if caption:
+            data["caption"] = caption
+        
+        if parse_mode:
+            data["parse_mode"] = parse_mode
+        
+        try:
+            response = requests.post(url, files=files, data=data)
+            response.raise_for_status()
+            
+            result = response.json()
+            if result.get("ok"):
+                logger.debug(f"Документ успешно отправлен в Telegram")
+                return {
+                    "success": True,
+                    "data": result.get("result")
+                }
+            else:
+                error_description = result.get("description", "Unknown error")
+                logger.warning(f"Ошибка отправки документа в Telegram: {error_description}")
+                return {
+                    "success": False,
+                    "error": error_description
+                }
+        
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Ошибка при отправке документа в Telegram: {e}")
+            return {
+                "success": False,
+                "error": str(e)
+            }
+    
+    def send_message_with_document(
+        self,
+        text: str,
+        document: BytesIO,
+        filename: str = "document.xlsx",
+        parse_mode: Optional[str] = None
+    ) -> dict:
+        """
+        Отправляет сообщение с документом. Текст используется как подпись к документу.
+        
+        Args:
+            text: Текст сообщения (будет использован как подпись к документу)
+            document: BytesIO объект с содержимым файла
+            filename: Имя файла для отправки (по умолчанию "document.xlsx")
+            parse_mode: Режим парсинга для подписи (HTML, Markdown, MarkdownV2). По умолчанию None
+        
+        Returns:
+            dict: Результат запроса с полями success (bool) и data/error
+        """
+        return self.send_document(
+            document=document,
+            filename=filename,
+            caption=text,
+            parse_mode=parse_mode
+        )
+

tests/parsers/__init__.py

@@ -0,0 +1,2 @@
+# Tests for parsers module
+

tests/parsers/test_excel_parcer.py

@@ -0,0 +1,320 @@
+import pytest
+import pandas as pd
+from io import BytesIO
+from unittest.mock import patch
+
+from mail_order_bot.parsers.excel_parcer import ExcelFileParcer
+
+
+class TestExcelFileParcer:
+    """Тесты для класса ExcelFileParcer"""
+
+    @pytest.fixture
+    def sample_config(self):
+        """Базовая конфигурация для тестов"""
+        return {
+            "sheet_name": 0,
+            "key_field": "Артикул",
+            "mapping": {
+                "article": "Артикул",
+                "manufacturer": "Производитель",
+                "name": "Наименование",
+                "price": "Цена",
+                "quantity": "Количество"
+            }
+        }
+
+    @pytest.fixture
+    def sample_excel_bytes(self):
+        """Создает тестовый Excel файл в виде байтов"""
+        df = pd.DataFrame({
+            'Артикул': ['ART001', 'ART002', 'ART003'],
+            'Производитель': ['MAN001', 'MAN002', 'MAN003'],
+            'Наименование': ['Товар 1', 'Товар 2', 'Товар 3'],
+            'Цена': [100.0, 200.0, 300.0],
+            'Количество': [1, 2, 3]
+        })
+        buf = BytesIO()
+        with pd.ExcelWriter(buf, engine='xlsxwriter') as writer:
+            df.to_excel(writer, sheet_name='Sheet1', index=False)
+        buf.seek(0)
+        return buf.getvalue()
+
+    @pytest.fixture
+    def excel_with_header_row(self):
+        """Создает Excel файл с заголовком не в первой строке"""
+        df = pd.DataFrame([
+            ['Заголовок документа', None, None, None, None],
+            ['Артикул', 'Производитель', 'Наименование', 'Цена', 'Количество'],
+            ['ART001', 'MAN001', 'Товар 1', 100.0, 1],
+            ['ART002', 'MAN002', 'Товар 2', 200.0, 2],
+            ['ART003', 'MAN003', 'Товар 3', 300.0, 3],
+            [None, None, None, None, None]  # Пустая строка для обрезки
+        ])
+        buf = BytesIO()
+        with pd.ExcelWriter(buf, engine='xlsxwriter') as writer:
+            df.to_excel(writer, sheet_name='Sheet1', index=False, header=False)
+        buf.seek(0)
+        return buf.getvalue()
+
+    def test_init_with_valid_file(self, sample_excel_bytes, sample_config):
+        """Тест инициализации с валидным файлом"""
+        parser = ExcelFileParcer(sample_excel_bytes, sample_config)
+        
+        assert parser.config == sample_config
+        assert parser.bytes == sample_excel_bytes
+        assert parser.sheet_name == 0
+        assert parser.df is not None
+        assert isinstance(parser.df, pd.DataFrame)
+
+    def test_init_with_custom_sheet_name(self, sample_excel_bytes):
+        """Тест инициализации с кастомным именем листа"""
+        config = {
+            "sheet_name": "Sheet2",
+            "key_field": "Артикул",
+            "mapping": {
+                "article": "Артикул",
+                "manufacturer": "Производитель"
+            }
+        }
+        parser = ExcelFileParcer(sample_excel_bytes, config)
+        assert parser.sheet_name == "Sheet2"
+
+    def test_init_with_default_sheet_name(self, sample_excel_bytes):
+        """Тест инициализации с дефолтным именем листа"""
+        config = {
+            "key_field": "Артикул",
+            "mapping": {
+                "article": "Артикул",
+                "manufacturer": "Производитель"
+            }
+        }
+        parser = ExcelFileParcer(sample_excel_bytes, config)
+        assert parser.sheet_name == 0
+
+    @patch('mail_order_bot.parsers.excel_parcer.pd.read_excel')
+    def test_init_with_invalid_file(self, mock_read_excel, sample_config):
+        """Тест инициализации с невалидным файлом"""
+        mock_read_excel.side_effect = Exception("Ошибка парсинга")
+        invalid_bytes = b"invalid excel content"
+        
+        parser = ExcelFileParcer(invalid_bytes, sample_config)
+        
+        assert parser.df is None
+
+    def test_parse_file_success(self, sample_excel_bytes, sample_config):
+        """Тест успешного парсинга файла"""
+        parser = ExcelFileParcer(sample_excel_bytes, sample_config)
+        assert parser.df is not None
+        assert len(parser.df) > 0
+
+    @patch('mail_order_bot.parsers.excel_parcer.pd.read_excel')
+    def test_parse_file_failure(self, mock_read_excel, sample_config):
+        """Тест обработки ошибки при парсинге файла"""
+        mock_read_excel.side_effect = Exception("Ошибка чтения")
+        invalid_bytes = b"invalid"
+        
+        parser = ExcelFileParcer(invalid_bytes, sample_config)
+        assert parser.df is None
+
+    def test_get_header_row(self, excel_with_header_row, sample_config):
+        """Тест поиска строки заголовка"""
+        parser = ExcelFileParcer(excel_with_header_row, sample_config)
+        header_row = parser._get_header_row()
+        
+        assert header_row == 1  # Заголовок во второй строке (индекс 1)
+
+    def test_get_attr_column(self, excel_with_header_row, sample_config):
+        """Тест поиска индекса колонки по имени"""
+        parser = ExcelFileParcer(excel_with_header_row, sample_config)
+        col_idx = parser._get_attr_column("Артикул")
+        
+        assert isinstance(col_idx, int)
+        assert col_idx >= 0
+
+    def test_get_attr_column_nonexistent(self, excel_with_header_row, sample_config):
+        """Тест поиска несуществующей колонки"""
+        parser = ExcelFileParcer(excel_with_header_row, sample_config)
+        
+        with pytest.raises((IndexError, KeyError)):
+            parser._get_attr_column("Несуществующая колонка")
+
+    def test_get_attr_row(self, excel_with_header_row, sample_config):
+        """Тест поиска строки по артикулу и производителю"""
+        parser = ExcelFileParcer(excel_with_header_row, sample_config)
+        row_idx = parser._get_attr_row("ART001", "MAN001")
+        
+        assert isinstance(row_idx, (int, pd.core.indexes.numeric.Int64Index))
+        # Проверяем, что индекс найден
+        assert row_idx is not None
+
+    def test_get_attr_row_nonexistent(self, excel_with_header_row, sample_config):
+        """Тест поиска несуществующей строки"""
+        parser = ExcelFileParcer(excel_with_header_row, sample_config)
+        
+        with pytest.raises((IndexError, KeyError)):
+            parser._get_attr_row("NONEXISTENT", "NONEXISTENT")
+
+    def test_set_value(self, excel_with_header_row, sample_config):
+        """Тест установки значения в ячейку"""
+        parser = ExcelFileParcer(excel_with_header_row, sample_config)
+        
+        # Получаем исходное значение
+        original_value = parser.df.iloc[2, 3]  # Строка с ART001, колонка "Цена"
+        
+        # Устанавливаем новое значение
+        parser.set_value("ART001", "MAN001", "Цена", 999.0)
+        
+        # Проверяем, что значение изменилось
+        new_value = parser.df.iloc[2, 3]
+        assert new_value == 999.0
+        assert new_value != original_value
+
+    def test_get_file_bytes(self, sample_excel_bytes, sample_config):
+        """Тест получения файла в виде байтов"""
+        parser = ExcelFileParcer(sample_excel_bytes, sample_config)
+        result_bytes = parser.get_file_bytes()
+        
+        assert result_bytes is not None
+        assert hasattr(result_bytes, 'read')
+        assert hasattr(result_bytes, 'seek')
+        
+        # Проверяем, что можно прочитать байты
+        result_bytes.seek(0)
+        content = result_bytes.read()
+        assert len(content) > 0
+
+    def test_get_file_bytes_creates_valid_excel(self, sample_excel_bytes, sample_config):
+        """Тест что get_file_bytes создает валидный Excel файл"""
+        parser = ExcelFileParcer(sample_excel_bytes, sample_config)
+        result_bytes = parser.get_file_bytes()
+        
+        # Пытаемся прочитать созданный файл
+        result_bytes.seek(0)
+        df = pd.read_excel(result_bytes, sheet_name=0, header=None)
+        
+        assert df is not None
+        assert len(df) > 0
+
+    def test_get_order_rows(self, excel_with_header_row, sample_config):
+        """Тест получения строк заказа"""
+        parser = ExcelFileParcer(excel_with_header_row, sample_config)
+        order_rows = parser.get_order_rows()
+        
+        assert order_rows is not None
+        assert isinstance(order_rows, pd.DataFrame)
+        assert len(order_rows) > 0
+        # Проверяем, что пустая строка обрезана
+        assert len(order_rows) == 3  # Только строки с данными
+
+    def test_get_order_rows_with_empty_file(self, sample_config):
+        """Тест получения строк заказа из пустого файла"""
+        # Создаем пустой DataFrame
+        df = pd.DataFrame([['Артикул', 'Производитель'], [None, None]])
+        buf = BytesIO()
+        with pd.ExcelWriter(buf, engine='xlsxwriter') as writer:
+            df.to_excel(writer, sheet_name='Sheet1', index=False, header=False)
+        buf.seek(0)
+        empty_bytes = buf.getvalue()
+        
+        parser = ExcelFileParcer(empty_bytes, sample_config)
+        
+        # Должен вернуть пустой DataFrame или вызвать ошибку
+        try:
+            order_rows = parser.get_order_rows()
+            assert len(order_rows) == 0
+        except (IndexError, KeyError):
+            # Ожидаемое поведение при отсутствии данных
+            pass
+
+    def test_set_value_updates_dataframe(self, excel_with_header_row, sample_config):
+        """Тест что set_value обновляет DataFrame"""
+        parser = ExcelFileParcer(excel_with_header_row, sample_config)
+        
+        # Находим строку с ART002
+        row_idx = parser._get_attr_row("ART002", "MAN002")
+        price_col_idx = parser._get_attr_column("Цена")
+        
+        original_price = parser.df.iloc[row_idx, price_col_idx]
+        
+        # Устанавливаем новое значение
+        parser.set_value("ART002", "MAN002", "Цена", 555.0)
+        
+        # Проверяем обновление
+        updated_price = parser.df.iloc[row_idx, price_col_idx]
+        assert updated_price == 555.0
+        assert updated_price != original_price
+
+    def test_multiple_set_value_operations(self, excel_with_header_row, sample_config):
+        """Тест множественных операций set_value"""
+        parser = ExcelFileParcer(excel_with_header_row, sample_config)
+        
+        # Устанавливаем несколько значений
+        parser.set_value("ART001", "MAN001", "Цена", 111.0)
+        parser.set_value("ART002", "MAN002", "Цена", 222.0)
+        parser.set_value("ART003", "MAN003", "Цена", 333.0)
+        
+        # Проверяем все значения
+        price_col_idx = parser._get_attr_column("Цена")
+        row1_idx = parser._get_attr_row("ART001", "MAN001")
+        row2_idx = parser._get_attr_row("ART002", "MAN002")
+        row3_idx = parser._get_attr_row("ART003", "MAN003")
+        
+        assert parser.df.iloc[row1_idx, price_col_idx] == 111.0
+        assert parser.df.iloc[row2_idx, price_col_idx] == 222.0
+        assert parser.df.iloc[row3_idx, price_col_idx] == 333.0
+
+    def test_get_order_rows_trimmed_correctly(self, sample_config):
+        """Тест что get_order_rows правильно обрезает пустые строки"""
+        # Создаем файл с пустой строкой в середине
+        df = pd.DataFrame([
+            ['Артикул', 'Производитель', 'Наименование'],
+            ['ART001', 'MAN001', 'Товар 1'],
+            ['ART002', 'MAN002', 'Товар 2'],
+            [None, None, None],  # Пустая строка
+            ['ART003', 'MAN003', 'Товар 3'],
+            [None, None, None]  # Еще одна пустая строка
+        ])
+        buf = BytesIO()
+        with pd.ExcelWriter(buf, engine='xlsxwriter') as writer:
+            df.to_excel(writer, sheet_name='Sheet1', index=False, header=False)
+        buf.seek(0)
+        excel_bytes = buf.getvalue()
+        
+        parser = ExcelFileParcer(excel_bytes, sample_config)
+        order_rows = parser.get_order_rows()
+        
+        # Должны остаться только строки до первой пустой
+        assert len(order_rows) == 2  # ART001 и ART002
+
+    @patch('mail_order_bot.parsers.excel_parcer.pd.read_excel')
+    def test_get_order_rows_with_calamine_engine(self, mock_read_excel, sample_config):
+        """Тест что get_order_rows использует calamine engine"""
+        # Создаем мок DataFrame
+        mock_df = pd.DataFrame({
+            'Артикул': ['ART001', 'ART002', None],
+            'Производитель': ['MAN001', 'MAN002', None]
+        })
+        mock_read_excel.return_value = mock_df
+        
+        # Создаем парсер с моком для первого чтения
+        df_init = pd.DataFrame([
+            ['Артикул', 'Производитель'],
+            ['ART001', 'MAN001'],
+            ['ART002', 'MAN002'],
+            [None, None]
+        ])
+        with patch('mail_order_bot.parsers.excel_parcer.pd.read_excel') as mock_init:
+            mock_init.return_value = df_init
+            parser = ExcelFileParcer(b"test", sample_config)
+        
+        # Тестируем get_order_rows
+        with patch('mail_order_bot.parsers.excel_parcer.pd.read_excel') as mock_get:
+            mock_get.return_value = mock_df
+            result = parser.get_order_rows()
+            
+            # Проверяем, что был вызван read_excel с engine='calamine'
+            mock_get.assert_called_once()
+            call_kwargs = mock_get.call_args[1]
+            assert call_kwargs.get('engine') == 'calamine'
+