Initial commit

Author: Derssen

.gitignore

@@ -0,0 +1,7 @@
+.env
+__pycache__/
+*.sqlite3
+venv/
+.DS_Store
+.github
+*.zip

config.py

@@ -0,0 +1,53 @@
+import os
+from dotenv import load_dotenv
+
+# Загружаем переменные из .env
+load_dotenv()
+
+class Settings:
+    # Telegram
+    BOT_TOKEN: str = os.getenv("BOT_TOKEN")
+    TARGET_CHAT_ID: int = int(os.getenv("TARGET_CHAT_ID", -1)) # Важный ID чата
+    
+    # Database
+    DATABASE_URL: str = os.getenv("DATABASE_URL", "sqlite+aiosqlite:///balances.sqlite3")
+    
+    # API Credentials
+    ZADARMA_KEY: str = os.getenv("ZADARMA_KEY")
+    ZADARMA_SECRET: str = os.getenv("ZADARMA_SECRET")
+    WAZZUP_TOKEN: str = os.getenv("WAZZUP_TOKEN")
+    DIDWW_KEY: str = os.getenv("DIDWW_KEY")
+    
+    # Constants
+    LOW_BALANCE_THRESHOLD: float = 10.0
+    MIN_TOP_UP_AMOUNT: float = 5.0
+    CALLII_DAILY_COST: float = float(os.getenv("CALLII_DAILY_COST", 2.2))
+    WAZZUP_DAILY_COST: float = float(os.getenv("WAZZUP_DAILY_COST", 400.0))
+    STREAMTELE_MONTHLY_FEE: float = float(os.getenv("STREAMTELE_MONTHLY_FEE", 1500.0))
+    WAZZUP_MONTHLY_FEE: float = float(os.getenv("WAZZUP_MONTHLY_FEE", 6000.0))
+    DIDWW_MONTHLY_FEE: float = float(os.getenv("DIDWW_MONTHLY_FEE", 45.0))
+    WAZZUP_PHONE: str = os.getenv("WAZZUP_PHONE", "+6281239838440")
+    SERVICE_CURRENCIES: dict = {
+        'Zadarma': 'USD',
+        'Wazzup24 Подписка': 'RUB',
+        'Wazzup24 Баланс номера': 'RUB',
+        'DIDWW': 'USD',
+        'Streamtele': 'UAH',
+        'Callii': 'USD',
+    }
+    CURRENCY_SIGNS: dict = {
+        'USD': '$',
+        'UAH': '₴',
+        'RUB': '₽',
+    }
+    # Статус активности API-сервисов
+    API_SERVICE_STATUSES: dict = {
+        'Zadarma': os.getenv("ZADARMA_ENABLED", "True").lower() in ('true', '1', 't'),
+        'DIDWW': os.getenv("DIDWW_ENABLED", "True").lower() in ('true', '1', 't'),
+    }
+
+SETTINGS = Settings()
+
+# Проверка, что критические переменные загружены
+if not SETTINGS.BOT_TOKEN or SETTINGS.TARGET_CHAT_ID == -1:
+    raise ValueError("BOT_TOKEN or TARGET_CHAT_ID is not configured correctly.")

db/models.py

@@ -0,0 +1,131 @@
+from sqlalchemy import Column, Integer, String, Float, Boolean, DateTime, text
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
+from sqlalchemy.orm import sessionmaker, declarative_base
+from datetime import datetime
+import pytz
+from config import SETTINGS
+
+# Используем вашу таймзону
+TIMEZONE = pytz.timezone('Europe/Kyiv') 
+
+Base = declarative_base()
+
+class Service(Base):
+    __tablename__ = 'services'
+    
+    id = Column(Integer, primary_key=True)
+    name = Column(String, unique=True, nullable=False)
+    
+    # Мониторинг баланса
+    last_balance = Column(Float, default=0.0)
+    low_balance_alert_sent = Column(Boolean, default=False)
+    currency = Column(String, default='USD')
+    daily_cost = Column(Float, nullable=True)
+    monthly_fee = Column(Float, nullable=True)
+    
+    # Для Callii и Streamtele
+    next_alert_date = Column(DateTime, nullable=True) 
+    next_monthly_alert = Column(DateTime, nullable=True)
+    
+    def __repr__(self):
+        return f"<Service(name='{self.name}', balance={self.last_balance})>"
+
+# Инициализация
+async def init_db(database_url: str):
+    engine = create_async_engine(database_url, echo=False)
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
+    
+    AsyncSessionLocal = sessionmaker(
+        autocommit=False,
+        autoflush=False,
+        bind=engine,
+        class_=AsyncSession,
+        expire_on_commit=False,
+    )
+    return AsyncSessionLocal
+
+# Вспомогательная функция для добавления начальных данных
+async def initialize_services(SessionLocal):
+    async with SessionLocal() as session:
+        # --- simple sqlite migration to add new columns if missing ---
+        pragma_stmt = text("PRAGMA table_info(services)")
+        result = await session.execute(pragma_stmt)
+        columns = {row[1] for row in result.fetchall()}
+
+        alter_statements = []
+        if 'currency' not in columns:
+            alter_statements.append("ALTER TABLE services ADD COLUMN currency VARCHAR")
+        if 'daily_cost' not in columns:
+            alter_statements.append("ALTER TABLE services ADD COLUMN daily_cost FLOAT")
+        if 'monthly_fee' not in columns:
+            alter_statements.append("ALTER TABLE services ADD COLUMN monthly_fee FLOAT")
+        if 'next_monthly_alert' not in columns:
+            alter_statements.append("ALTER TABLE services ADD COLUMN next_monthly_alert DATETIME")
+
+        for stmt in alter_statements:
+            try:
+                await session.execute(text(stmt))
+            except Exception:
+                # ignore if already exists or other minor issues
+                pass
+        if alter_statements:
+            await session.commit()
+
+        services_to_add = [
+            # API сервисы
+            {
+                'name': 'Zadarma',
+                'last_balance': 0.0,
+                'currency': SETTINGS.SERVICE_CURRENCIES.get('Zadarma', 'USD'),
+            },
+            # Wazzup разделён: подписка (ежемесячно) и баланс номера (ежедневный расход)
+            {
+                'name': 'Wazzup24 Подписка',
+                'last_balance': 0.0,
+                'currency': SETTINGS.SERVICE_CURRENCIES.get('Wazzup24 Подписка', 'RUB'),
+                'monthly_fee': SETTINGS.WAZZUP_MONTHLY_FEE,
+                'next_monthly_alert': TIMEZONE.localize(datetime(2025, 12, 11, 10, 0)),
+            },
+            {
+                'name': 'Wazzup24 Баланс номера',
+                'last_balance': 0.0,
+                'currency': SETTINGS.SERVICE_CURRENCIES.get('Wazzup24 Баланс номера', 'RUB'),
+                'daily_cost': SETTINGS.WAZZUP_DAILY_COST,
+                'next_alert_date': TIMEZONE.localize(datetime(2025, 12, 11, 10, 0)),
+            },
+            {
+                'name': 'DIDWW',
+                'last_balance': 0.0,
+                'currency': SETTINGS.SERVICE_CURRENCIES.get('DIDWW', 'USD'),
+                'monthly_fee': SETTINGS.DIDWW_MONTHLY_FEE,
+                'next_monthly_alert': TIMEZONE.localize(datetime(2025, 12, 20, 10, 0)),
+            },
+            # Callii (Управляемый FSM)
+            {
+                'name': 'Callii',
+                'next_alert_date': TIMEZONE.localize(datetime(2025, 12, 11, 10, 0)),
+                'currency': SETTINGS.SERVICE_CURRENCIES.get('Callii', 'USD'),
+                'daily_cost': SETTINGS.CALLII_DAILY_COST,
+            }, 
+            # Streamtele (Ежемесячное напоминание)
+            {
+                'name': 'Streamtele',
+                'currency': SETTINGS.SERVICE_CURRENCIES.get('Streamtele', 'UAH'),
+                'monthly_fee': SETTINGS.STREAMTELE_MONTHLY_FEE,
+                'next_monthly_alert': TIMEZONE.localize(datetime(2025, 12, 11, 10, 0)),
+            },
+        ]
+
+        from sqlalchemy.future import select # Дополнительный импорт нужен для 'select'
+        
+        for data in services_to_add:
+            # ИСПРАВЛЕНО: Используем select и where для поиска по уникальному полю 'name'
+            stmt = select(Service).filter(Service.name == data['name'])
+            result = await session.execute(stmt)
+            exists = result.scalar_one_or_none()
+            
+            if not exists:
+                session.add(Service(name=data['name'], **{k: v for k, v in data.items() if k != 'name'}))
+        
+        await session.commit()

handlers/balance.py

@@ -0,0 +1,83 @@
+from aiogram import Router
+from aiogram.filters import Command
+from aiogram.types import Message
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.future import select
+from db.models import Service, TIMEZONE
+from services.api_clients import API_CLIENTS
+from config import SETTINGS # Новый импорт
+
+router = Router()
+
+CURRENCY_SIGNS = SETTINGS.CURRENCY_SIGNS
+MANUAL_APPROX_VALUES = {
+    'Streamtele': SETTINGS.STREAMTELE_MONTHLY_FEE,
+    'Callii': 10.0,
+    'Wazzup24 Подписка': SETTINGS.WAZZUP_MONTHLY_FEE,
+    'Wazzup24 Баланс номера': SETTINGS.WAZZUP_DAILY_COST,
+}
+MANUAL_SERVICES = ('Streamtele', 'Callii', 'Wazzup24 Подписка', 'Wazzup24 Баланс номера')
+
+@router.message(Command("balance"))
+async def handle_balance_command(message: Message, session: AsyncSession):
+    
+    response_parts = ["💰 **Текущие балансы сервисов:**"]
+    
+    # --- 1. Проверка API-сервисов ---
+    for service_name, client in API_CLIENTS.items():
+        
+        if not SETTINGS.API_SERVICE_STATUSES.get(service_name, False):
+            # Сервис отключен
+            response_parts.append(f"• **{service_name}:** _Отключен в конфигурации_ 🚫")
+            continue
+            
+        try:
+            current_balance = await client.get_balance()
+            if current_balance is None:
+                response_parts.append(f"• **{service_name}:** _Баланс через API недоступен_ ⚙️")
+            else:
+                currency = SETTINGS.SERVICE_CURRENCIES.get(service_name, 'USD')
+                symbol = CURRENCY_SIGNS.get(currency, currency)
+                response_parts.append(f"• **{service_name}:** `{symbol}{current_balance:.2f}` (API)")
+        except Exception as e:
+            response_parts.append(f"• **{service_name}:** Ошибка API (см. логи)")
+
+    # --- 2. Проверка ручных сервисов ---
+    stmt = select(Service).where(Service.name.in_(MANUAL_SERVICES))
+    manual_result = await session.execute(stmt)
+    manual_services = {service.name: service for service in manual_result.scalars()}
+
+    for name in MANUAL_SERVICES:
+        approx = MANUAL_APPROX_VALUES.get(name, 0.0)
+        service = manual_services.get(name)
+        currency = SETTINGS.SERVICE_CURRENCIES.get(name, 'USD')
+        symbol = CURRENCY_SIGNS.get(currency, currency)
+
+        if name == 'Callii':
+            next_date = service.next_alert_date.astimezone(TIMEZONE).strftime('%Y-%m-%d') if service and service.next_alert_date else "N/A"
+            response_parts.append(
+                f"• **{name}:** `{symbol}{approx:.2f}` (примерно)\n"
+                f"  _След. оплата:_ **{next_date}**"
+            )
+        elif name == 'Streamtele':
+            next_monthly = service.next_monthly_alert.astimezone(TIMEZONE).strftime('%Y-%m-%d') if service and service.next_monthly_alert else "N/A"
+            response_parts.append(
+                f"• **{name}:** Подписка: `{symbol}{approx:.2f}`)\n"
+                f"  _След. оплата:_ **{next_monthly}**"
+            )
+        elif name == 'Wazzup24 Подписка':
+            next_monthly = service.next_monthly_alert.astimezone(TIMEZONE).strftime('%Y-%m-%d') if service and service.next_monthly_alert else "N/A"
+            response_parts.append(
+                f"• **{name}:** `{symbol}{approx:.2f}`\n"
+                f"  _След. оплата:_ **{next_monthly}**"
+            )
+        elif name == 'Wazzup24 Баланс номера':
+            next_daily = service.next_alert_date.astimezone(TIMEZONE).strftime('%Y-%m-%d') if service and service.next_alert_date else "N/A"
+            current_balance = service.last_balance if service and service.last_balance is not None else approx
+            response_parts.append(
+                f"• **{name}:** `{symbol}{current_balance:.1f}`\n"
+                f"  _След. оплата:_ **{next_daily}**"
+            )
+
+
+    await message.answer('\n'.join(response_parts))

handlers/callii.py

@@ -0,0 +1,75 @@
+from aiogram import Router, F, Bot
+from aiogram.types import CallbackQuery, Message
+from aiogram.fsm.context import FSMContext
+from aiogram.fsm.state import State, StatesGroup
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.future import select
+from db.models import Service, TIMEZONE
+from config import SETTINGS
+from datetime import datetime, timedelta
+import math
+
+router = Router()
+
+# FSM States
+class CalliiPayment(StatesGroup):
+    waiting_for_amount = State()
+
+@router.callback_query(F.data == "callii_paid")
+async def process_callii_paid(callback: CallbackQuery, state: FSMContext):
+    """Обработка нажатия кнопки 'Оплатил'."""
+    await callback.message.edit_text(
+        "Спасибо за оплату. **Введите сумму пополнения (числом в USD):**"
+    )
+    await state.set_state(CalliiPayment.waiting_for_amount)
+    await callback.answer() # Убираем "часы" с кнопки
+
+@router.message(CalliiPayment.waiting_for_amount, F.text.regexp(r'^\d+(\.\d{1,2})?$'))
+async def process_callii_amount(message: Message, state: FSMContext, session: AsyncSession):
+    """Обработка ввода суммы пополнения и расчет следующей даты."""
+    
+    try:
+        amount = float(message.text)
+        daily_cost = SETTINGS.CALLII_DAILY_COST
+        
+        # Расчет дней (округляем до меньшего)
+        days = math.floor(amount / daily_cost)
+        
+        if days < 1:
+            await message.answer(
+                "Сумма слишком мала для покрытия дневного расхода. Пожалуйста, введите сумму, достаточную для хотя бы одного дня."
+            )
+            return
+
+        # Расчет следующей даты
+        next_alert_datetime = datetime.now(TIMEZONE) + timedelta(days=days)
+        # Устанавливаем время оповещения на 10:00
+        next_alert_datetime = next_alert_datetime.replace(hour=10, minute=0, second=0, microsecond=0)
+
+        # Обновление БД
+        stmt = select(Service).where(Service.name == 'Callii')
+        result = await session.execute(stmt)
+        service = result.scalar_one()
+        
+        service.next_alert_date = next_alert_datetime
+        await session.commit()
+        
+        await message.answer(
+            f"💰 **Платеж Callii обработан!**\n"
+            f"Сумма: **${amount:.2f}**\n"
+            f"Дней покрытия (расход ${daily_cost}/день): **{days}**.\n"
+            f"Следующий контроль баланса запланирован на **{next_alert_datetime.strftime('%Y-%m-%d в 10:00')}**."
+        )
+        await state.clear()
+        
+    except ValueError:
+        await message.answer("Ошибка: Пожалуйста, введите сумму пополнения как число.")
+    except Exception as e:
+        await message.answer(f"Произошла ошибка при сохранении данных: {e}")
+        await state.clear()
+
+
+@router.message(CalliiPayment.waiting_for_amount)
+async def process_callii_amount_invalid(message: Message):
+    """Обработка невалидного ввода суммы."""
+    await message.answer("Неверный формат. Пожалуйста, введите сумму числом (например, 50 или 50.50).")