forked from kodorvan/stcs
320 lines
13 KiB
Python
320 lines
13 KiB
Python
from datetime import datetime
|
||
import logging.config
|
||
from sqlalchemy.sql.sqltypes import DateTime
|
||
|
||
from sqlalchemy import BigInteger, Boolean, Integer, String, ForeignKey
|
||
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
|
||
from sqlalchemy.ext.asyncio import AsyncAttrs, async_sessionmaker, create_async_engine
|
||
from logger_helper.logger_helper import LOGGING_CONFIG
|
||
from sqlalchemy import select, insert
|
||
|
||
logging.config.dictConfig(LOGGING_CONFIG)
|
||
logger = logging.getLogger("models")
|
||
|
||
engine = create_async_engine(url='sqlite+aiosqlite:///db.sqlite3')
|
||
|
||
async_session = async_sessionmaker(engine)
|
||
|
||
|
||
class Base(AsyncAttrs, DeclarativeBase):
|
||
"""Базовый класс для declarative моделей SQLAlchemy с поддержкой async."""
|
||
pass
|
||
|
||
|
||
class User_Telegram_Id(Base):
|
||
"""
|
||
Модель таблицы user_telegram_id.
|
||
|
||
Хранит идентификаторы Telegram пользователей.
|
||
|
||
Атрибуты:
|
||
id (int): Внутренний первичный ключ записи.
|
||
tg_id (int): Уникальный идентификатор пользователя Telegram.
|
||
"""
|
||
__tablename__ = 'user_telegram_id'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
tg_id = mapped_column(BigInteger)
|
||
|
||
|
||
class User_Bybit_API(Base):
|
||
"""
|
||
Модель таблицы user_bybit_api.
|
||
|
||
Хранит API ключи и секреты Bybit для каждого Telegram пользователя.
|
||
|
||
Атрибуты:
|
||
id (int): Внутренний первичный ключ записи.
|
||
tg_id (int): Внешний ключ на Telegram пользователя (user_telegram_id.tg_id).
|
||
api_key (str): API ключ Bybit (уникальный для пользователя).
|
||
secret_key (str): Секретный ключ Bybit (уникальный для пользователя).
|
||
"""
|
||
__tablename__ = 'user_bybit_api'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
tg_id = mapped_column(ForeignKey("user_telegram_id.tg_id"))
|
||
|
||
api_key = mapped_column(String(18), default='None')
|
||
secret_key = mapped_column(String(36), default='None')
|
||
|
||
|
||
class User_Symbol(Base):
|
||
"""
|
||
Модель таблицы user_main_settings.
|
||
|
||
Хранит основные настройки торговли для пользователя.
|
||
|
||
Атрибуты:
|
||
id (int): Внутренний первичный ключ записи.
|
||
tg_id (int): Внешний ключ на Telegram пользователя.
|
||
trading_mode (str): Режим торговли (ForeignKey на trading_modes.mode).
|
||
margin_type (str): Тип маржи (ForeignKey на margin_types.type).
|
||
size_leverage (int): Кредитное плечо.
|
||
starting_quantity (int): Начальный объём позиции.
|
||
martingale_factor (int): Коэффициент мартингейла.
|
||
martingale_step (int): Текущий шаг мартингейла.
|
||
maximal_quantity (int): Максимальное количество шагов мартингейла.
|
||
entry_order_type (str): Тип входа (Market или Limit).
|
||
limit_order_price (str, optional): Цена лимитного ордера, если используется.
|
||
"""
|
||
__tablename__ = 'user_symbols'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
tg_id = mapped_column(ForeignKey("user_telegram_id.tg_id"))
|
||
|
||
symbol = mapped_column(String(18), default='PENGUUSDT')
|
||
|
||
|
||
class Trading_Mode(Base):
|
||
"""
|
||
Справочник доступных режимов торговли.
|
||
|
||
Атрибуты:
|
||
id (int): Первичный ключ.
|
||
mode (str): Уникальный режим (например, 'Long', 'Short').
|
||
"""
|
||
__tablename__ = 'trading_modes'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
mode = mapped_column(String(10), unique=True)
|
||
|
||
|
||
class Margin_type(Base):
|
||
"""
|
||
Справочник типов маржинальной торговли.
|
||
|
||
Атрибуты:
|
||
id (int): Первичный ключ.
|
||
type (str): Тип маржи (например, 'Isolated', 'Cross').
|
||
"""
|
||
__tablename__ = 'margin_types'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
type = mapped_column(String(15), unique=True)
|
||
|
||
|
||
class Trigger(Base):
|
||
"""
|
||
Справочник видов триггеров для сделок.
|
||
|
||
Атрибуты:
|
||
id (int): Первичный ключ.
|
||
trigger (str): Название триггера (например, 'Ручной', 'Автоматический').
|
||
"""
|
||
__tablename__ = 'triggers'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
trigger = mapped_column(String(15), unique=True)
|
||
|
||
|
||
class User_Main_Settings(Base):
|
||
"""
|
||
Основные настройки пользователя для торговли.
|
||
|
||
Атрибуты:
|
||
id (int): Первичный ключ.
|
||
tg_id (int): Внешний ключ на Telegram пользователя.
|
||
trading_mode (str): Режим торговли, FK на trading_modes.mode.
|
||
margin_type (str): Тип маржи, FK на margin_types.type.
|
||
size_leverage (int): Кредитное плечо.
|
||
starting_quantity (int): Начальный объем позиции.
|
||
martingale_factor (int): Коэффициент мартингейла.
|
||
martingale_step (int): Текущий шаг мартингейла.
|
||
maximal_quantity (int): Максимальное число шагов мартингейла.
|
||
entry_order_type (str): Тип ордера входа (Market/Limit).
|
||
limit_order_price (Optional[str]): Цена лимитного ордера, если есть.
|
||
"""
|
||
__tablename__ = 'user_main_settings'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
tg_id = mapped_column(ForeignKey("user_telegram_id.tg_id"))
|
||
|
||
trading_mode = mapped_column(ForeignKey("trading_modes.mode"))
|
||
margin_type = mapped_column(ForeignKey("margin_types.type"))
|
||
switch_mode_enabled = mapped_column(String(15), default="Выключен")
|
||
switch_state = mapped_column(String(10), default='Long')
|
||
size_leverage = mapped_column(Integer(), default=1)
|
||
starting_quantity = mapped_column(Integer(), default=1)
|
||
martingale_factor = mapped_column(Integer(), default=1)
|
||
martingale_step = mapped_column(Integer(), default=1)
|
||
maximal_quantity = mapped_column(Integer(), default=10)
|
||
entry_order_type = mapped_column(String(10), default='Market')
|
||
limit_order_price = mapped_column(String(20), nullable=True)
|
||
|
||
|
||
class User_Risk_Management_Settings(Base):
|
||
"""
|
||
Настройки управления рисками пользователя.
|
||
|
||
Атрибуты:
|
||
id (int): Первичный ключ.
|
||
tg_id (int): Внешний ключ на Telegram пользователя.
|
||
price_profit (int): Процент прибыли для трейда.
|
||
price_loss (int): Процент убытка для трейда.
|
||
max_risk_deal (int): Максимально допустимый риск по сделке в процентах.
|
||
commission_fee (str): Учитывать ли комиссию в расчетах ("Да"/"Нет").
|
||
"""
|
||
__tablename__ = 'user_risk_management_settings'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
tg_id = mapped_column(ForeignKey("user_telegram_id.tg_id"))
|
||
|
||
price_profit = mapped_column(Integer(), default=1)
|
||
price_loss = mapped_column(Integer(), default=1)
|
||
max_risk_deal = mapped_column(Integer(), default=100)
|
||
commission_fee = mapped_column(String(), default="Да")
|
||
|
||
|
||
class User_Condition_Settings(Base):
|
||
"""
|
||
Дополнительные пользовательские условия для торговли.
|
||
|
||
Атрибуты:
|
||
id (int): Первичный ключ.
|
||
tg_id (int): Внешний ключ на Telegram пользователя.
|
||
trigger (str): Тип триггера, FK на triggers.trigger.
|
||
filter_time (str): Временной фильтр.
|
||
filter_volatility (bool): Фильтр по волатильности.
|
||
external_cues (bool): Внешние сигналы.
|
||
tradingview_cues (bool): Сигналы TradingView.
|
||
webhook (str): URL webhook.
|
||
ai_analytics (bool): Использование AI для аналитики.
|
||
"""
|
||
__tablename__ = 'user_condition_settings'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
tg_id = mapped_column(ForeignKey("user_telegram_id.tg_id"))
|
||
|
||
trigger = mapped_column(String(15), default='Ручной')
|
||
filter_time = mapped_column(String(25), default='???')
|
||
filter_volatility = mapped_column(Boolean, default=False)
|
||
external_cues = mapped_column(Boolean, default=False)
|
||
tradingview_cues = mapped_column(Boolean, default=False)
|
||
webhook = mapped_column(String(40), default='')
|
||
ai_analytics = mapped_column(Boolean, default=False)
|
||
|
||
|
||
class User_Additional_Settings(Base):
|
||
"""
|
||
Прочие дополнительные настройки пользователя.
|
||
|
||
Атрибуты:
|
||
id (int): Первичный ключ.
|
||
tg_id (int): Внешний ключ на Telegram пользователя.
|
||
pattern_save (bool): Сохранять ли шаблоны.
|
||
autostart (bool): Автоматический запуск.
|
||
notifications (bool): Получение уведомлений.
|
||
"""
|
||
__tablename__ = 'user_additional_settings'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
tg_id = mapped_column(ForeignKey("user_telegram_id.tg_id"))
|
||
|
||
pattern_save = mapped_column(Boolean, default=False)
|
||
autostart = mapped_column(Boolean, default=False)
|
||
notifications = mapped_column(Boolean, default=False)
|
||
|
||
|
||
class USER_DEALS(Base):
|
||
"""
|
||
Таблица сделок пользователя.
|
||
|
||
Атрибуты:
|
||
id (int): Первичный ключ.
|
||
tg_id (int): Внешний ключ на Telegram пользователя.
|
||
symbol (str): Торговая пара.
|
||
side (str): Направление сделки (Buy/Sell).
|
||
open_price (int): Цена открытия.
|
||
positive_percent (int): Процент доходности.
|
||
"""
|
||
__tablename__ = 'user_deals'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
|
||
tg_id = mapped_column(ForeignKey("user_telegram_id.tg_id"))
|
||
|
||
symbol = mapped_column(String(18), default='PENGUUSDT')
|
||
side = mapped_column(String(10), nullable=False)
|
||
open_price = mapped_column(Integer(), nullable=False)
|
||
positive_percent = mapped_column(Integer(), nullable=False)
|
||
|
||
|
||
class UserTimer(Base):
|
||
"""
|
||
Таймер пользователя для отсроченного запуска сделок.
|
||
|
||
Атрибуты:
|
||
id (int): Первичный ключ.
|
||
tg_id (int): Внешний ключ на Telegram пользователя.
|
||
timer_minutes (int): Количество минут таймера.
|
||
timer_start (datetime): Время начала таймера.
|
||
timer_end (Optional[datetime]): Время окончания таймера (если установлено).
|
||
"""
|
||
__tablename__ = 'user_timers'
|
||
|
||
id: Mapped[int] = mapped_column(primary_key=True)
|
||
tg_id = mapped_column(ForeignKey("user_telegram_id.tg_id"))
|
||
timer_minutes = mapped_column(Integer, nullable=False, default=0)
|
||
timer_start = mapped_column(DateTime, default=datetime.utcnow)
|
||
timer_end = mapped_column(DateTime, nullable=True)
|
||
|
||
|
||
async def async_main():
|
||
"""
|
||
Асинхронное создание всех таблиц и заполнение справочников начальными данными.
|
||
"""
|
||
async with engine.begin() as conn:
|
||
await conn.run_sync(Base.metadata.create_all)
|
||
|
||
# Заполнение таблиц
|
||
modes = ['Long', 'Short', 'Switch', 'Smart']
|
||
for mode in modes:
|
||
result = await conn.execute(select(Trading_Mode).where(Trading_Mode.mode == mode))
|
||
if not result.first():
|
||
logger.info("Заполение таблицы режима торговли")
|
||
await conn.execute(Trading_Mode.__table__.insert().values(mode=mode))
|
||
|
||
types = ['Isolated', 'Cross']
|
||
for type in types:
|
||
result = await conn.execute(select(Margin_type).where(Margin_type.type == type))
|
||
if not result.first():
|
||
logger.info("Заполение таблицы типов маржи")
|
||
await conn.execute(Margin_type.__table__.insert().values(type=type))
|
||
|
||
triggers = ['Ручной', 'Автоматический']
|
||
for trigger in triggers:
|
||
result = await conn.execute(select(Trigger).where(Trigger.trigger == trigger))
|
||
if not result.first():
|
||
logger.info("Заполение таблицы триггеров")
|
||
await conn.execute(Trigger.__table__.insert().values(trigger=trigger))
|