forked from kodorvan/stcs
Compare commits
67 Commits
42f0f8ddc0
...
stable
| Author | SHA1 | Date | |
|---|---|---|---|
| 8706449c78 | |||
| 89a3c70e4b | |||
| 98cc3c248c | |||
|
|
cb6499e347 | ||
| 7494f85202 | |||
|
|
867802b2a7 | ||
| 28c9614ecb | |||
|
|
1f123c77e7 | ||
|
|
56729c287b | ||
|
|
f268d3290b | ||
|
|
856169cba9 | ||
| d6b36799dc | |||
|
|
0bc74ed188 | ||
| 0b3e9ff476 | |||
|
|
d7b558664b | ||
| ec7e10f7a1 | |||
|
|
34922e6998 | ||
| 37257d2ec2 | |||
|
|
7c85c03d10 | ||
|
|
39bbe8d997 | ||
| 88c358b90e | |||
|
|
3ba32660ea | ||
|
|
e0167ea406 | ||
|
|
dbf0a30d54 | ||
| a7a23a4662 | |||
|
|
78f21e6718 | ||
|
|
6416bd6dc9 | ||
|
|
29c168e31d | ||
| 245cadf650 | |||
|
|
e043a2429f | ||
|
|
a8119d2811 | ||
|
|
7e4c936ef5 | ||
|
|
d8866af185 | ||
| 8d32439a15 | |||
|
|
9497cca3e0 | ||
| 690d793e8c | |||
|
|
a0ef48810a | ||
| ab752b5dd8 | |||
|
|
a2164853d9 | ||
|
|
92bb052151 | ||
| ca7bd5c795 | |||
|
|
62e923cefa | ||
|
|
faae2475c1 | ||
|
|
3ef8eae997 | ||
|
|
2bebada215 | ||
|
|
8be1636279 | ||
|
|
b756aadf26 | ||
| 46c890f7af | |||
|
|
f10500cc79 | ||
| 2d7acb491e | |||
|
|
d767399988 | ||
|
|
89603f0b62 | ||
|
|
14f2a9e773 | ||
|
|
a43fc6a66b | ||
|
|
869458b2e1 | ||
|
|
07948d93cf | ||
| 12d1db16d3 | |||
|
|
7350c86927 | ||
| 0a369b10f2 | |||
| 7b1a803db4 | |||
| 9fcd92cc72 | |||
| 97a199f31e | |||
| 951bc15957 | |||
| 5937058899 | |||
| f0732607e2 | |||
| 56af1d8f3b | |||
| 9f069df68a |
3
.gitignore
vendored
3
.gitignore
vendored
@@ -210,3 +210,6 @@ cython_debug/
|
|||||||
marimo/_static/
|
marimo/_static/
|
||||||
marimo/_lsp/
|
marimo/_lsp/
|
||||||
__marimo__/
|
__marimo__/
|
||||||
|
|
||||||
|
stcs_venv
|
||||||
|
venv
|
||||||
|
|||||||
16
README.md
16
README.md
@@ -1,6 +1,11 @@
|
|||||||
Crypto Trading Telegram Bot
|
# Crypto Trading Telegram Bot by [KODORVAN](https://git.svoboda.works/kodorvan)
|
||||||
|
|
||||||
Этот бот — автоматизированный торговый помощник для работы с криптовалютной биржей Bybit на основе стратегии мартингейла. Он позволяет торговать бессрочными контрактами с управлением рисками, тейк-профитами, стоп-лоссами и кредитным плечом.
|
Автоматизированный торговый помощник для работы с криптовалютной биржей Bybit на основе стратегии мартингейла.<br>
|
||||||
|
Он позволяет торговать бессрочными контрактами с управлением рисками, тейк-профитами, стоп-лоссами и кредитным плечом.
|
||||||
|
|
||||||
|
|
||||||
|
**Разработано командой [КОДОРВАНЬ](https://git.svoboda.works/kodorvan)**<br>
|
||||||
|
_Мы окажем полное содействие и поддержку, пишите в телеграм: https://t.me/kodorvan_
|
||||||
|
|
||||||
## Основные возможности
|
## Основные возможности
|
||||||
|
|
||||||
@@ -59,7 +64,12 @@ nvim .env
|
|||||||
alembic upgrade head
|
alembic upgrade head
|
||||||
```
|
```
|
||||||
|
|
||||||
5. Запустите бота:
|
6. Убедитесь в том, что установлен redis и открыт порт 6789
|
||||||
|
```bash
|
||||||
|
sudo ufw allow 6789
|
||||||
|
```
|
||||||
|
|
||||||
|
7. Запустите бота:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
python run.py
|
python run.py
|
||||||
|
|||||||
@@ -84,7 +84,7 @@ path_separator = os
|
|||||||
# database URL. This is consumed by the user-maintained env.py script only.
|
# database URL. This is consumed by the user-maintained env.py script only.
|
||||||
# other means of configuring database URLs may be customized within the env.py
|
# other means of configuring database URLs may be customized within the env.py
|
||||||
# file.
|
# file.
|
||||||
sqlalchemy.url = sqlite+aiosqlite:///./database/db/stcs.db
|
sqlalchemy.url = sqlite+aiosqlite:///./database/stcs.db
|
||||||
|
|
||||||
|
|
||||||
[post_write_hooks]
|
[post_write_hooks]
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
"""Added side_mode column
|
"""initial
|
||||||
|
|
||||||
Revision ID: fbf4e3658310
|
Revision ID: f6e7eb3f25c0
|
||||||
Revises:
|
Revises:
|
||||||
Create Date: 2025-10-22 13:08:02.317419
|
Create Date: 2025-11-12 22:53:02.189445
|
||||||
|
|
||||||
"""
|
"""
|
||||||
from typing import Sequence, Union
|
from typing import Sequence, Union
|
||||||
@@ -12,7 +12,7 @@ import sqlalchemy as sa
|
|||||||
|
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
# revision identifiers, used by Alembic.
|
||||||
revision: str = 'fbf4e3658310'
|
revision: str = 'f6e7eb3f25c0'
|
||||||
down_revision: Union[str, Sequence[str], None] = None
|
down_revision: Union[str, Sequence[str], None] = None
|
||||||
branch_labels: Union[str, Sequence[str], None] = None
|
branch_labels: Union[str, Sequence[str], None] = None
|
||||||
depends_on: Union[str, Sequence[str], None] = None
|
depends_on: Union[str, Sequence[str], None] = None
|
||||||
@@ -21,12 +21,12 @@ depends_on: Union[str, Sequence[str], None] = None
|
|||||||
def upgrade() -> None:
|
def upgrade() -> None:
|
||||||
"""Upgrade schema."""
|
"""Upgrade schema."""
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
# ### commands auto generated by Alembic - please adjust! ###
|
||||||
op.add_column('user_deals', sa.Column('side_mode', sa.String(), nullable=True))
|
pass
|
||||||
# ### end Alembic commands ###
|
# ### end Alembic commands ###
|
||||||
|
|
||||||
|
|
||||||
def downgrade() -> None:
|
def downgrade() -> None:
|
||||||
"""Downgrade schema."""
|
"""Downgrade schema."""
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
# ### commands auto generated by Alembic - please adjust! ###
|
||||||
op.drop_column('user_deals', 'side_mode')
|
pass
|
||||||
# ### end Alembic commands ###
|
# ### end Alembic commands ###
|
||||||
@@ -15,7 +15,7 @@ async def get_bybit_client(tg_id: int) -> HTTP | None:
|
|||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
api_key, api_secret = await rq.get_user_api(tg_id=tg_id)
|
api_key, api_secret = await rq.get_user_api(tg_id=tg_id)
|
||||||
return HTTP(api_key=api_key, api_secret=api_secret)
|
return HTTP(demo=True, api_key=api_key, api_secret=api_secret)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error("Error getting bybit client for user %s: %s", tg_id, e)
|
logger.error("Error getting bybit client for user %s: %s", tg_id, e)
|
||||||
return None
|
return None
|
||||||
|
|||||||
@@ -38,7 +38,7 @@ async def get_active_positions(tg_id: int) -> list | None:
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
async def get_active_positions_by_symbol(tg_id: int, symbol: str) -> dict | None:
|
async def get_active_positions_by_symbol(tg_id: int, symbol: str):
|
||||||
"""
|
"""
|
||||||
Get active positions for a user by symbol
|
Get active positions for a user by symbol
|
||||||
"""
|
"""
|
||||||
@@ -62,8 +62,12 @@ async def get_active_positions_by_symbol(tg_id: int, symbol: str) -> dict | None
|
|||||||
)
|
)
|
||||||
return None
|
return None
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error("Error getting active positions for user %s: %s", tg_id, e)
|
errors = str(e)
|
||||||
return None
|
if errors.startswith("Permission denied, please check your API key permissions"):
|
||||||
|
return "Invalid API key permissions"
|
||||||
|
else:
|
||||||
|
logger.error("Error getting active positions for user %s: %s", tg_id, e)
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
async def get_active_orders(tg_id: int) -> list | None:
|
async def get_active_orders(tg_id: int) -> list | None:
|
||||||
|
|||||||
@@ -28,11 +28,9 @@ async def start_trading_cycle(
|
|||||||
symbol = await rq.get_user_symbol(tg_id=tg_id)
|
symbol = await rq.get_user_symbol(tg_id=tg_id)
|
||||||
additional_data = await rq.get_user_additional_settings(tg_id=tg_id)
|
additional_data = await rq.get_user_additional_settings(tg_id=tg_id)
|
||||||
risk_management_data = await rq.get_user_risk_management(tg_id=tg_id)
|
risk_management_data = await rq.get_user_risk_management(tg_id=tg_id)
|
||||||
user_deals_data = await rq.get_user_deal_by_symbol(
|
|
||||||
tg_id=tg_id, symbol=symbol
|
|
||||||
)
|
|
||||||
trade_mode = additional_data.trade_mode
|
trade_mode = additional_data.trade_mode
|
||||||
switch_side = additional_data.switch_side
|
switch_side = additional_data.switch_side
|
||||||
|
side = additional_data.side
|
||||||
margin_type = additional_data.margin_type
|
margin_type = additional_data.margin_type
|
||||||
leverage = additional_data.leverage
|
leverage = additional_data.leverage
|
||||||
order_quantity = additional_data.order_quantity
|
order_quantity = additional_data.order_quantity
|
||||||
@@ -41,21 +39,11 @@ async def start_trading_cycle(
|
|||||||
max_bets_in_series = additional_data.max_bets_in_series
|
max_bets_in_series = additional_data.max_bets_in_series
|
||||||
take_profit_percent = risk_management_data.take_profit_percent
|
take_profit_percent = risk_management_data.take_profit_percent
|
||||||
stop_loss_percent = risk_management_data.stop_loss_percent
|
stop_loss_percent = risk_management_data.stop_loss_percent
|
||||||
total_commission = 0
|
commission_fee = risk_management_data.commission_fee
|
||||||
|
commission_place = risk_management_data.commission_place
|
||||||
get_side = "Buy"
|
|
||||||
|
|
||||||
if user_deals_data:
|
|
||||||
get_side = user_deals_data.last_side or "Buy"
|
|
||||||
|
|
||||||
if trade_mode == "Switch":
|
if trade_mode == "Switch":
|
||||||
if switch_side == "По направлению":
|
side = side
|
||||||
side = get_side
|
|
||||||
else:
|
|
||||||
if get_side == "Buy":
|
|
||||||
side = "Sell"
|
|
||||||
else:
|
|
||||||
side = "Buy"
|
|
||||||
else:
|
else:
|
||||||
if trade_mode == "Long":
|
if trade_mode == "Long":
|
||||||
side = "Buy"
|
side = "Buy"
|
||||||
@@ -66,11 +54,32 @@ async def start_trading_cycle(
|
|||||||
tg_id=tg_id,
|
tg_id=tg_id,
|
||||||
symbol=symbol,
|
symbol=symbol,
|
||||||
mode=0)
|
mode=0)
|
||||||
await set_margin_mode(tg_id=tg_id, margin_mode=margin_type)
|
|
||||||
await set_leverage(
|
await rq.set_user_deal(
|
||||||
tg_id=tg_id,
|
tg_id=tg_id,
|
||||||
symbol=symbol,
|
symbol=symbol,
|
||||||
|
current_step=1,
|
||||||
|
current_series=1,
|
||||||
|
trade_mode=trade_mode,
|
||||||
|
side_mode=switch_side,
|
||||||
|
margin_type=margin_type,
|
||||||
leverage=leverage,
|
leverage=leverage,
|
||||||
|
order_quantity=order_quantity,
|
||||||
|
trigger_price=trigger_price,
|
||||||
|
martingale_factor=martingale_factor,
|
||||||
|
max_bets_in_series=max_bets_in_series,
|
||||||
|
take_profit_percent=take_profit_percent,
|
||||||
|
stop_loss_percent=stop_loss_percent,
|
||||||
|
base_quantity=order_quantity,
|
||||||
|
commission_fee=commission_fee,
|
||||||
|
commission_place=commission_place,
|
||||||
|
pnl_series=0
|
||||||
|
)
|
||||||
|
await rq.set_total_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, total_fee=0
|
||||||
|
)
|
||||||
|
await rq.set_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, fee=0
|
||||||
)
|
)
|
||||||
|
|
||||||
res = await open_positions(
|
res = await open_positions(
|
||||||
@@ -79,30 +88,10 @@ async def start_trading_cycle(
|
|||||||
side=side,
|
side=side,
|
||||||
order_quantity=order_quantity,
|
order_quantity=order_quantity,
|
||||||
trigger_price=trigger_price,
|
trigger_price=trigger_price,
|
||||||
margin_type=margin_type,
|
leverage=leverage
|
||||||
leverage=leverage,
|
|
||||||
take_profit_percent=take_profit_percent,
|
|
||||||
stop_loss_percent=stop_loss_percent,
|
|
||||||
commission_fee_percent=total_commission
|
|
||||||
)
|
)
|
||||||
|
|
||||||
if res == "OK":
|
if res == "OK":
|
||||||
await rq.set_user_deal(
|
|
||||||
tg_id=tg_id,
|
|
||||||
symbol=symbol,
|
|
||||||
current_step=1,
|
|
||||||
trade_mode=trade_mode,
|
|
||||||
side_mode=switch_side,
|
|
||||||
margin_type=margin_type,
|
|
||||||
leverage=leverage,
|
|
||||||
order_quantity=order_quantity,
|
|
||||||
trigger_price=trigger_price,
|
|
||||||
martingale_factor=martingale_factor,
|
|
||||||
max_bets_in_series=max_bets_in_series,
|
|
||||||
take_profit_percent=take_profit_percent,
|
|
||||||
stop_loss_percent=stop_loss_percent,
|
|
||||||
base_quantity=order_quantity
|
|
||||||
)
|
|
||||||
return "OK"
|
return "OK"
|
||||||
return (
|
return (
|
||||||
res
|
res
|
||||||
@@ -118,7 +107,9 @@ async def start_trading_cycle(
|
|||||||
"position idx not match position mode",
|
"position idx not match position mode",
|
||||||
"Qty invalid",
|
"Qty invalid",
|
||||||
"The number of contracts exceeds maximum limit allowed",
|
"The number of contracts exceeds maximum limit allowed",
|
||||||
"The number of contracts exceeds minimum limit allowed"
|
"The number of contracts exceeds minimum limit allowed",
|
||||||
|
"Order placement failed as your position may exceed the max",
|
||||||
|
"Permission denied, please check your API key permissions"
|
||||||
}
|
}
|
||||||
else None
|
else None
|
||||||
)
|
)
|
||||||
@@ -132,8 +123,6 @@ async def trading_cycle_profit(
|
|||||||
tg_id: int, symbol: str, side: str) -> str | None:
|
tg_id: int, symbol: str, side: str) -> str | None:
|
||||||
try:
|
try:
|
||||||
user_deals_data = await rq.get_user_deal_by_symbol(tg_id=tg_id, symbol=symbol)
|
user_deals_data = await rq.get_user_deal_by_symbol(tg_id=tg_id, symbol=symbol)
|
||||||
user_auto_trading_data = await rq.get_user_auto_trading(tg_id=tg_id, symbol=symbol)
|
|
||||||
total_fee = user_auto_trading_data.total_fee
|
|
||||||
trade_mode = user_deals_data.trade_mode
|
trade_mode = user_deals_data.trade_mode
|
||||||
margin_type = user_deals_data.margin_type
|
margin_type = user_deals_data.margin_type
|
||||||
leverage = user_deals_data.leverage
|
leverage = user_deals_data.leverage
|
||||||
@@ -144,6 +133,9 @@ async def trading_cycle_profit(
|
|||||||
martingale_factor = user_deals_data.martingale_factor
|
martingale_factor = user_deals_data.martingale_factor
|
||||||
side_mode = user_deals_data.side_mode
|
side_mode = user_deals_data.side_mode
|
||||||
base_quantity = user_deals_data.base_quantity
|
base_quantity = user_deals_data.base_quantity
|
||||||
|
current_series = user_deals_data.current_series
|
||||||
|
commission_fee = user_deals_data.commission_fee
|
||||||
|
commission_place = user_deals_data.commission_place
|
||||||
|
|
||||||
await set_margin_mode(tg_id=tg_id, margin_mode=margin_type)
|
await set_margin_mode(tg_id=tg_id, margin_mode=margin_type)
|
||||||
await set_leverage(
|
await set_leverage(
|
||||||
@@ -152,7 +144,6 @@ async def trading_cycle_profit(
|
|||||||
leverage=leverage,
|
leverage=leverage,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
if trade_mode == "Switch":
|
if trade_mode == "Switch":
|
||||||
if side_mode == "Противоположно":
|
if side_mode == "Противоположно":
|
||||||
s_side = "Sell" if side == "Buy" else "Buy"
|
s_side = "Sell" if side == "Buy" else "Buy"
|
||||||
@@ -161,36 +152,39 @@ async def trading_cycle_profit(
|
|||||||
else:
|
else:
|
||||||
s_side = side
|
s_side = side
|
||||||
|
|
||||||
|
next_series = current_series + 1
|
||||||
|
|
||||||
|
await rq.set_user_deal(
|
||||||
|
tg_id=tg_id,
|
||||||
|
symbol=symbol,
|
||||||
|
current_step=1,
|
||||||
|
current_series=next_series,
|
||||||
|
trade_mode=trade_mode,
|
||||||
|
side_mode=side_mode,
|
||||||
|
margin_type=margin_type,
|
||||||
|
leverage=leverage,
|
||||||
|
order_quantity=base_quantity,
|
||||||
|
trigger_price=trigger_price,
|
||||||
|
martingale_factor=martingale_factor,
|
||||||
|
max_bets_in_series=max_bets_in_series,
|
||||||
|
take_profit_percent=take_profit_percent,
|
||||||
|
stop_loss_percent=stop_loss_percent,
|
||||||
|
base_quantity=base_quantity,
|
||||||
|
commission_fee=commission_fee,
|
||||||
|
commission_place=commission_place,
|
||||||
|
pnl_series=0
|
||||||
|
)
|
||||||
|
|
||||||
res = await open_positions(
|
res = await open_positions(
|
||||||
tg_id=tg_id,
|
tg_id=tg_id,
|
||||||
symbol=symbol,
|
symbol=symbol,
|
||||||
side=s_side,
|
side=s_side,
|
||||||
order_quantity=base_quantity,
|
order_quantity=base_quantity,
|
||||||
trigger_price=trigger_price,
|
trigger_price=trigger_price,
|
||||||
margin_type=margin_type,
|
leverage=leverage
|
||||||
leverage=leverage,
|
|
||||||
take_profit_percent=take_profit_percent,
|
|
||||||
stop_loss_percent=stop_loss_percent,
|
|
||||||
commission_fee_percent=total_fee
|
|
||||||
)
|
)
|
||||||
|
|
||||||
if res == "OK":
|
if res == "OK":
|
||||||
await rq.set_user_deal(
|
|
||||||
tg_id=tg_id,
|
|
||||||
symbol=symbol,
|
|
||||||
current_step=1,
|
|
||||||
trade_mode=trade_mode,
|
|
||||||
side_mode=side_mode,
|
|
||||||
margin_type=margin_type,
|
|
||||||
leverage=leverage,
|
|
||||||
order_quantity=base_quantity,
|
|
||||||
trigger_price=trigger_price,
|
|
||||||
martingale_factor=martingale_factor,
|
|
||||||
max_bets_in_series=max_bets_in_series,
|
|
||||||
take_profit_percent=take_profit_percent,
|
|
||||||
stop_loss_percent=stop_loss_percent,
|
|
||||||
base_quantity=base_quantity
|
|
||||||
)
|
|
||||||
return "OK"
|
return "OK"
|
||||||
|
|
||||||
return (
|
return (
|
||||||
@@ -201,6 +195,7 @@ async def trading_cycle_profit(
|
|||||||
"ab not enough for new order",
|
"ab not enough for new order",
|
||||||
"InvalidRequestError",
|
"InvalidRequestError",
|
||||||
"The number of contracts exceeds maximum limit allowed",
|
"The number of contracts exceeds maximum limit allowed",
|
||||||
|
"Order placement failed as your position may exceed the max",
|
||||||
}
|
}
|
||||||
else None
|
else None
|
||||||
)
|
)
|
||||||
@@ -215,6 +210,8 @@ async def trading_cycle(
|
|||||||
try:
|
try:
|
||||||
user_deals_data = await rq.get_user_deal_by_symbol(tg_id=tg_id, symbol=symbol)
|
user_deals_data = await rq.get_user_deal_by_symbol(tg_id=tg_id, symbol=symbol)
|
||||||
user_auto_trading_data = await rq.get_user_auto_trading(tg_id=tg_id, symbol=symbol)
|
user_auto_trading_data = await rq.get_user_auto_trading(tg_id=tg_id, symbol=symbol)
|
||||||
|
commission_fee = user_deals_data.commission_fee
|
||||||
|
commission_place = user_deals_data.commission_place
|
||||||
total_fee = user_auto_trading_data.total_fee
|
total_fee = user_auto_trading_data.total_fee
|
||||||
trade_mode = user_deals_data.trade_mode
|
trade_mode = user_deals_data.trade_mode
|
||||||
margin_type = user_deals_data.margin_type
|
margin_type = user_deals_data.margin_type
|
||||||
@@ -228,21 +225,24 @@ async def trading_cycle(
|
|||||||
order_quantity = user_deals_data.order_quantity
|
order_quantity = user_deals_data.order_quantity
|
||||||
base_quantity = user_deals_data.base_quantity
|
base_quantity = user_deals_data.base_quantity
|
||||||
side_mode = user_deals_data.side_mode
|
side_mode = user_deals_data.side_mode
|
||||||
|
current_series = user_deals_data.current_series
|
||||||
|
pnl_series = user_deals_data.pnl_series
|
||||||
|
|
||||||
next_quantity = safe_float(order_quantity) * (
|
next_quantity = safe_float(order_quantity) * (
|
||||||
safe_float(martingale_factor)
|
safe_float(martingale_factor)
|
||||||
)
|
)
|
||||||
current_step += 1
|
current_step += 1
|
||||||
|
|
||||||
if max_bets_in_series < current_step:
|
|
||||||
return "Max bets in series"
|
|
||||||
|
|
||||||
await set_margin_mode(tg_id=tg_id, margin_mode=margin_type)
|
await set_margin_mode(tg_id=tg_id, margin_mode=margin_type)
|
||||||
await set_leverage(
|
await set_leverage(
|
||||||
tg_id=tg_id,
|
tg_id=tg_id,
|
||||||
symbol=symbol,
|
symbol=symbol,
|
||||||
leverage=leverage,
|
leverage=leverage,
|
||||||
)
|
)
|
||||||
|
total_quantity = next_quantity
|
||||||
|
if commission_fee == "Yes_commission_fee":
|
||||||
|
if commission_place == "Commission_for_qty":
|
||||||
|
total_quantity = next_quantity + total_fee
|
||||||
|
|
||||||
if trade_mode == "Switch":
|
if trade_mode == "Switch":
|
||||||
if side == "Buy":
|
if side == "Buy":
|
||||||
@@ -252,36 +252,37 @@ async def trading_cycle(
|
|||||||
else:
|
else:
|
||||||
r_side = side
|
r_side = side
|
||||||
|
|
||||||
|
await rq.set_user_deal(
|
||||||
|
tg_id=tg_id,
|
||||||
|
symbol=symbol,
|
||||||
|
current_step=current_step,
|
||||||
|
current_series=current_series,
|
||||||
|
trade_mode=trade_mode,
|
||||||
|
side_mode=side_mode,
|
||||||
|
margin_type=margin_type,
|
||||||
|
leverage=leverage,
|
||||||
|
order_quantity=next_quantity,
|
||||||
|
trigger_price=trigger_price,
|
||||||
|
martingale_factor=martingale_factor,
|
||||||
|
max_bets_in_series=max_bets_in_series,
|
||||||
|
take_profit_percent=take_profit_percent,
|
||||||
|
stop_loss_percent=stop_loss_percent,
|
||||||
|
base_quantity=base_quantity,
|
||||||
|
commission_fee=commission_fee,
|
||||||
|
commission_place=commission_place,
|
||||||
|
pnl_series=pnl_series
|
||||||
|
)
|
||||||
|
|
||||||
res = await open_positions(
|
res = await open_positions(
|
||||||
tg_id=tg_id,
|
tg_id=tg_id,
|
||||||
symbol=symbol,
|
symbol=symbol,
|
||||||
side=r_side,
|
side=r_side,
|
||||||
order_quantity=next_quantity,
|
order_quantity=total_quantity,
|
||||||
trigger_price=trigger_price,
|
trigger_price=trigger_price,
|
||||||
margin_type=margin_type,
|
leverage=leverage
|
||||||
leverage=leverage,
|
|
||||||
take_profit_percent=take_profit_percent,
|
|
||||||
stop_loss_percent=stop_loss_percent,
|
|
||||||
commission_fee_percent=total_fee
|
|
||||||
)
|
)
|
||||||
|
|
||||||
if res == "OK":
|
if res == "OK":
|
||||||
await rq.set_user_deal(
|
|
||||||
tg_id=tg_id,
|
|
||||||
symbol=symbol,
|
|
||||||
current_step=current_step,
|
|
||||||
trade_mode=trade_mode,
|
|
||||||
side_mode=side_mode,
|
|
||||||
margin_type=margin_type,
|
|
||||||
leverage=leverage,
|
|
||||||
order_quantity=next_quantity,
|
|
||||||
trigger_price=trigger_price,
|
|
||||||
martingale_factor=martingale_factor,
|
|
||||||
max_bets_in_series=max_bets_in_series,
|
|
||||||
take_profit_percent=take_profit_percent,
|
|
||||||
stop_loss_percent=stop_loss_percent,
|
|
||||||
base_quantity=base_quantity
|
|
||||||
)
|
|
||||||
return "OK"
|
return "OK"
|
||||||
|
|
||||||
return (
|
return (
|
||||||
@@ -292,6 +293,7 @@ async def trading_cycle(
|
|||||||
"ab not enough for new order",
|
"ab not enough for new order",
|
||||||
"InvalidRequestError",
|
"InvalidRequestError",
|
||||||
"The number of contracts exceeds maximum limit allowed",
|
"The number of contracts exceeds maximum limit allowed",
|
||||||
|
"Order placement failed as your position may exceed the max",
|
||||||
}
|
}
|
||||||
else None
|
else None
|
||||||
)
|
)
|
||||||
@@ -307,23 +309,24 @@ async def open_positions(
|
|||||||
symbol: str,
|
symbol: str,
|
||||||
order_quantity: float,
|
order_quantity: float,
|
||||||
trigger_price: float,
|
trigger_price: float,
|
||||||
margin_type: str,
|
leverage: str
|
||||||
leverage: str,
|
|
||||||
take_profit_percent: float,
|
|
||||||
stop_loss_percent: float,
|
|
||||||
commission_fee_percent: float
|
|
||||||
) -> str | None:
|
) -> str | None:
|
||||||
try:
|
try:
|
||||||
client = await get_bybit_client(tg_id=tg_id)
|
client = await get_bybit_client(tg_id=tg_id)
|
||||||
get_ticker = await get_tickers(tg_id, symbol=symbol)
|
get_ticker = await get_tickers(tg_id, symbol=symbol)
|
||||||
price_symbol = safe_float(get_ticker.get("lastPrice")) or 0
|
|
||||||
|
if get_ticker is None:
|
||||||
|
price_symbol = 0
|
||||||
|
else:
|
||||||
|
price_symbol = safe_float(get_ticker.get("lastPrice"))
|
||||||
|
|
||||||
instruments_info = await get_instruments_info(tg_id=tg_id, symbol=symbol)
|
instruments_info = await get_instruments_info(tg_id=tg_id, symbol=symbol)
|
||||||
qty_step_str = instruments_info.get("lotSizeFilter").get("qtyStep")
|
qty_step_str = instruments_info.get("lotSizeFilter").get("qtyStep")
|
||||||
qty_step = safe_float(qty_step_str)
|
qty_step = safe_float(qty_step_str)
|
||||||
qty = (safe_float(order_quantity) * safe_float(leverage)) / safe_float(price_symbol)
|
qty = (safe_float(order_quantity) * safe_float(leverage)) / safe_float(price_symbol)
|
||||||
decimals = abs(int(round(math.log10(qty_step))))
|
decimals = abs(int(round(math.log10(qty_step))))
|
||||||
qty_formatted = math.floor(qty / qty_step) * qty_step
|
qty_format = math.floor(qty / qty_step) * qty_step
|
||||||
qty_formatted = round(qty_formatted, decimals)
|
qty_formatted = round(qty_format, decimals)
|
||||||
|
|
||||||
if trigger_price > 0:
|
if trigger_price > 0:
|
||||||
po_trigger_price = str(trigger_price)
|
po_trigger_price = str(trigger_price)
|
||||||
@@ -332,33 +335,6 @@ async def open_positions(
|
|||||||
po_trigger_price = None
|
po_trigger_price = None
|
||||||
trigger_direction = None
|
trigger_direction = None
|
||||||
|
|
||||||
price_for_cals = trigger_price if po_trigger_price is not None else price_symbol
|
|
||||||
|
|
||||||
if qty_formatted <= 0:
|
|
||||||
return "Order does not meet minimum order value"
|
|
||||||
|
|
||||||
if margin_type == "ISOLATED_MARGIN":
|
|
||||||
if side == "Buy":
|
|
||||||
take_profit_price = price_for_cals * (
|
|
||||||
1 + take_profit_percent / 100) + commission_fee_percent / qty_formatted
|
|
||||||
stop_loss_price = None
|
|
||||||
else:
|
|
||||||
take_profit_price = price_for_cals * (
|
|
||||||
1 - take_profit_percent / 100) - commission_fee_percent / qty_formatted
|
|
||||||
stop_loss_price = None
|
|
||||||
else:
|
|
||||||
if side == "Buy":
|
|
||||||
take_profit_price = price_for_cals * (
|
|
||||||
1 + take_profit_percent / 100) + commission_fee_percent / qty_formatted
|
|
||||||
stop_loss_price = price_for_cals * (1 - stop_loss_percent / 100)
|
|
||||||
else:
|
|
||||||
take_profit_price = price_for_cals * (
|
|
||||||
1 - take_profit_percent / 100) - commission_fee_percent / qty_formatted
|
|
||||||
stop_loss_price = price_for_cals * (1 + stop_loss_percent / 100)
|
|
||||||
|
|
||||||
take_profit_price = max(take_profit_price, 0)
|
|
||||||
stop_loss_price = max(stop_loss_price, 0)
|
|
||||||
|
|
||||||
# Place order
|
# Place order
|
||||||
order_params = {
|
order_params = {
|
||||||
"category": "linear",
|
"category": "linear",
|
||||||
@@ -371,9 +347,6 @@ async def open_positions(
|
|||||||
"triggerBy": "LastPrice",
|
"triggerBy": "LastPrice",
|
||||||
"timeInForce": "GTC",
|
"timeInForce": "GTC",
|
||||||
"positionIdx": 0,
|
"positionIdx": 0,
|
||||||
"tpslMode": "Full",
|
|
||||||
"takeProfit": str(take_profit_price) if take_profit_price else None,
|
|
||||||
"stopLoss": str(stop_loss_price) if stop_loss_price else None,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
response = client.place_order(**order_params)
|
response = client.place_order(**order_params)
|
||||||
@@ -395,6 +368,8 @@ async def open_positions(
|
|||||||
"Qty invalid": "Qty invalid",
|
"Qty invalid": "Qty invalid",
|
||||||
"The number of contracts exceeds maximum limit allowed": "The number of contracts exceeds maximum limit allowed",
|
"The number of contracts exceeds maximum limit allowed": "The number of contracts exceeds maximum limit allowed",
|
||||||
"The number of contracts exceeds minimum limit allowed": "The number of contracts exceeds minimum limit allowed",
|
"The number of contracts exceeds minimum limit allowed": "The number of contracts exceeds minimum limit allowed",
|
||||||
|
"Order placement failed as your position may exceed the max": "Order placement failed as your position may exceed the max",
|
||||||
|
"Permission denied, please check your API key permissions": "Permission denied, please check your API key permissions"
|
||||||
}
|
}
|
||||||
for key, msg in known_errors.items():
|
for key, msg in known_errors.items():
|
||||||
if key in error_text:
|
if key in error_text:
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ async def user_profile_bybit(tg_id: int, message: Message, state: FSMContext) ->
|
|||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
await message.answer(
|
await message.answer(
|
||||||
text="Ошибка при подключении, повторите попытку",
|
text="Ошибка при подключении к платформе. Проверьте корректность и разрешения API ключа и добавьте повторно.",
|
||||||
reply_markup=kbi.connect_the_platform,
|
reply_markup=kbi.connect_the_platform,
|
||||||
)
|
)
|
||||||
logger.error("Error processing user profile for user %s", tg_id)
|
logger.error("Error processing user profile for user %s", tg_id)
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ async def set_tp_sl_for_position(
|
|||||||
take_profit_price: float,
|
take_profit_price: float,
|
||||||
stop_loss_price: float,
|
stop_loss_price: float,
|
||||||
position_idx: int,
|
position_idx: int,
|
||||||
) -> bool:
|
) -> bool | str:
|
||||||
"""
|
"""
|
||||||
Set take profit and stop loss for a symbol.
|
Set take profit and stop loss for a symbol.
|
||||||
:param tg_id: Telegram user ID
|
:param tg_id: Telegram user ID
|
||||||
@@ -21,15 +21,17 @@ async def set_tp_sl_for_position(
|
|||||||
:param take_profit_price: Take profit price
|
:param take_profit_price: Take profit price
|
||||||
:param stop_loss_price: Stop loss price
|
:param stop_loss_price: Stop loss price
|
||||||
:param position_idx: Position index
|
:param position_idx: Position index
|
||||||
:return: bool
|
:return: bool or str
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
client = await get_bybit_client(tg_id)
|
client = await get_bybit_client(tg_id)
|
||||||
|
take_profit = round(take_profit_price, 6) if take_profit_price is not None else None
|
||||||
|
stop_loss = round(stop_loss_price, 6) if stop_loss_price is not None else None
|
||||||
resp = client.set_trading_stop(
|
resp = client.set_trading_stop(
|
||||||
category="linear",
|
category="linear",
|
||||||
symbol=symbol,
|
symbol=symbol,
|
||||||
takeProfit=str(round(take_profit_price, 5)),
|
takeProfit=str(take_profit) if take_profit is not None else None,
|
||||||
stopLoss=str(round(stop_loss_price, 5)),
|
stopLoss=str(stop_loss) if stop_loss is not None else None,
|
||||||
positionIdx=position_idx,
|
positionIdx=position_idx,
|
||||||
tpslMode="Full",
|
tpslMode="Full",
|
||||||
)
|
)
|
||||||
@@ -38,8 +40,18 @@ async def set_tp_sl_for_position(
|
|||||||
logger.info("TP/SL for %s has been set", symbol)
|
logger.info("TP/SL for %s has been set", symbol)
|
||||||
return True
|
return True
|
||||||
else:
|
else:
|
||||||
logger.error("Error setting TP/SL for %s: %s", symbol, resp.get("retMsg"))
|
error_msg = resp.get("retMsg")
|
||||||
return False
|
if "not modified" in error_msg.lower():
|
||||||
|
logger.info("TP/SL for %s not modified: %s", symbol, error_msg)
|
||||||
|
return "not modified"
|
||||||
|
else:
|
||||||
|
logger.error("Error setting TP/SL for %s: %s", symbol, error_msg)
|
||||||
|
return False
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error("Error setting TP/SL for %s: %s", symbol, e)
|
error_msg = str(e)
|
||||||
return False
|
if "not modified" in error_msg.lower():
|
||||||
|
logger.info("TP/SL for %s not modified: %s", symbol, error_msg)
|
||||||
|
return "not modified"
|
||||||
|
else:
|
||||||
|
logger.error("Error set TP/SL for %s: %s", symbol, e)
|
||||||
|
return False
|
||||||
|
|||||||
@@ -1,10 +1,13 @@
|
|||||||
import logging.config
|
import logging.config
|
||||||
|
import math
|
||||||
|
import json
|
||||||
import app.telegram.keyboards.inline as kbi
|
import app.telegram.keyboards.inline as kbi
|
||||||
import database.request as rq
|
import database.request as rq
|
||||||
|
from app.bybit.get_functions.get_instruments_info import get_instruments_info
|
||||||
from app.bybit.logger_bybit.logger_bybit import LOGGING_CONFIG
|
from app.bybit.logger_bybit.logger_bybit import LOGGING_CONFIG
|
||||||
from app.bybit.open_positions import trading_cycle, trading_cycle_profit
|
from app.bybit.open_positions import trading_cycle, trading_cycle_profit
|
||||||
from app.helper_functions import format_value, safe_float
|
from app.bybit.set_functions.set_tp_sl import set_tp_sl_for_position
|
||||||
|
from app.helper_functions import format_value, safe_float, truncate_float
|
||||||
|
|
||||||
logging.config.dictConfig(LOGGING_CONFIG)
|
logging.config.dictConfig(LOGGING_CONFIG)
|
||||||
logger = logging.getLogger("telegram_message_handler")
|
logger = logging.getLogger("telegram_message_handler")
|
||||||
@@ -12,193 +15,351 @@ logger = logging.getLogger("telegram_message_handler")
|
|||||||
|
|
||||||
class TelegramMessageHandler:
|
class TelegramMessageHandler:
|
||||||
def __init__(self, telegram_bot):
|
def __init__(self, telegram_bot):
|
||||||
|
"""Initialize the TelegramMessageHandler class."""
|
||||||
self.telegram_bot = telegram_bot
|
self.telegram_bot = telegram_bot
|
||||||
|
|
||||||
async def format_position_update(self, message):
|
|
||||||
pass
|
|
||||||
|
|
||||||
async def format_order_update(self, message, tg_id):
|
async def format_order_update(self, message, tg_id):
|
||||||
|
"""Handle order updates."""
|
||||||
try:
|
try:
|
||||||
order_data = message.get("data", [{}])[0]
|
# logger.info("Order update: %s", json.dumps(message))
|
||||||
symbol = format_value(order_data.get("symbol"))
|
user_additional_data = await rq.get_user_additional_settings(tg_id=tg_id)
|
||||||
qty = format_value(order_data.get("qty"))
|
trigger_price = safe_float(user_additional_data.trigger_price)
|
||||||
side = format_value(order_data.get("side"))
|
if trigger_price > 0:
|
||||||
side_rus = (
|
order_data = message.get("data", [{}])[0]
|
||||||
"Покупка"
|
symbol = format_value(order_data.get("symbol"))
|
||||||
if side == "Buy"
|
side = format_value(order_data.get("side"))
|
||||||
else "Продажа" if side == "Sell" else "Нет данных"
|
side_rus = (
|
||||||
)
|
"Покупка"
|
||||||
order_status = format_value(order_data.get("orderStatus"))
|
if side == "Buy"
|
||||||
price = format_value(order_data.get("price"))
|
else "Продажа" if side == "Sell" else "Нет данных"
|
||||||
trigger_price = format_value(order_data.get("triggerPrice"))
|
)
|
||||||
take_profit = format_value(order_data.get("takeProfit"))
|
order_status = format_value(order_data.get("orderStatus"))
|
||||||
stop_loss = format_value(order_data.get("stopLoss"))
|
tr_price = format_value(order_data.get("triggerPrice"))
|
||||||
|
|
||||||
status_map = {
|
status_map = {
|
||||||
"Untriggered": "Условный ордер выставлен",
|
"Untriggered": "Условный ордер выставлен",
|
||||||
}
|
}
|
||||||
|
|
||||||
if order_status == "Filled" or order_status not in status_map:
|
if order_status == "Filled" or order_status not in status_map:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
user_auto_trading = await rq.get_user_auto_trading(
|
text = (
|
||||||
tg_id=tg_id, symbol=symbol
|
f"Торговая пара: {symbol}\n"
|
||||||
)
|
f"Движение: {side_rus}\n"
|
||||||
auto_trading = (
|
)
|
||||||
user_auto_trading.auto_trading if user_auto_trading else False
|
if tr_price and tr_price != "Нет данных":
|
||||||
)
|
text += f"Триггер цена: {tr_price}\n"
|
||||||
user_deals_data = await rq.get_user_deal_by_symbol(
|
|
||||||
tg_id=tg_id, symbol=symbol
|
|
||||||
)
|
|
||||||
|
|
||||||
text = (
|
await self.telegram_bot.send_message(
|
||||||
f"Торговая пара: {symbol}\n"
|
chat_id=tg_id, text=text, reply_markup=kbi.profile_bybit
|
||||||
f"Движение: {side_rus}\n"
|
)
|
||||||
)
|
await rq.set_trigger_price(tg_id=tg_id, trigger_price=0)
|
||||||
|
|
||||||
if user_deals_data is not None and auto_trading:
|
|
||||||
text += f"Текущая ставка: {user_deals_data.order_quantity} USDT\n"
|
|
||||||
else:
|
|
||||||
text += f"Количество: {qty}\n"
|
|
||||||
|
|
||||||
if price and price != "0":
|
|
||||||
text += f"Цена: {price}\n"
|
|
||||||
if take_profit and take_profit != "Нет данных":
|
|
||||||
text += f"Тейк-профит: {take_profit}\n"
|
|
||||||
if stop_loss and stop_loss != "Нет данных":
|
|
||||||
text += f"Стоп-лосс: {stop_loss}\n"
|
|
||||||
if trigger_price and trigger_price != "Нет данных":
|
|
||||||
text += f"Триггер цена: {trigger_price}\n"
|
|
||||||
|
|
||||||
await self.telegram_bot.send_message(
|
|
||||||
chat_id=tg_id, text=text, reply_markup=kbi.profile_bybit
|
|
||||||
)
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error("Error in format_order_update: %s", e)
|
logger.error("Error in format_order_update: %s", e)
|
||||||
|
|
||||||
async def format_execution_update(self, message, tg_id):
|
async def format_execution_update(self, message, tg_id):
|
||||||
|
"""Handle execution updates without duplicate processing."""
|
||||||
try:
|
try:
|
||||||
|
# logger.info("Execution update: %s", json.dumps(message))
|
||||||
execution = message.get("data", [{}])[0]
|
execution = message.get("data", [{}])[0]
|
||||||
closed_size = format_value(execution.get("closedSize"))
|
exec_type = format_value(execution.get("execType"))
|
||||||
symbol = format_value(execution.get("symbol"))
|
if exec_type == "Trade" or exec_type == "BustTrade":
|
||||||
exec_price = format_value(execution.get("execPrice"))
|
closed_size = format_value(execution.get("closedSize"))
|
||||||
exec_qty = format_value(execution.get("execQty"))
|
symbol = format_value(execution.get("symbol"))
|
||||||
exec_fees = format_value(execution.get("execFee"))
|
exec_price = format_value(execution.get("execPrice"))
|
||||||
fee_rate = format_value(execution.get("feeRate"))
|
exec_qty = format_value(execution.get("execQty"))
|
||||||
side = format_value(execution.get("side"))
|
exec_fees = format_value(execution.get("execFee"))
|
||||||
side_rus = (
|
fee_rate = format_value(execution.get("feeRate"))
|
||||||
"Покупка"
|
side = format_value(execution.get("side"))
|
||||||
if side == "Buy"
|
exec_pnl = format_value(execution.get("execPnl"))
|
||||||
else "Продажа" if side == "Sell" else "Нет данных"
|
stop_order_type = format_value(execution.get("stopOrderType"))
|
||||||
)
|
create_type = format_value(execution.get("createType"))
|
||||||
if safe_float(exec_fees) == 0:
|
|
||||||
exec_fee = safe_float(exec_price) * safe_float(exec_qty) * safe_float(
|
|
||||||
fee_rate
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
exec_fee = safe_float(exec_fees)
|
|
||||||
|
|
||||||
if safe_float(closed_size) == 0:
|
user_auto_trading = await rq.get_user_auto_trading(
|
||||||
await rq.set_fee_user_auto_trading(
|
tg_id=tg_id, symbol=symbol
|
||||||
tg_id=tg_id, symbol=symbol, fee=safe_float(exec_fee)
|
)
|
||||||
|
auto_trading = (
|
||||||
|
user_auto_trading.auto_trading if user_auto_trading else False
|
||||||
)
|
)
|
||||||
|
|
||||||
user_auto_trading = await rq.get_user_auto_trading(
|
if auto_trading:
|
||||||
tg_id=tg_id, symbol=symbol
|
side_rus = (
|
||||||
)
|
"Покупка"
|
||||||
|
if side == "Buy"
|
||||||
|
else "Продажа" if side == "Sell" else "Нет данных"
|
||||||
|
)
|
||||||
|
if safe_float(exec_fees) == 0:
|
||||||
|
exec_fee = safe_float(exec_price) * safe_float(exec_qty) * safe_float(
|
||||||
|
fee_rate
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
exec_fee = safe_float(exec_fees)
|
||||||
|
|
||||||
get_total_fee = user_auto_trading.total_fee
|
if safe_float(closed_size) == 0:
|
||||||
total_fee = safe_float(exec_fee) + safe_float(get_total_fee)
|
await rq.set_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, fee=safe_float(exec_fee)
|
||||||
|
)
|
||||||
|
|
||||||
|
get_total_fee = 0
|
||||||
|
|
||||||
if user_auto_trading is not None and user_auto_trading.fee is not None:
|
if user_auto_trading is not None:
|
||||||
fee = user_auto_trading.fee
|
get_total_fee = user_auto_trading.total_fee
|
||||||
else:
|
|
||||||
fee = 0
|
|
||||||
|
|
||||||
exec_pnl = format_value(execution.get("execPnl"))
|
total_fee = safe_float(exec_fee) + safe_float(get_total_fee)
|
||||||
risk_management_data = await rq.get_user_risk_management(tg_id=tg_id)
|
ex_pnl = safe_float(exec_pnl)
|
||||||
commission_fee = risk_management_data.commission_fee
|
|
||||||
|
|
||||||
if commission_fee == "Yes_commission_fee":
|
header = (
|
||||||
total_pnl = safe_float(exec_pnl) - safe_float(exec_fee) - fee
|
"Сделка закрыта:" if safe_float(closed_size) > 0 else "Сделка открыта:"
|
||||||
else:
|
)
|
||||||
total_pnl = safe_float(exec_pnl)
|
text = f"{header}\n" f"Торговая пара: {symbol}\n"
|
||||||
|
user_deals_data = await rq.get_user_deal_by_symbol(
|
||||||
|
tg_id=tg_id, symbol=symbol
|
||||||
|
)
|
||||||
|
|
||||||
header = (
|
commission_fee = user_deals_data.commission_fee
|
||||||
"Сделка закрыта:" if safe_float(closed_size) > 0 else "Сделка открыта:"
|
commission_place = user_deals_data.commission_place
|
||||||
)
|
current_series = user_deals_data.current_series
|
||||||
text = f"{header}\n" f"Торговая пара: {symbol}\n"
|
current_step = user_deals_data.current_step
|
||||||
|
order_quantity = user_deals_data.order_quantity
|
||||||
|
pnl_series = user_deals_data.pnl_series
|
||||||
|
take_profit_percent = user_deals_data.take_profit_percent
|
||||||
|
stop_loss_percent = user_deals_data.stop_loss_percent
|
||||||
|
leverage = safe_float(user_deals_data.leverage)
|
||||||
|
fee = safe_float(user_auto_trading.fee)
|
||||||
|
total_pnl = safe_float(exec_pnl) - safe_float(exec_fee) - fee
|
||||||
|
|
||||||
auto_trading = (
|
if commission_fee == "Yes_commission_fee":
|
||||||
user_auto_trading.auto_trading if user_auto_trading else False
|
if commission_place == "Commission_for_qty":
|
||||||
)
|
total_quantity = safe_float(order_quantity) + safe_float(
|
||||||
user_deals_data = await rq.get_user_deal_by_symbol(
|
total_fee
|
||||||
tg_id=tg_id, symbol=symbol
|
) * 2
|
||||||
)
|
else:
|
||||||
if user_deals_data is not None and auto_trading:
|
total_quantity = safe_float(order_quantity)
|
||||||
await rq.set_total_fee_user_auto_trading(
|
else:
|
||||||
tg_id=tg_id, symbol=symbol, total_fee=total_fee
|
total_quantity = safe_float(order_quantity)
|
||||||
)
|
|
||||||
text += f"Текущая ставка: {user_deals_data.order_quantity} USDT\n"
|
|
||||||
|
|
||||||
text += (
|
if user_deals_data is not None and safe_float(closed_size) == 0:
|
||||||
f"Цена исполнения: {exec_price}\n"
|
await rq.set_total_fee_user_auto_trading(
|
||||||
f"Комиссия: {exec_fee:.8f}\n"
|
tg_id=tg_id, symbol=symbol, total_fee=total_fee
|
||||||
)
|
)
|
||||||
|
text += f"Текущая ставка: {total_quantity:.2f} USDT\n"
|
||||||
|
text += f"Серия №: {current_series}\n"
|
||||||
|
text += f"Сделка №: {current_step}\n"
|
||||||
|
|
||||||
if safe_float(closed_size) == 0:
|
text += (
|
||||||
text += f"Движение: {side_rus}\n"
|
f"Цена исполнения: {exec_price}\n"
|
||||||
else:
|
f"Комиссия: {exec_fee:.8f}\n"
|
||||||
text += f"\nРеализованная прибыль: {total_pnl:.7f}\n"
|
)
|
||||||
|
|
||||||
await self.telegram_bot.send_message(
|
if safe_float(closed_size) == 0:
|
||||||
chat_id=tg_id, text=text, reply_markup=kbi.profile_bybit
|
instruments_info = await get_instruments_info(tg_id=tg_id, symbol=symbol)
|
||||||
)
|
qty_step_str = instruments_info.get("lotSizeFilter").get("qtyStep")
|
||||||
|
qty_step = safe_float(qty_step_str)
|
||||||
|
qty = (safe_float(order_quantity) * safe_float(leverage)) / safe_float(exec_price)
|
||||||
|
decimals = abs(int(round(math.log10(qty_step))))
|
||||||
|
qty_format = math.floor(qty / qty_step) * qty_step
|
||||||
|
qty_formatted = round(qty_format, decimals)
|
||||||
|
total_commission = 0
|
||||||
|
|
||||||
user_symbols = user_auto_trading.symbol if user_auto_trading else None
|
if commission_fee == "Yes_commission_fee":
|
||||||
|
if commission_place == "Commission_for_tp":
|
||||||
|
total_commission = safe_float(total_fee) / qty_formatted
|
||||||
|
|
||||||
|
if side == "Buy":
|
||||||
|
take_profit_price = safe_float(exec_price) * (
|
||||||
|
1 + take_profit_percent / 100) + total_commission
|
||||||
|
stop_loss_price = safe_float(exec_price) * (1 - stop_loss_percent / 100)
|
||||||
|
else:
|
||||||
|
take_profit_price = safe_float(exec_price) * (
|
||||||
|
1 - take_profit_percent / 100) - total_commission
|
||||||
|
stop_loss_price = safe_float(exec_price) * (1 + stop_loss_percent / 100)
|
||||||
|
|
||||||
|
ress = await set_tp_sl_for_position(tg_id=tg_id,
|
||||||
|
symbol=symbol,
|
||||||
|
take_profit_price=take_profit_price,
|
||||||
|
stop_loss_price=stop_loss_price,
|
||||||
|
position_idx=0)
|
||||||
|
if ress or ress == "not modified":
|
||||||
|
take_profit_truncated = await truncate_float(take_profit_price, 6)
|
||||||
|
stop_loss_truncated = await truncate_float(stop_loss_price, 6)
|
||||||
|
text += (f"Движение: {side_rus}\n"
|
||||||
|
f"Тейк-профит: {take_profit_truncated}\n"
|
||||||
|
f"Стоп-лосс: {stop_loss_truncated}\n"
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
text += (f"Движение: {side_rus}\n"
|
||||||
|
"Не удалось установить ТП и СЛ\n")
|
||||||
|
|
||||||
|
elif safe_float(closed_size) > 0 and auto_trading:
|
||||||
|
new_pnl = safe_float(pnl_series) + total_pnl
|
||||||
|
await rq.set_pnl_series_by_symbol(
|
||||||
|
tg_id=tg_id, symbol=symbol, pnl_series=new_pnl)
|
||||||
|
text += f"\nПрибыль без комиссии: {ex_pnl:.4f}\n"
|
||||||
|
text += f"Реализованная прибыль: {total_pnl:.4f}\n"
|
||||||
|
text += f"Прибыль серии: {safe_float(new_pnl):.4f}\n"
|
||||||
|
|
||||||
if (
|
|
||||||
auto_trading
|
|
||||||
and safe_float(closed_size) > 0
|
|
||||||
and user_symbols is not None
|
|
||||||
):
|
|
||||||
if safe_float(total_pnl) > 0:
|
|
||||||
profit_text = "📈 Прибыль достигнута. Начинаем новую серию с базовой ставки\n"
|
|
||||||
await self.telegram_bot.send_message(
|
await self.telegram_bot.send_message(
|
||||||
chat_id=tg_id, text=profit_text, reply_markup=kbi.profile_bybit
|
chat_id=tg_id, text=text, reply_markup=kbi.profile_bybit
|
||||||
)
|
)
|
||||||
|
|
||||||
if side == "Buy":
|
if stop_order_type == "TakeProfit":
|
||||||
r_side = "Sell"
|
profit_text = "📈 Начинаю новую серию с базовой ставки\n"
|
||||||
else:
|
await self.telegram_bot.send_message(
|
||||||
r_side = "Buy"
|
chat_id=tg_id, text=profit_text, reply_markup=kbi.profile_bybit
|
||||||
|
)
|
||||||
|
|
||||||
await rq.set_last_side_by_symbol(
|
if side == "Buy":
|
||||||
tg_id=tg_id, symbol=symbol, last_side=r_side)
|
r_side = "Sell"
|
||||||
await rq.set_total_fee_user_auto_trading(
|
else:
|
||||||
tg_id=tg_id, symbol=symbol, total_fee=0
|
r_side = "Buy"
|
||||||
)
|
|
||||||
await rq.set_fee_user_auto_trading(
|
|
||||||
tg_id=tg_id, symbol=symbol, fee=0
|
|
||||||
)
|
|
||||||
|
|
||||||
res = await trading_cycle_profit(
|
await rq.set_last_side_by_symbol(
|
||||||
tg_id=tg_id, symbol=symbol, side=r_side
|
tg_id=tg_id, symbol=symbol, last_side=r_side)
|
||||||
)
|
await rq.set_total_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, total_fee=0
|
||||||
|
)
|
||||||
|
await rq.set_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, fee=0
|
||||||
|
)
|
||||||
|
await rq.set_pnl_series_by_symbol(tg_id=tg_id, symbol=symbol, pnl_series=0)
|
||||||
|
|
||||||
if res == "OK":
|
res = await trading_cycle_profit(
|
||||||
pass
|
tg_id=tg_id, symbol=symbol, side=r_side
|
||||||
else:
|
)
|
||||||
errors = {
|
if res == "OK":
|
||||||
"Max bets in series": "❗️ Максимальное количество сделок в серии достигнуто",
|
pass
|
||||||
"Risk is too high for this trade": "❗️ Риск сделки слишком высок для продолжения",
|
else:
|
||||||
"ab not enough for new order": "❗️ Недостаточно средств для продолжения торговли",
|
errors = {
|
||||||
"InvalidRequestError": "❗️ Недостаточно средств для размещения нового ордера с заданным количеством и плечом.",
|
"Risk is too high for this trade": "❗️ Риск сделки слишком высок для продолжения",
|
||||||
"The number of contracts exceeds maximum limit allowed": "❗️ Превышен максимальный лимит ставки",
|
"ab not enough for new order": "❗️ Недостаточно средств для продолжения торговли",
|
||||||
}
|
"InvalidRequestError": "❗️ Недостаточно средств для размещения нового ордера с заданным количеством и плечом.",
|
||||||
error_text = errors.get(
|
"The number of contracts exceeds maximum limit allowed": "❗️ Превышен максимальный лимит ставки",
|
||||||
res, "❗️ Не удалось открыть новую сделку"
|
"Order placement failed as your position may exceed the max": "❗️ Превышен максимальный лимит ставки",
|
||||||
|
}
|
||||||
|
error_text = errors.get(
|
||||||
|
res, "❗️ Не удалось открыть новую сделку"
|
||||||
|
)
|
||||||
|
await rq.set_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, auto_trading=False
|
||||||
|
)
|
||||||
|
|
||||||
|
await rq.set_total_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, total_fee=0
|
||||||
|
)
|
||||||
|
await rq.set_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, fee=0
|
||||||
|
)
|
||||||
|
await self.telegram_bot.send_message(
|
||||||
|
chat_id=tg_id,
|
||||||
|
text=error_text,
|
||||||
|
reply_markup=kbi.profile_bybit,
|
||||||
|
)
|
||||||
|
|
||||||
|
elif stop_order_type == "StopLoss" or exec_type == "BustTrade":
|
||||||
|
current_step = user_deals_data.current_step
|
||||||
|
max_bets_in_series = user_deals_data.max_bets_in_series
|
||||||
|
current_step += 1
|
||||||
|
|
||||||
|
if max_bets_in_series < current_step:
|
||||||
|
text_series = ("\n❗️ Максимальное количество сделок в серии достигнуто.\n"
|
||||||
|
"📈 Начинаю новую серию с базовой ставки\n")
|
||||||
|
await self.telegram_bot.send_message(
|
||||||
|
chat_id=tg_id, text=text_series
|
||||||
|
)
|
||||||
|
if side == "Buy":
|
||||||
|
r_side = "Sell"
|
||||||
|
else:
|
||||||
|
r_side = "Buy"
|
||||||
|
|
||||||
|
await rq.set_last_side_by_symbol(
|
||||||
|
tg_id=tg_id, symbol=symbol, last_side=r_side)
|
||||||
|
await rq.set_total_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, total_fee=0
|
||||||
|
)
|
||||||
|
await rq.set_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, fee=0
|
||||||
|
)
|
||||||
|
await rq.set_pnl_series_by_symbol(tg_id=tg_id, symbol=symbol, pnl_series=0)
|
||||||
|
|
||||||
|
res = await trading_cycle_profit(
|
||||||
|
tg_id=tg_id, symbol=symbol, side=r_side
|
||||||
|
)
|
||||||
|
if res == "OK":
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
errors = {
|
||||||
|
"Risk is too high for this trade": "❗️ Риск сделки слишком высок для продолжения",
|
||||||
|
"ab not enough for new order": "❗️ Недостаточно средств для продолжения торговли",
|
||||||
|
"InvalidRequestError": "❗️ Недостаточно средств для размещения нового ордера с заданным количеством и плечом.",
|
||||||
|
"The number of contracts exceeds maximum limit allowed": "❗️ Превышен максимальный лимит ставки",
|
||||||
|
"Order placement failed as your position may exceed the max": "❗️ Превышен максимальный лимит ставки",
|
||||||
|
}
|
||||||
|
error_text = errors.get(
|
||||||
|
res, "❗️ Не удалось открыть новую сделку"
|
||||||
|
)
|
||||||
|
await rq.set_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, auto_trading=False
|
||||||
|
)
|
||||||
|
|
||||||
|
await rq.set_total_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, total_fee=0
|
||||||
|
)
|
||||||
|
await rq.set_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, fee=0
|
||||||
|
)
|
||||||
|
await self.telegram_bot.send_message(
|
||||||
|
chat_id=tg_id,
|
||||||
|
text=error_text,
|
||||||
|
reply_markup=kbi.profile_bybit,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
open_order_text = "\n❗️ Открываю новую сделку с увеличенной ставкой.\n"
|
||||||
|
await self.telegram_bot.send_message(
|
||||||
|
chat_id=tg_id, text=open_order_text
|
||||||
|
)
|
||||||
|
|
||||||
|
if side == "Buy":
|
||||||
|
r_side = "Sell"
|
||||||
|
else:
|
||||||
|
r_side = "Buy"
|
||||||
|
|
||||||
|
res = await trading_cycle(
|
||||||
|
tg_id=tg_id, symbol=symbol, side=r_side
|
||||||
|
)
|
||||||
|
|
||||||
|
if res == "OK":
|
||||||
|
pass
|
||||||
|
else:
|
||||||
|
errors = {
|
||||||
|
"Risk is too high for this trade": "❗️ Риск сделки слишком высок для продолжения",
|
||||||
|
"ab not enough for new order": "❗️ Недостаточно средств для продолжения торговли",
|
||||||
|
"InvalidRequestError": "❗️ Недостаточно средств для размещения нового ордера с заданным количеством и плечом.",
|
||||||
|
"The number of contracts exceeds maximum limit allowed": "❗️ Превышен максимальный лимит ставки",
|
||||||
|
"Order placement failed as your position may exceed the max": "❗️ Превышен максимальный лимит ставки с текущим плечом",
|
||||||
|
}
|
||||||
|
error_text = errors.get(
|
||||||
|
res, "❗️ Не удалось открыть новую сделку"
|
||||||
|
)
|
||||||
|
await rq.set_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, auto_trading=False
|
||||||
|
)
|
||||||
|
|
||||||
|
await rq.set_total_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, total_fee=0
|
||||||
|
)
|
||||||
|
await rq.set_fee_user_auto_trading(
|
||||||
|
tg_id=tg_id, symbol=symbol, fee=0
|
||||||
|
)
|
||||||
|
await self.telegram_bot.send_message(
|
||||||
|
chat_id=tg_id,
|
||||||
|
text=error_text,
|
||||||
|
reply_markup=kbi.profile_bybit,
|
||||||
|
)
|
||||||
|
elif create_type == "CreateByClosing":
|
||||||
|
await self.telegram_bot.send_message(
|
||||||
|
chat_id=tg_id,
|
||||||
|
text=f"❗️ Торговля для {symbol} остановлена",
|
||||||
|
reply_markup=kbi.profile_bybit,
|
||||||
)
|
)
|
||||||
await rq.set_auto_trading(
|
await rq.set_auto_trading(
|
||||||
tg_id=tg_id, symbol=symbol, auto_trading=False
|
tg_id=tg_id, symbol=symbol, auto_trading=False
|
||||||
@@ -210,54 +371,9 @@ class TelegramMessageHandler:
|
|||||||
await rq.set_fee_user_auto_trading(
|
await rq.set_fee_user_auto_trading(
|
||||||
tg_id=tg_id, symbol=symbol, fee=0
|
tg_id=tg_id, symbol=symbol, fee=0
|
||||||
)
|
)
|
||||||
await self.telegram_bot.send_message(
|
logger.info("Stop trading for symbol: %s, create_type: %s, stop_order_type: %s: %s",
|
||||||
chat_id=tg_id,
|
symbol, create_type, stop_order_type, tg_id)
|
||||||
text=error_text,
|
else:
|
||||||
reply_markup=kbi.profile_bybit,
|
logger.info("Execution update: %s", json.dumps(message))
|
||||||
)
|
|
||||||
else:
|
|
||||||
open_order_text = "\n❗️ Сделка закрылась в минус, открываю новую сделку с увеличенной ставкой.\n"
|
|
||||||
await self.telegram_bot.send_message(
|
|
||||||
chat_id=tg_id, text=open_order_text
|
|
||||||
)
|
|
||||||
|
|
||||||
if side == "Buy":
|
|
||||||
r_side = "Sell"
|
|
||||||
else:
|
|
||||||
r_side = "Buy"
|
|
||||||
|
|
||||||
res = await trading_cycle(
|
|
||||||
tg_id=tg_id, symbol=symbol, side=r_side
|
|
||||||
)
|
|
||||||
|
|
||||||
if res == "OK":
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
errors = {
|
|
||||||
"Max bets in series": "❗️ Максимальное количество сделок в серии достигнуто",
|
|
||||||
"Risk is too high for this trade": "❗️ Риск сделки слишком высок для продолжения",
|
|
||||||
"ab not enough for new order": "❗️ Недостаточно средств для продолжения торговли",
|
|
||||||
"InvalidRequestError": "❗️ Недостаточно средств для размещения нового ордера с заданным количеством и плечом.",
|
|
||||||
"The number of contracts exceeds maximum limit allowed": "❗️ Превышен максимальный лимит ставки",
|
|
||||||
}
|
|
||||||
error_text = errors.get(
|
|
||||||
res, "❗️ Не удалось открыть новую сделку"
|
|
||||||
)
|
|
||||||
await rq.set_auto_trading(
|
|
||||||
tg_id=tg_id, symbol=symbol, auto_trading=False
|
|
||||||
)
|
|
||||||
|
|
||||||
await rq.set_total_fee_user_auto_trading(
|
|
||||||
tg_id=tg_id, symbol=symbol, total_fee=0
|
|
||||||
)
|
|
||||||
await rq.set_fee_user_auto_trading(
|
|
||||||
tg_id=tg_id, symbol=symbol, fee=0
|
|
||||||
)
|
|
||||||
await self.telegram_bot.send_message(
|
|
||||||
chat_id=tg_id,
|
|
||||||
text=error_text,
|
|
||||||
reply_markup=kbi.profile_bybit,
|
|
||||||
)
|
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error("Error in telegram_message_handler: %s", e)
|
logger.error("Error in telegram_message_handler: %s", e, exc_info=True)
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
import asyncio
|
import asyncio
|
||||||
|
from collections import deque
|
||||||
import logging.config
|
import logging.config
|
||||||
|
|
||||||
from pybit.unified_trading import WebSocket
|
from pybit.unified_trading import WebSocket
|
||||||
@@ -11,112 +12,178 @@ logging.config.dictConfig(LOGGING_CONFIG)
|
|||||||
logger = logging.getLogger("web_socket")
|
logger = logging.getLogger("web_socket")
|
||||||
|
|
||||||
|
|
||||||
|
class CustomWebSocket(WebSocket):
|
||||||
|
"""Custom WebSocket wrapper with enhanced error handling."""
|
||||||
|
|
||||||
|
def _on_error(self, error):
|
||||||
|
logger.error(f"WebSocket error: {error}")
|
||||||
|
return super()._on_error(error)
|
||||||
|
|
||||||
|
def _on_close(self):
|
||||||
|
logger.warning("WebSocket connection closed")
|
||||||
|
super()._on_close()
|
||||||
|
|
||||||
|
|
||||||
class WebSocketBot:
|
class WebSocketBot:
|
||||||
"""
|
"""
|
||||||
Class to handle WebSocket connections and messages.
|
Manages multiple Bybit private WebSocket connections for Telegram users.
|
||||||
|
Uses queue-based message processing to handle thread-safe async calls.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, telegram_bot):
|
def __init__(self, telegram_bot):
|
||||||
"""Initialize the TradingBot class."""
|
"""
|
||||||
|
Initialize WebSocketBot.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
telegram_bot: Telegram bot instance for message handling
|
||||||
|
"""
|
||||||
self.telegram_bot = telegram_bot
|
self.telegram_bot = telegram_bot
|
||||||
self.ws_private = None
|
|
||||||
self.user_messages = {}
|
|
||||||
self.user_sockets = {}
|
self.user_sockets = {}
|
||||||
|
self.user_messages = {}
|
||||||
self.user_keys = {}
|
self.user_keys = {}
|
||||||
self.loop = None
|
self.loop = None
|
||||||
self.message_handler = TelegramMessageHandler(telegram_bot)
|
self.message_handler = TelegramMessageHandler(telegram_bot)
|
||||||
|
|
||||||
|
self.order_queues = {} # {tg_id: deque}
|
||||||
|
self.execution_queues = {} # {tg_id: deque}
|
||||||
|
self.processing_tasks = {} # {tg_id: task}
|
||||||
|
|
||||||
async def run_user_check_loop(self):
|
async def run_user_check_loop(self):
|
||||||
"""Run a loop to check for users and connect them to the WebSocket."""
|
"""Main loop that continuously checks users and maintains connections."""
|
||||||
self.loop = asyncio.get_running_loop()
|
self.loop = asyncio.get_running_loop()
|
||||||
|
logger.info("Starting WebSocket user check loop")
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
users = await WebSocketBot.get_users_from_db()
|
try:
|
||||||
for user in users:
|
users = await WebSocketBot.get_users_from_db()
|
||||||
tg_id = user.tg_id
|
for user in users:
|
||||||
api_key, api_secret = await rq.get_user_api(tg_id=tg_id)
|
tg_id = user.tg_id
|
||||||
|
api_key, api_secret = await rq.get_user_api(tg_id=tg_id)
|
||||||
|
|
||||||
if not api_key or not api_secret:
|
if not api_key or not api_secret:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
keys_stored = self.user_keys.get(tg_id)
|
keys_stored = self.user_keys.get(tg_id)
|
||||||
if tg_id in self.user_sockets and keys_stored == (api_key, api_secret):
|
socket_exists = tg_id in self.user_sockets
|
||||||
continue
|
|
||||||
|
|
||||||
if tg_id in self.user_sockets:
|
if socket_exists and keys_stored == (api_key, api_secret):
|
||||||
self.user_sockets.clear()
|
continue
|
||||||
self.user_messages.clear()
|
|
||||||
self.user_keys.clear()
|
|
||||||
logger.info(
|
|
||||||
"Closed old websocket for user %s due to key change", tg_id
|
|
||||||
)
|
|
||||||
|
|
||||||
success = await self.try_connect_user(api_key, api_secret, tg_id)
|
if socket_exists:
|
||||||
if success:
|
await self.close_user_socket(tg_id)
|
||||||
self.user_keys[tg_id] = (api_key, api_secret)
|
|
||||||
self.user_messages.setdefault(
|
success = await self.try_connect_user(api_key, api_secret, tg_id)
|
||||||
tg_id, {"position": None, "order": None, "execution": None}
|
if success:
|
||||||
)
|
self.user_keys[tg_id] = (api_key, api_secret)
|
||||||
logger.info("User %s connected to WebSocket", tg_id)
|
self.user_messages.setdefault(
|
||||||
else:
|
tg_id, {"position": None, "order": None, "execution": None}
|
||||||
await asyncio.sleep(30)
|
)
|
||||||
|
logger.info("User %s successfully connected", tg_id)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error in user check loop: %s", e)
|
||||||
|
|
||||||
await asyncio.sleep(10)
|
await asyncio.sleep(10)
|
||||||
|
|
||||||
async def clear_user_sockets(self):
|
|
||||||
"""Clear the user_sockets and user_messages dictionaries."""
|
|
||||||
self.user_sockets.clear()
|
|
||||||
self.user_messages.clear()
|
|
||||||
self.user_keys.clear()
|
|
||||||
logger.info("Cleared user_sockets")
|
|
||||||
|
|
||||||
async def try_connect_user(self, api_key, api_secret, tg_id):
|
async def try_connect_user(self, api_key, api_secret, tg_id):
|
||||||
"""Try to connect a user to the WebSocket."""
|
"""
|
||||||
|
Create and setup WebSocket streams with thread-safe queues.
|
||||||
|
"""
|
||||||
try:
|
try:
|
||||||
self.ws_private = WebSocket(
|
ws = CustomWebSocket(
|
||||||
|
demo=True,
|
||||||
testnet=False,
|
testnet=False,
|
||||||
channel_type="private",
|
channel_type="private",
|
||||||
api_key=api_key,
|
api_key=api_key,
|
||||||
api_secret=api_secret,
|
api_secret=api_secret
|
||||||
)
|
)
|
||||||
|
|
||||||
self.user_sockets[tg_id] = self.ws_private
|
self.user_sockets[tg_id] = ws
|
||||||
# Connect to the WebSocket private channel
|
|
||||||
# Handle position updates
|
self.order_queues[tg_id] = deque()
|
||||||
self.ws_private.position_stream(
|
self.execution_queues[tg_id] = deque()
|
||||||
lambda msg: self.loop.call_soon_threadsafe(
|
|
||||||
asyncio.create_task, self.handle_position_update(msg)
|
self.processing_tasks[tg_id] = asyncio.create_task(
|
||||||
)
|
self._process_order_queue(tg_id)
|
||||||
)
|
)
|
||||||
# Handle order updates
|
self.processing_tasks[tg_id + 1] = asyncio.create_task(
|
||||||
self.ws_private.order_stream(
|
self._process_execution_queue(tg_id)
|
||||||
lambda msg: self.loop.call_soon_threadsafe(
|
|
||||||
asyncio.create_task, self.handle_order_update(msg, tg_id)
|
|
||||||
)
|
|
||||||
)
|
|
||||||
# Handle execution updates
|
|
||||||
self.ws_private.execution_stream(
|
|
||||||
lambda msg: self.loop.call_soon_threadsafe(
|
|
||||||
asyncio.create_task, self.handle_execution_update(msg, tg_id)
|
|
||||||
)
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def order_callback(msg):
|
||||||
|
self.order_queues[tg_id].append(msg)
|
||||||
|
|
||||||
|
def execution_callback(msg):
|
||||||
|
self.execution_queues[tg_id].append(msg)
|
||||||
|
|
||||||
|
ws.order_stream(order_callback)
|
||||||
|
ws.execution_stream(execution_callback)
|
||||||
|
|
||||||
|
logger.info("WebSocket streams configured for user %s", tg_id)
|
||||||
return True
|
return True
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error("Error connecting user %s: %s", tg_id, e)
|
logger.error("Error connecting user %s: %s", tg_id, e)
|
||||||
|
self.user_sockets.pop(tg_id, None)
|
||||||
return False
|
return False
|
||||||
|
|
||||||
async def handle_position_update(self, message):
|
async def _process_order_queue(self, tg_id):
|
||||||
"""Handle position updates."""
|
"""Continuously process order queue for user."""
|
||||||
await self.message_handler.format_position_update(message)
|
while tg_id in self.user_sockets:
|
||||||
|
try:
|
||||||
|
if self.order_queues[tg_id]:
|
||||||
|
msg = self.order_queues[tg_id].popleft()
|
||||||
|
await self.handle_order_update(msg, tg_id)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error processing order queue %s: %s", tg_id, e)
|
||||||
|
await asyncio.sleep(0.01)
|
||||||
|
|
||||||
|
async def _process_execution_queue(self, tg_id):
|
||||||
|
"""Continuously process execution queue for user."""
|
||||||
|
while tg_id in self.user_sockets:
|
||||||
|
try:
|
||||||
|
if self.execution_queues[tg_id]:
|
||||||
|
msg = self.execution_queues[tg_id].popleft()
|
||||||
|
await self.handle_execution_update(msg, tg_id)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error processing execution queue %s: %s", tg_id, e)
|
||||||
|
await asyncio.sleep(0.01)
|
||||||
|
|
||||||
|
async def close_user_socket(self, tg_id):
|
||||||
|
"""Gracefully close user connection."""
|
||||||
|
if tg_id in self.user_sockets:
|
||||||
|
self.user_sockets.pop(tg_id, None)
|
||||||
|
|
||||||
|
for key in (tg_id, tg_id + 1):
|
||||||
|
task = self.processing_tasks.pop(key, None)
|
||||||
|
if task and not task.done():
|
||||||
|
task.cancel()
|
||||||
|
|
||||||
|
self.order_queues.pop(tg_id, None)
|
||||||
|
self.execution_queues.pop(tg_id, None)
|
||||||
|
self.user_messages.pop(tg_id, None)
|
||||||
|
self.user_keys.pop(tg_id, None)
|
||||||
|
logger.info("Cleaned up user %s", tg_id)
|
||||||
|
|
||||||
async def handle_order_update(self, message, tg_id):
|
async def handle_order_update(self, message, tg_id):
|
||||||
"""Handle order updates."""
|
"""Process order updates."""
|
||||||
await self.message_handler.format_order_update(message, tg_id)
|
try:
|
||||||
|
await self.message_handler.format_order_update(message, tg_id)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error handling order update for %s: %s", tg_id, e)
|
||||||
|
|
||||||
async def handle_execution_update(self, message, tg_id):
|
async def handle_execution_update(self, message, tg_id):
|
||||||
"""Handle execution updates."""
|
"""Process execution updates."""
|
||||||
await self.message_handler.format_execution_update(message, tg_id)
|
try:
|
||||||
|
await self.message_handler.format_execution_update(message, tg_id)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error handling execution update for %s: %s", tg_id, e)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def get_users_from_db():
|
async def get_users_from_db():
|
||||||
"""Get all users from the database."""
|
"""Fetch all users from database."""
|
||||||
return await rq.get_users()
|
try:
|
||||||
|
return await rq.get_users()
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error getting users from DB: %s", e)
|
||||||
|
return []
|
||||||
|
|||||||
@@ -179,3 +179,9 @@ async def calculate_total_budget(
|
|||||||
|
|
||||||
total += r_quantity
|
total += r_quantity
|
||||||
return total
|
return total
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
async def truncate_float(f, decimals=4):
|
||||||
|
factor = 10 ** decimals
|
||||||
|
return int(f * factor) / factor
|
||||||
@@ -121,7 +121,7 @@ async def set_symbol(message: Message, state: FSMContext) -> None:
|
|||||||
)
|
)
|
||||||
|
|
||||||
await rq.set_leverage(tg_id=message.from_user.id, leverage=str(max_leverage))
|
await rq.set_leverage(tg_id=message.from_user.id, leverage=str(max_leverage))
|
||||||
risk_percent = 100 / safe_float(max_leverage)
|
risk_percent = 10 / safe_float(max_leverage)
|
||||||
await rq.set_stop_loss_percent(
|
await rq.set_stop_loss_percent(
|
||||||
tg_id=message.from_user.id, stop_loss_percent=risk_percent)
|
tg_id=message.from_user.id, stop_loss_percent=risk_percent)
|
||||||
await rq.set_take_profit_percent(
|
await rq.set_take_profit_percent(
|
||||||
|
|||||||
@@ -85,7 +85,19 @@ async def cmd_to_main(message: Message, state: FSMContext) -> None:
|
|||||||
None: Exceptions are caught and logged internally.
|
None: Exceptions are caught and logged internally.
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
await user_profile_tg(tg_id=message.from_user.id, message=message)
|
await state.clear()
|
||||||
|
user = await rq.get_user(tg_id=message.from_user.id)
|
||||||
|
if user:
|
||||||
|
await user_profile_tg(tg_id=message.from_user.id, message=message)
|
||||||
|
else:
|
||||||
|
await rq.create_user(
|
||||||
|
tg_id=message.from_user.id, username=message.from_user.username
|
||||||
|
)
|
||||||
|
await rq.set_user_symbol(tg_id=message.from_user.id, symbol="BTCUSDT")
|
||||||
|
await rq.create_user_additional_settings(tg_id=message.from_user.id)
|
||||||
|
await rq.create_user_risk_management(tg_id=message.from_user.id)
|
||||||
|
await rq.create_user_conditional_settings(tg_id=message.from_user.id)
|
||||||
|
await user_profile_tg(tg_id=message.from_user.id, message=message)
|
||||||
logger.debug(
|
logger.debug(
|
||||||
"Command to_profile_tg processed successfully for user: %s",
|
"Command to_profile_tg processed successfully for user: %s",
|
||||||
message.from_user.id,
|
message.from_user.id,
|
||||||
@@ -117,9 +129,21 @@ async def profile_bybit(message: Message, state: FSMContext) -> None:
|
|||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
await state.clear()
|
await state.clear()
|
||||||
await user_profile_bybit(
|
user = await rq.get_user(tg_id=message.from_user.id)
|
||||||
tg_id=message.from_user.id, message=message, state=state
|
if user:
|
||||||
)
|
await user_profile_bybit(
|
||||||
|
tg_id=message.from_user.id, message=message, state=state
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
await rq.create_user(
|
||||||
|
tg_id=message.from_user.id, username=message.from_user.username
|
||||||
|
)
|
||||||
|
await rq.set_user_symbol(tg_id=message.from_user.id, symbol="BTCUSDT")
|
||||||
|
await rq.create_user_additional_settings(tg_id=message.from_user.id)
|
||||||
|
await rq.create_user_risk_management(tg_id=message.from_user.id)
|
||||||
|
await rq.create_user_conditional_settings(tg_id=message.from_user.id)
|
||||||
|
await user_profile_bybit(
|
||||||
|
tg_id=message.from_user.id, message=message, state=state)
|
||||||
logger.debug(
|
logger.debug(
|
||||||
"Command to_profile_bybit processed successfully for user: %s",
|
"Command to_profile_bybit processed successfully for user: %s",
|
||||||
message.from_user.id,
|
message.from_user.id,
|
||||||
@@ -150,15 +174,31 @@ async def profile_bybit_callback(
|
|||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
await state.clear()
|
await state.clear()
|
||||||
await user_profile_bybit(
|
user = await rq.get_user(tg_id=callback_query.from_user.id)
|
||||||
tg_id=callback_query.from_user.id,
|
|
||||||
message=callback_query.message,
|
if user:
|
||||||
state=state,
|
await user_profile_bybit(
|
||||||
)
|
tg_id=callback_query.from_user.id,
|
||||||
logger.debug(
|
message=callback_query.message,
|
||||||
"Callback profile_bybit processed successfully for user: %s",
|
state=state,
|
||||||
callback_query.from_user.id,
|
)
|
||||||
)
|
logger.debug(
|
||||||
|
"Callback profile_bybit processed successfully for user: %s",
|
||||||
|
callback_query.from_user.id,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
await rq.create_user(
|
||||||
|
tg_id=callback_query.from_user.id, username=callback_query.from_user.username
|
||||||
|
)
|
||||||
|
await rq.set_user_symbol(tg_id=callback_query.from_user.id, symbol="BTCUSDT")
|
||||||
|
await rq.create_user_additional_settings(tg_id=callback_query.from_user.id)
|
||||||
|
await rq.create_user_risk_management(tg_id=callback_query.from_user.id)
|
||||||
|
await rq.create_user_conditional_settings(tg_id=callback_query.from_user.id)
|
||||||
|
await user_profile_bybit(
|
||||||
|
tg_id=callback_query.from_user.id,
|
||||||
|
message=callback_query.message,
|
||||||
|
state=state,
|
||||||
|
)
|
||||||
await callback_query.answer()
|
await callback_query.answer()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(
|
logger.error(
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ async def settings_for_trade_mode(
|
|||||||
text="Выберите режим торговли:\n\n"
|
text="Выберите режим торговли:\n\n"
|
||||||
"Лонг - все сделки серии открываются на покупку.\n"
|
"Лонг - все сделки серии открываются на покупку.\n"
|
||||||
"Шорт - все сделки серии открываются на продажу.\n"
|
"Шорт - все сделки серии открываются на продажу.\n"
|
||||||
"Свитч - направление первой сделки серии меняется по переменно.\n",
|
"Свитч - направление каждой сделки в рамках серии меняется попеременно.\n",
|
||||||
reply_markup=kbi.trade_mode,
|
reply_markup=kbi.trade_mode,
|
||||||
)
|
)
|
||||||
logger.debug(
|
logger.debug(
|
||||||
@@ -105,10 +105,10 @@ async def trade_mode(callback_query: CallbackQuery, state: FSMContext) -> None:
|
|||||||
await state.clear()
|
await state.clear()
|
||||||
|
|
||||||
|
|
||||||
@router_additional_settings.callback_query(F.data == "switch_side_start")
|
@router_additional_settings.callback_query(F.data == "switch_side_second")
|
||||||
async def switch_side_start(callback_query: CallbackQuery, state: FSMContext) -> None:
|
async def switch_side_second(callback_query: CallbackQuery, state: FSMContext) -> None:
|
||||||
"""
|
"""
|
||||||
Handles the 'switch_side_start' callback query.
|
Handles the 'switch_side_second' callback query.
|
||||||
|
|
||||||
Clears the current FSM state, edits the message text to display the switch side start message,
|
Clears the current FSM state, edits the message text to display the switch side start message,
|
||||||
and shows an inline keyboard for selection.
|
and shows an inline keyboard for selection.
|
||||||
@@ -123,13 +123,13 @@ async def switch_side_start(callback_query: CallbackQuery, state: FSMContext) ->
|
|||||||
try:
|
try:
|
||||||
await state.clear()
|
await state.clear()
|
||||||
await callback_query.message.edit_text(
|
await callback_query.message.edit_text(
|
||||||
text="Выберите направление первой сделки серии:\n\n"
|
text="Выберите направление первой сделки последующих серии:\n\n"
|
||||||
"По направлению - сделка открывается в направлении последней сделки предыдущей серии.\n"
|
"По направлению - сделка открывается в направлении последней сделки предыдущей серии.\n"
|
||||||
"Противоположно - сделка открывается в противоположном направлении последней сделки предыдущей серии.\n",
|
"Противоположно - сделка открывается в противоположном направлении последней сделки предыдущей серии.\n",
|
||||||
reply_markup=kbi.switch_side,
|
reply_markup=kbi.switch_side,
|
||||||
)
|
)
|
||||||
logger.debug(
|
logger.debug(
|
||||||
"Command switch_side_start processed successfully for user: %s",
|
"Command switch_side_second processed successfully for user: %s",
|
||||||
callback_query.from_user.id,
|
callback_query.from_user.id,
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
@@ -137,7 +137,7 @@ async def switch_side_start(callback_query: CallbackQuery, state: FSMContext) ->
|
|||||||
text="Произошла ошибка. Пожалуйста, попробуйте позже."
|
text="Произошла ошибка. Пожалуйста, попробуйте позже."
|
||||||
)
|
)
|
||||||
logger.error(
|
logger.error(
|
||||||
"Error processing command switch_side_start for user %s: %s",
|
"Error processing command switch_side_second for user %s: %s",
|
||||||
callback_query.from_user.id,
|
callback_query.from_user.id,
|
||||||
e,
|
e,
|
||||||
)
|
)
|
||||||
@@ -193,6 +193,89 @@ async def switch_side_handler(callback_query: CallbackQuery, state: FSMContext)
|
|||||||
await state.clear()
|
await state.clear()
|
||||||
|
|
||||||
|
|
||||||
|
@router_additional_settings.callback_query(F.data == "switch_side_start")
|
||||||
|
async def switch_side_start(callback_query: CallbackQuery, state: FSMContext) -> None:
|
||||||
|
"""
|
||||||
|
Handles the 'switch_side_start' callback query.
|
||||||
|
|
||||||
|
Clears the current FSM state, edits the message text to display the switch side second message,
|
||||||
|
and shows an inline keyboard for selection.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
callback_query (CallbackQuery): Incoming callback query from Telegram inline keyboard.
|
||||||
|
state (FSMContext): Finite State Machine context for the current user session.
|
||||||
|
|
||||||
|
Logs:
|
||||||
|
Success or error messages with user identification.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
await state.clear()
|
||||||
|
await callback_query.message.edit_text(
|
||||||
|
text="Выберите направление первой сделки:\n\n", reply_markup=kbi.side_for_switch
|
||||||
|
)
|
||||||
|
logger.debug(
|
||||||
|
"Command switch_side_start processed successfully for user: %s",
|
||||||
|
callback_query.from_user.id,
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
await callback_query.answer(
|
||||||
|
text="Произошла ошибка. Пожалуйста, попробуйте позже."
|
||||||
|
)
|
||||||
|
logger.error(
|
||||||
|
"Error processing command switch_side_start for user %s: %s",
|
||||||
|
callback_query.from_user.id,
|
||||||
|
e,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@router_additional_settings.callback_query(lambda c: c.data == "buy_switch" or c.data == "sell_switch")
|
||||||
|
async def switch_side_handler_2(callback_query: CallbackQuery, state: FSMContext) -> None:
|
||||||
|
"""
|
||||||
|
Handles callback queries related to switch side selection.
|
||||||
|
|
||||||
|
Updates FSM context with selected switch side and persists the choice in database.
|
||||||
|
Sends an acknowledgement to user and clears FSM state afterward.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
callback_query (CallbackQuery): Incoming callback query indicating selected switch side.
|
||||||
|
state (FSMContext): Finite State Machine context for the current user session.
|
||||||
|
|
||||||
|
Logs:
|
||||||
|
Success or error messages with user identification.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
if callback_query.data == "sell_switch":
|
||||||
|
side = "Sell"
|
||||||
|
side_rus = "Продажа"
|
||||||
|
else:
|
||||||
|
side = "Buy"
|
||||||
|
side_rus = "Покупка"
|
||||||
|
|
||||||
|
req = await rq.set_side(
|
||||||
|
tg_id=callback_query.from_user.id, side=side
|
||||||
|
)
|
||||||
|
|
||||||
|
if not req:
|
||||||
|
await callback_query.answer(
|
||||||
|
text="Произошла ошибка при смене направления. Пожалуйста, попробуйте позже."
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
|
await callback_query.answer(text=f"Выбрано: {side_rus}")
|
||||||
|
logger.debug(
|
||||||
|
"Switch side changed successfully for user: %s", callback_query.from_user.id
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
await callback_query.answer(text="Произошла ошибка при смене направления.")
|
||||||
|
logger.error(
|
||||||
|
"Error processing set switch_side for user %s: %s",
|
||||||
|
callback_query.from_user.id,
|
||||||
|
e,
|
||||||
|
)
|
||||||
|
finally:
|
||||||
|
await state.clear()
|
||||||
|
|
||||||
|
|
||||||
@router_additional_settings.callback_query(F.data == "margin_type")
|
@router_additional_settings.callback_query(F.data == "margin_type")
|
||||||
async def settings_for_margin_type(
|
async def settings_for_margin_type(
|
||||||
callback_query: CallbackQuery, state: FSMContext
|
callback_query: CallbackQuery, state: FSMContext
|
||||||
@@ -216,6 +299,12 @@ async def settings_for_margin_type(
|
|||||||
deals = await get_active_positions_by_symbol(
|
deals = await get_active_positions_by_symbol(
|
||||||
tg_id=callback_query.from_user.id, symbol=symbol
|
tg_id=callback_query.from_user.id, symbol=symbol
|
||||||
)
|
)
|
||||||
|
if deals == "Invalid API key permissions":
|
||||||
|
await callback_query.answer(
|
||||||
|
text="API ключ не имеет достаточных прав для смены маржи",
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
position = next((d for d in deals if d.get("symbol") == symbol), None)
|
position = next((d for d in deals if d.get("symbol") == symbol), None)
|
||||||
|
|
||||||
if position:
|
if position:
|
||||||
@@ -577,7 +666,7 @@ async def set_leverage_handler(message: Message, state: FSMContext) -> None:
|
|||||||
text=f"Кредитное плечо успешно установлено на {leverage_float}",
|
text=f"Кредитное плечо успешно установлено на {leverage_float}",
|
||||||
reply_markup=kbi.back_to_additional_settings,
|
reply_markup=kbi.back_to_additional_settings,
|
||||||
)
|
)
|
||||||
risk_percent = 100 / safe_float(leverage_float)
|
risk_percent = 10 / safe_float(leverage_float)
|
||||||
await rq.set_stop_loss_percent(
|
await rq.set_stop_loss_percent(
|
||||||
tg_id=message.from_user.id, stop_loss_percent=risk_percent)
|
tg_id=message.from_user.id, stop_loss_percent=risk_percent)
|
||||||
await rq.set_take_profit_percent(
|
await rq.set_take_profit_percent(
|
||||||
@@ -593,10 +682,19 @@ async def set_leverage_handler(message: Message, state: FSMContext) -> None:
|
|||||||
|
|
||||||
await state.clear()
|
await state.clear()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
await message.answer(
|
errors_text = str(e)
|
||||||
text="Произошла ошибка при установке кредитного плеча. Пожалуйста, попробуйте позже.",
|
known_errors = {
|
||||||
reply_markup=kbi.back_to_additional_settings,
|
"Permission denied, please check your API key permissions": "API ключ не имеет достаточных прав для установки кредитного плеча"
|
||||||
)
|
|
||||||
|
}
|
||||||
|
for key, msg in known_errors.items():
|
||||||
|
if key in errors_text:
|
||||||
|
await message.answer(msg, reply_markup=kbi.back_to_additional_settings)
|
||||||
|
else:
|
||||||
|
await message.answer(
|
||||||
|
text="Произошла ошибка при установке кредитного плеча. Пожалуйста, попробуйте позже.",
|
||||||
|
reply_markup=kbi.back_to_additional_settings,
|
||||||
|
)
|
||||||
logger.error(
|
logger.error(
|
||||||
"Error processing command leverage for user %s: %s", message.from_user.id, e
|
"Error processing command leverage for user %s: %s", message.from_user.id, e
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -98,7 +98,7 @@ async def set_take_profit_percent(message: Message, state: FSMContext) -> None:
|
|||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
if safe_float(take_profit_percent_value) < 1 or safe_float(take_profit_percent_value) > 100:
|
if safe_float(take_profit_percent_value) < 0.1 or safe_float(take_profit_percent_value) > 100:
|
||||||
await message.answer(
|
await message.answer(
|
||||||
text="Ошибка: введите число от 1 до 100.",
|
text="Ошибка: введите число от 1 до 100.",
|
||||||
reply_markup=kbi.back_to_risk_management,
|
reply_markup=kbi.back_to_risk_management,
|
||||||
@@ -219,7 +219,7 @@ async def set_stop_loss_percent(message: Message, state: FSMContext) -> None:
|
|||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
if safe_float(stop_loss_percent_value) < 1 or safe_float(stop_loss_percent_value) > 100:
|
if safe_float(stop_loss_percent_value) < 0.1 or safe_float(stop_loss_percent_value) > 100:
|
||||||
await message.answer(
|
await message.answer(
|
||||||
text="Ошибка: введите число от 1 до 100.",
|
text="Ошибка: введите число от 1 до 100.",
|
||||||
reply_markup=kbi.back_to_risk_management,
|
reply_markup=kbi.back_to_risk_management,
|
||||||
@@ -341,3 +341,83 @@ async def set_commission_fee(callback_query: CallbackQuery, state: FSMContext) -
|
|||||||
)
|
)
|
||||||
finally:
|
finally:
|
||||||
await state.clear()
|
await state.clear()
|
||||||
|
|
||||||
|
|
||||||
|
@router_risk_management.callback_query(F.data == "compensation_commission")
|
||||||
|
async def compensation_commission(callback_query: CallbackQuery, state: FSMContext) -> None:
|
||||||
|
"""
|
||||||
|
Handles the 'compensation_commission' callback query.
|
||||||
|
|
||||||
|
Clears the current FSM state, edits the message text to display the compensation commission options,
|
||||||
|
and shows an inline keyboard for selection.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
callback_query (CallbackQuery): Incoming callback query from Telegram inline keyboard.
|
||||||
|
state (FSMContext): Finite State Machine context for the current user session.
|
||||||
|
|
||||||
|
Logs:
|
||||||
|
Success or error messages with user identification.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
await state.clear()
|
||||||
|
msg = await callback_query.message.edit_text(
|
||||||
|
text="Выберите за счет чего будет происходить компенсация комиссии: ",
|
||||||
|
reply_markup=kbi.commission_place,
|
||||||
|
)
|
||||||
|
await state.update_data(prompt_message_id=msg.message_id)
|
||||||
|
logger.debug(
|
||||||
|
"Command compensation_commission processed successfully for user: %s",
|
||||||
|
callback_query.from_user.id,
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
await callback_query.answer(
|
||||||
|
text="Произошла ошибка. Пожалуйста, попробуйте позже."
|
||||||
|
)
|
||||||
|
logger.error(
|
||||||
|
"Error processing command compensation_commission for user %s: %s",
|
||||||
|
callback_query.from_user.id,
|
||||||
|
e,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@router_risk_management.callback_query(
|
||||||
|
lambda c: c.data in ["Commission_for_qty", "Commission_for_tp"]
|
||||||
|
)
|
||||||
|
async def set_compensation_commission(callback_query: CallbackQuery, state: FSMContext) -> None:
|
||||||
|
"""
|
||||||
|
Handles user input for setting the compensation commission.
|
||||||
|
|
||||||
|
Updates FSM context with the selected option and persists the choice in database.
|
||||||
|
Sends an acknowledgement to user and clears FSM state afterward.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
callback_query (CallbackQuery): Incoming callback query from Telegram inline keyboard.
|
||||||
|
state (FSMContext): Finite State Machine context for the current user session.
|
||||||
|
|
||||||
|
Logs:
|
||||||
|
Success or error messages with user identification.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
|
||||||
|
req = await rq.set_commission_place(
|
||||||
|
tg_id=callback_query.from_user.id, commission_place=callback_query.data
|
||||||
|
)
|
||||||
|
|
||||||
|
if not req:
|
||||||
|
await callback_query.answer(
|
||||||
|
text="Произошла ошибка при установке компенсации комиссии. Пожалуйста, попробуйте позже."
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
|
if callback_query.data == "Commission_for_qty":
|
||||||
|
await callback_query.answer(text="Комиссия компенсируется по ставке.")
|
||||||
|
else:
|
||||||
|
await callback_query.answer(text="Комиссия компенсируется по тейк-профиту.")
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(
|
||||||
|
"Error processing command compensation_commission for user %s: %s",
|
||||||
|
callback_query.from_user.id,
|
||||||
|
e,
|
||||||
|
)
|
||||||
|
finally:
|
||||||
|
await state.clear()
|
||||||
@@ -62,10 +62,19 @@ async def additional_settings(callback_query: CallbackQuery, state: FSMContext)
|
|||||||
max_bets = additional_data.max_bets_in_series
|
max_bets = additional_data.max_bets_in_series
|
||||||
quantity = f(additional_data.order_quantity)
|
quantity = f(additional_data.order_quantity)
|
||||||
trigger_price = f(additional_data.trigger_price) or 0
|
trigger_price = f(additional_data.trigger_price) or 0
|
||||||
|
side = additional_data.side
|
||||||
|
|
||||||
|
side_map = {
|
||||||
|
"Buy": "Лонг",
|
||||||
|
"Sell": "Шорт",
|
||||||
|
}
|
||||||
|
side_rus = side_map.get(side, side)
|
||||||
|
|
||||||
switch_side_mode = ""
|
switch_side_mode = ""
|
||||||
|
side = ""
|
||||||
if trade_mode == "Switch":
|
if trade_mode == "Switch":
|
||||||
switch_side_mode = f"- Направление первой сделки: {switch_side}\n"
|
side = f"- Направление первой сделки: {side_rus}\n"
|
||||||
|
switch_side_mode = f"- Направление первой сделки последующих серии: {switch_side}\n"
|
||||||
|
|
||||||
total_budget = await calculate_total_budget(
|
total_budget = await calculate_total_budget(
|
||||||
quantity=quantity,
|
quantity=quantity,
|
||||||
@@ -75,6 +84,7 @@ async def additional_settings(callback_query: CallbackQuery, state: FSMContext)
|
|||||||
text = (
|
text = (
|
||||||
f"Основные настройки:\n\n"
|
f"Основные настройки:\n\n"
|
||||||
f"- Режим торговли: {trade_mode_rus}\n"
|
f"- Режим торговли: {trade_mode_rus}\n"
|
||||||
|
f"{side}"
|
||||||
f"{switch_side_mode}"
|
f"{switch_side_mode}"
|
||||||
f"- Тип маржи: {margin_type_rus}\n"
|
f"- Тип маржи: {margin_type_rus}\n"
|
||||||
f"- Размер кредитного плеча: {leverage:.2f}\n"
|
f"- Размер кредитного плеча: {leverage:.2f}\n"
|
||||||
@@ -120,12 +130,17 @@ async def risk_management(callback_query: CallbackQuery, state: FSMContext) -> N
|
|||||||
commission_fee_rus = (
|
commission_fee_rus = (
|
||||||
"Да" if commission_fee == "Yes_commission_fee" else "Нет"
|
"Да" if commission_fee == "Yes_commission_fee" else "Нет"
|
||||||
)
|
)
|
||||||
|
commission_place = risk_management_data.commission_place
|
||||||
|
commission_place_rus = (
|
||||||
|
"Ставке" if commission_place == "Commission_for_qty" else "Тейк-профиту"
|
||||||
|
)
|
||||||
|
|
||||||
await callback_query.message.edit_text(
|
await callback_query.message.edit_text(
|
||||||
text=f"Риск-менеджмент:\n\n"
|
text=f"Риск-менеджмент:\n\n"
|
||||||
f"- Процент изменения цены для фиксации прибыли: {take_profit_percent:.2f}%\n"
|
f"- Процент изменения цены для фиксации прибыли: {take_profit_percent:.2f}%\n"
|
||||||
f"- Процент изменения цены для фиксации убытка: {stop_loss_percent:.2f}%\n\n"
|
f"- Процент изменения цены для фиксации убытка: {stop_loss_percent:.2f}%\n\n"
|
||||||
f"- Комиссия биржи для расчета прибыли: {commission_fee_rus}\n\n",
|
f"- Комиссия биржи для расчета прибыли: {commission_fee_rus}\n\n"
|
||||||
|
f"- Компенсация комиссии по: {commission_place_rus}",
|
||||||
reply_markup=kbi.risk_management,
|
reply_markup=kbi.risk_management,
|
||||||
)
|
)
|
||||||
logger.debug(
|
logger.debug(
|
||||||
@@ -164,7 +179,7 @@ async def conditions(callback_query: CallbackQuery, state: FSMContext) -> None:
|
|||||||
start_timer = conditional_settings_data.timer_start or 0
|
start_timer = conditional_settings_data.timer_start or 0
|
||||||
await callback_query.message.edit_text(
|
await callback_query.message.edit_text(
|
||||||
text="Условия торговли:\n\n"
|
text="Условия торговли:\n\n"
|
||||||
f"- Таймер для старта: {start_timer} мин.\n",
|
f"- Таймер для старта: {start_timer} мин.\n",
|
||||||
reply_markup=kbi.conditions,
|
reply_markup=kbi.conditions,
|
||||||
)
|
)
|
||||||
logger.debug(
|
logger.debug(
|
||||||
|
|||||||
@@ -38,6 +38,12 @@ async def start_trading(callback_query: CallbackQuery, state: FSMContext) -> Non
|
|||||||
deals = await get_active_positions_by_symbol(
|
deals = await get_active_positions_by_symbol(
|
||||||
tg_id=callback_query.from_user.id, symbol=symbol
|
tg_id=callback_query.from_user.id, symbol=symbol
|
||||||
)
|
)
|
||||||
|
if deals == "Invalid API key permissions":
|
||||||
|
await callback_query.answer(
|
||||||
|
text="API ключ не имеет достаточных прав для запуска торговли",
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
position = next((d for d in deals if d.get("symbol") == symbol), None)
|
position = next((d for d in deals if d.get("symbol") == symbol), None)
|
||||||
|
|
||||||
if position:
|
if position:
|
||||||
@@ -97,7 +103,8 @@ async def start_trading(callback_query: CallbackQuery, state: FSMContext) -> Non
|
|||||||
"Limit price is out min price": "Цена лимитного ордера меньше допустимого",
|
"Limit price is out min price": "Цена лимитного ордера меньше допустимого",
|
||||||
"Limit price is out max price": "Цена лимитного ордера больше допустимого",
|
"Limit price is out max price": "Цена лимитного ордера больше допустимого",
|
||||||
"Risk is too high for this trade": "Риск сделки превышает допустимый убыток",
|
"Risk is too high for this trade": "Риск сделки превышает допустимый убыток",
|
||||||
"estimated will trigger liq": "Лимитный ордер может вызвать мгновенную ликвидацию. Проверьте параметры ордера.",
|
"estimated will trigger liq": "Лимитный ордер может вызвать мгновенную ликвидацию. "
|
||||||
|
"Проверьте параметры ордера.",
|
||||||
"ab not enough for new order": "Недостаточно средств для создания нового ордера",
|
"ab not enough for new order": "Недостаточно средств для создания нового ордера",
|
||||||
"InvalidRequestError": "Произошла ошибка при запуске торговли.",
|
"InvalidRequestError": "Произошла ошибка при запуске торговли.",
|
||||||
"Order does not meet minimum order value": "Сумма ставки меньше допустимого для запуска торговли. "
|
"Order does not meet minimum order value": "Сумма ставки меньше допустимого для запуска торговли. "
|
||||||
@@ -106,6 +113,11 @@ async def start_trading(callback_query: CallbackQuery, state: FSMContext) -> Non
|
|||||||
"Qty invalid": "Некорректное значение ставки для данного инструмента",
|
"Qty invalid": "Некорректное значение ставки для данного инструмента",
|
||||||
"The number of contracts exceeds maximum limit allowed": "️️Превышен максимальный лимит ставки",
|
"The number of contracts exceeds maximum limit allowed": "️️Превышен максимальный лимит ставки",
|
||||||
"The number of contracts exceeds minimum limit allowed": "️️Лимит ставки меньше минимально допустимого",
|
"The number of contracts exceeds minimum limit allowed": "️️Лимит ставки меньше минимально допустимого",
|
||||||
|
"Order placement failed as your position may exceed the max":
|
||||||
|
"Не удалось разместить ордер, так как ваша позиция может превышать максимальный лимит."
|
||||||
|
"Пожалуйста, уменьшите кредитное плечо, чтобы увеличить максимальное значение",
|
||||||
|
"Permission denied, please check your API key permissions": "API ключ не имеет достаточных прав для запуска торговли"
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if res == "OK":
|
if res == "OK":
|
||||||
@@ -127,7 +139,16 @@ async def start_trading(callback_query: CallbackQuery, state: FSMContext) -> Non
|
|||||||
await add_start_task_merged(user_id=callback_query.from_user.id, task=task)
|
await add_start_task_merged(user_id=callback_query.from_user.id, task=task)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
await callback_query.answer(text="Произошла ошибка при запуске торговли")
|
error_text = str(e)
|
||||||
|
known_errors = {
|
||||||
|
"Permission denied, please check your API key permissions": "API ключ не имеет достаточных прав для запуска торговли"
|
||||||
|
|
||||||
|
}
|
||||||
|
for key, msg in known_errors.items():
|
||||||
|
if key in error_text:
|
||||||
|
await callback_query.answer(msg)
|
||||||
|
else:
|
||||||
|
await callback_query.answer(text="Произошла ошибка при запуске торговли")
|
||||||
logger.error(
|
logger.error(
|
||||||
"Error processing command start_trading for user %s: %s",
|
"Error processing command start_trading for user %s: %s",
|
||||||
callback_query.from_user.id,
|
callback_query.from_user.id,
|
||||||
@@ -141,7 +162,7 @@ async def start_trading(callback_query: CallbackQuery, state: FSMContext) -> Non
|
|||||||
lambda c: c.data == "cancel_timer_merged"
|
lambda c: c.data == "cancel_timer_merged"
|
||||||
)
|
)
|
||||||
async def cancel_start_trading(
|
async def cancel_start_trading(
|
||||||
callback_query: CallbackQuery, state: FSMContext
|
callback_query: CallbackQuery, state: FSMContext
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
Handles the "cancel_timer" callback query.
|
Handles the "cancel_timer" callback query.
|
||||||
|
|||||||
@@ -39,21 +39,15 @@ async def stop_all_trading(callback_query: CallbackQuery, state: FSMContext):
|
|||||||
await rq.set_stop_timer(tg_id=callback_query.from_user.id, timer_end=0)
|
await rq.set_stop_timer(tg_id=callback_query.from_user.id, timer_end=0)
|
||||||
await asyncio.sleep(timer_end * 60)
|
await asyncio.sleep(timer_end * 60)
|
||||||
|
|
||||||
user_auto_trading = await rq.get_user_auto_trading(
|
await close_position_by_symbol(
|
||||||
tg_id=callback_query.from_user.id, symbol=symbol
|
tg_id=callback_query.from_user.id, symbol=symbol)
|
||||||
|
await rq.set_auto_trading(
|
||||||
|
tg_id=callback_query.from_user.id,
|
||||||
|
symbol=symbol,
|
||||||
|
auto_trading=False,
|
||||||
)
|
)
|
||||||
|
await callback_query.message.edit_text(text=f"Торговля для {symbol} остановлена", reply_markup=kbi.profile_bybit)
|
||||||
|
|
||||||
if user_auto_trading and user_auto_trading.auto_trading:
|
|
||||||
await rq.set_auto_trading(
|
|
||||||
tg_id=callback_query.from_user.id,
|
|
||||||
symbol=symbol,
|
|
||||||
auto_trading=False,
|
|
||||||
)
|
|
||||||
await close_position_by_symbol(
|
|
||||||
tg_id=callback_query.from_user.id, symbol=symbol)
|
|
||||||
await callback_query.message.edit_text(text=f"Торговля для {symbol} остановлена", reply_markup=kbi.profile_bybit)
|
|
||||||
else:
|
|
||||||
await callback_query.message.edit_text(text=f"Нет активной торговли для {symbol}", reply_markup=kbi.profile_bybit)
|
|
||||||
|
|
||||||
task = asyncio.create_task(delay_start())
|
task = asyncio.create_task(delay_start())
|
||||||
await add_stop_task(user_id=callback_query.from_user.id, task=task)
|
await add_stop_task(user_id=callback_query.from_user.id, task=task)
|
||||||
|
|||||||
@@ -94,7 +94,10 @@ def get_additional_settings_keyboard(mode: str
|
|||||||
|
|
||||||
if mode == "Switch":
|
if mode == "Switch":
|
||||||
buttons.append(
|
buttons.append(
|
||||||
[InlineKeyboardButton(text="Направление первой сделки", callback_data="switch_side_start")]
|
[InlineKeyboardButton(text="Направление первой сделки первой серии", callback_data="switch_side_start")]
|
||||||
|
)
|
||||||
|
buttons.append(
|
||||||
|
[InlineKeyboardButton(text="Направление первой сделки последующих серии", callback_data="switch_side_second")]
|
||||||
)
|
)
|
||||||
|
|
||||||
buttons.append(
|
buttons.append(
|
||||||
@@ -148,6 +151,19 @@ switch_side = InlineKeyboardMarkup(
|
|||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
side_for_switch = InlineKeyboardMarkup(
|
||||||
|
inline_keyboard=[
|
||||||
|
[
|
||||||
|
InlineKeyboardButton(text="Лонг", callback_data="buy_switch"),
|
||||||
|
InlineKeyboardButton(text="Шорт", callback_data="sell_switch"),
|
||||||
|
],
|
||||||
|
[
|
||||||
|
InlineKeyboardButton(text="Назад", callback_data="additional_settings"),
|
||||||
|
InlineKeyboardButton(text="На главную", callback_data="profile_bybit"),
|
||||||
|
],
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
margin_type = InlineKeyboardMarkup(
|
margin_type = InlineKeyboardMarkup(
|
||||||
inline_keyboard=[
|
inline_keyboard=[
|
||||||
[
|
[
|
||||||
@@ -194,6 +210,7 @@ risk_management = InlineKeyboardMarkup(
|
|||||||
),
|
),
|
||||||
],
|
],
|
||||||
[InlineKeyboardButton(text="Комиссия биржи", callback_data="commission_fee")],
|
[InlineKeyboardButton(text="Комиссия биржи", callback_data="commission_fee")],
|
||||||
|
[InlineKeyboardButton(text="Компенсация комиссии", callback_data="compensation_commission")],
|
||||||
[
|
[
|
||||||
InlineKeyboardButton(text="Назад", callback_data="main_settings"),
|
InlineKeyboardButton(text="Назад", callback_data="main_settings"),
|
||||||
InlineKeyboardButton(text="На главную", callback_data="profile_bybit"),
|
InlineKeyboardButton(text="На главную", callback_data="profile_bybit"),
|
||||||
@@ -223,6 +240,20 @@ commission_fee = InlineKeyboardMarkup(
|
|||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
commission_place = InlineKeyboardMarkup(
|
||||||
|
inline_keyboard=[
|
||||||
|
[
|
||||||
|
InlineKeyboardButton(text="По ставке", callback_data="Commission_for_qty"),
|
||||||
|
InlineKeyboardButton(text="По тейк-профиту", callback_data="Commission_for_tp"),
|
||||||
|
],
|
||||||
|
[
|
||||||
|
InlineKeyboardButton(text="Назад", callback_data="risk_management"),
|
||||||
|
InlineKeyboardButton(text="На главную", callback_data="profile_bybit"),
|
||||||
|
],
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
# conditions
|
# conditions
|
||||||
conditions = InlineKeyboardMarkup(
|
conditions = InlineKeyboardMarkup(
|
||||||
inline_keyboard=[
|
inline_keyboard=[
|
||||||
|
|||||||
@@ -10,10 +10,9 @@ logging.config.dictConfig(LOGGING_CONFIG)
|
|||||||
logger = logging.getLogger("database")
|
logger = logging.getLogger("database")
|
||||||
|
|
||||||
BASE_DIR = Path(__file__).parent.resolve()
|
BASE_DIR = Path(__file__).parent.resolve()
|
||||||
DATA_DIR = BASE_DIR / "db"
|
BASE_DIR.mkdir(parents=True, exist_ok=True)
|
||||||
DATA_DIR.mkdir(parents=True, exist_ok=True)
|
|
||||||
|
|
||||||
DATABASE_URL = f"sqlite+aiosqlite:///{DATA_DIR / 'stcs.db'}"
|
DATABASE_URL = f"sqlite+aiosqlite:///{BASE_DIR / 'stcs.db'}"
|
||||||
|
|
||||||
async_engine = create_async_engine(
|
async_engine = create_async_engine(
|
||||||
DATABASE_URL,
|
DATABASE_URL,
|
||||||
@@ -39,7 +38,7 @@ async_session = async_sessionmaker(
|
|||||||
async def init_db():
|
async def init_db():
|
||||||
try:
|
try:
|
||||||
async with async_engine.begin() as conn:
|
async with async_engine.begin() as conn:
|
||||||
await conn.run_sync(Base.metadata.create_all)
|
await conn.run_sync(lambda sync_conn: Base.metadata.create_all(bind=sync_conn, checkfirst=True))
|
||||||
logger.info("Database initialized.")
|
logger.info("Database initialized.")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error("Database initialization failed: %s", e)
|
logger.error("Database initialization failed: %s", e)
|
||||||
@@ -92,6 +92,7 @@ class UserAdditionalSettings(Base):
|
|||||||
nullable=False, unique=True)
|
nullable=False, unique=True)
|
||||||
trade_mode = Column(String, nullable=False, default="Merged_Single")
|
trade_mode = Column(String, nullable=False, default="Merged_Single")
|
||||||
switch_side = Column(String, nullable=False, default="По направлению")
|
switch_side = Column(String, nullable=False, default="По направлению")
|
||||||
|
side = Column(String, nullable=False, default="Buy")
|
||||||
trigger_price = Column(Float, nullable=False, default=0.0)
|
trigger_price = Column(Float, nullable=False, default=0.0)
|
||||||
margin_type = Column(String, nullable=False, default="ISOLATED_MARGIN")
|
margin_type = Column(String, nullable=False, default="ISOLATED_MARGIN")
|
||||||
leverage = Column(String, nullable=False, default="10")
|
leverage = Column(String, nullable=False, default="10")
|
||||||
@@ -113,6 +114,7 @@ class UserRiskManagement(Base):
|
|||||||
take_profit_percent = Column(Float, nullable=False, default=1)
|
take_profit_percent = Column(Float, nullable=False, default=1)
|
||||||
stop_loss_percent = Column(Float, nullable=False, default=1)
|
stop_loss_percent = Column(Float, nullable=False, default=1)
|
||||||
commission_fee = Column(String, nullable=False, default="Yes_commission_fee")
|
commission_fee = Column(String, nullable=False, default="Yes_commission_fee")
|
||||||
|
commission_place = Column(String, nullable=False, default="Commission_for_qty")
|
||||||
|
|
||||||
user = relationship("User", back_populates="user_risk_management")
|
user = relationship("User", back_populates="user_risk_management")
|
||||||
|
|
||||||
@@ -152,9 +154,15 @@ class UserDeals(Base):
|
|||||||
order_quantity = Column(Float, nullable=True)
|
order_quantity = Column(Float, nullable=True)
|
||||||
martingale_factor = Column(Float, nullable=True)
|
martingale_factor = Column(Float, nullable=True)
|
||||||
max_bets_in_series = Column(Integer, nullable=True)
|
max_bets_in_series = Column(Integer, nullable=True)
|
||||||
take_profit_percent = Column(Integer, nullable=True)
|
take_profit_percent = Column(Float, nullable=True)
|
||||||
stop_loss_percent = Column(Integer, nullable=True)
|
stop_loss_percent = Column(Float, nullable=True)
|
||||||
trigger_price = Column(Float, nullable=True)
|
trigger_price = Column(Float, nullable=True)
|
||||||
|
current_series = Column(Integer, nullable=True)
|
||||||
|
commission_fee = Column(String, nullable=True)
|
||||||
|
commission_place = Column(String, nullable=True)
|
||||||
|
pnl_series = Column(Float, nullable=True)
|
||||||
|
take_profit = Column(Float, nullable=False, default=0.0)
|
||||||
|
stop_loss = Column(Float, nullable=False, default=0.0)
|
||||||
|
|
||||||
user = relationship("User", back_populates="user_deals")
|
user = relationship("User", back_populates="user_deals")
|
||||||
|
|
||||||
|
|||||||
@@ -86,7 +86,7 @@ async def set_user_api(tg_id: int, api_key: str, api_secret: str) -> bool:
|
|||||||
else:
|
else:
|
||||||
# Creating new record
|
# Creating new record
|
||||||
user_api = UserApi(
|
user_api = UserApi(
|
||||||
user=user, api_key=api_key, api_secret=api_secret
|
user_id=user.id, api_key=api_key, api_secret=api_secret
|
||||||
)
|
)
|
||||||
session.add(user_api)
|
session.add(user_api)
|
||||||
|
|
||||||
@@ -141,7 +141,7 @@ async def set_user_symbol(tg_id: int, symbol: str) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_symbol = UserSymbol(
|
user_symbol = UserSymbol(
|
||||||
symbol=symbol,
|
symbol=symbol,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_symbol)
|
session.add(user_symbol)
|
||||||
|
|
||||||
@@ -197,9 +197,11 @@ async def create_user_additional_settings(tg_id: int) -> None:
|
|||||||
|
|
||||||
# Create the user additional settings
|
# Create the user additional settings
|
||||||
user_additional_settings = UserAdditionalSettings(
|
user_additional_settings = UserAdditionalSettings(
|
||||||
user=user,
|
user_id=user.id,
|
||||||
trade_mode="Long", # Default value
|
trade_mode="Long", # Default value
|
||||||
switch_side="По направлению",
|
switch_side="По направлению",
|
||||||
|
side="Buy",
|
||||||
|
trigger_price=0.0,
|
||||||
margin_type="ISOLATED_MARGIN",
|
margin_type="ISOLATED_MARGIN",
|
||||||
leverage="10",
|
leverage="10",
|
||||||
order_quantity=1.0,
|
order_quantity=1.0,
|
||||||
@@ -265,7 +267,7 @@ async def set_trade_mode(tg_id: int, trade_mode: str) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_additional_settings = UserAdditionalSettings(
|
user_additional_settings = UserAdditionalSettings(
|
||||||
trade_mode=trade_mode,
|
trade_mode=trade_mode,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_additional_settings)
|
session.add(user_additional_settings)
|
||||||
|
|
||||||
@@ -304,7 +306,7 @@ async def set_margin_type(tg_id: int, margin_type: str) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_additional_settings = UserAdditionalSettings(
|
user_additional_settings = UserAdditionalSettings(
|
||||||
margin_type=margin_type,
|
margin_type=margin_type,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_additional_settings)
|
session.add(user_additional_settings)
|
||||||
|
|
||||||
@@ -343,7 +345,7 @@ async def set_switch_side(tg_id: int, switch_side: str) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_additional_settings = UserAdditionalSettings(
|
user_additional_settings = UserAdditionalSettings(
|
||||||
switch_side=switch_side,
|
switch_side=switch_side,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_additional_settings)
|
session.add(user_additional_settings)
|
||||||
|
|
||||||
@@ -358,6 +360,45 @@ async def set_switch_side(tg_id: int, switch_side: str) -> bool:
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def set_side(tg_id: int, side: str) -> bool:
|
||||||
|
"""
|
||||||
|
Set side for a user in the database.
|
||||||
|
:param tg_id: Telegram user ID
|
||||||
|
:param side: "BUY" or "SELL"
|
||||||
|
:return: True if successful, False otherwise
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
async with async_session() as session:
|
||||||
|
result = await session.execute(
|
||||||
|
select(User)
|
||||||
|
.options(joinedload(User.user_additional_settings))
|
||||||
|
.filter_by(tg_id=tg_id)
|
||||||
|
)
|
||||||
|
user = result.scalars().first()
|
||||||
|
|
||||||
|
if user:
|
||||||
|
if user.user_additional_settings:
|
||||||
|
# Updating existing record
|
||||||
|
user.user_additional_settings.side = side
|
||||||
|
else:
|
||||||
|
# Creating new record
|
||||||
|
user_additional_settings = UserAdditionalSettings(
|
||||||
|
side=side,
|
||||||
|
user_id=user.id,
|
||||||
|
)
|
||||||
|
session.add(user_additional_settings)
|
||||||
|
|
||||||
|
await session.commit()
|
||||||
|
logger.info("User side updated for user: %s", tg_id)
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
logger.error("User not found with tg_id: %s", tg_id)
|
||||||
|
return False
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error adding/updating user side for user %s: %s", tg_id, e)
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
async def set_leverage(tg_id: int, leverage: str) -> bool:
|
async def set_leverage(tg_id: int, leverage: str) -> bool:
|
||||||
"""
|
"""
|
||||||
Set leverage for a user in the database.
|
Set leverage for a user in the database.
|
||||||
@@ -382,7 +423,7 @@ async def set_leverage(tg_id: int, leverage: str) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_additional_settings = UserAdditionalSettings(
|
user_additional_settings = UserAdditionalSettings(
|
||||||
leverage=leverage,
|
leverage=leverage,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_additional_settings)
|
session.add(user_additional_settings)
|
||||||
|
|
||||||
@@ -421,7 +462,7 @@ async def set_order_quantity(tg_id: int, order_quantity: float) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_additional_settings = UserAdditionalSettings(
|
user_additional_settings = UserAdditionalSettings(
|
||||||
order_quantity=order_quantity,
|
order_quantity=order_quantity,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_additional_settings)
|
session.add(user_additional_settings)
|
||||||
|
|
||||||
@@ -462,7 +503,7 @@ async def set_martingale_factor(tg_id: int, martingale_factor: float) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_additional_settings = UserAdditionalSettings(
|
user_additional_settings = UserAdditionalSettings(
|
||||||
martingale_factor=martingale_factor,
|
martingale_factor=martingale_factor,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_additional_settings)
|
session.add(user_additional_settings)
|
||||||
|
|
||||||
@@ -505,7 +546,7 @@ async def set_max_bets_in_series(tg_id: int, max_bets_in_series: int) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_additional_settings = UserAdditionalSettings(
|
user_additional_settings = UserAdditionalSettings(
|
||||||
max_bets_in_series=max_bets_in_series,
|
max_bets_in_series=max_bets_in_series,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_additional_settings)
|
session.add(user_additional_settings)
|
||||||
|
|
||||||
@@ -546,7 +587,7 @@ async def set_trigger_price(tg_id: int, trigger_price: float) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_additional_settings = UserAdditionalSettings(
|
user_additional_settings = UserAdditionalSettings(
|
||||||
trigger_price=trigger_price,
|
trigger_price=trigger_price,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_additional_settings)
|
session.add(user_additional_settings)
|
||||||
|
|
||||||
@@ -586,10 +627,11 @@ async def create_user_risk_management(tg_id: int) -> None:
|
|||||||
|
|
||||||
# Create the user risk management
|
# Create the user risk management
|
||||||
user_risk_management = UserRiskManagement(
|
user_risk_management = UserRiskManagement(
|
||||||
user=user,
|
user_id=user.id,
|
||||||
take_profit_percent=1.0,
|
take_profit_percent=1.0,
|
||||||
stop_loss_percent=1.0,
|
stop_loss_percent=1.0,
|
||||||
commission_fee="Yes_commission_fee",
|
commission_fee="Yes_commission_fee",
|
||||||
|
commission_place="Commission_for_qty"
|
||||||
)
|
)
|
||||||
session.add(user_risk_management)
|
session.add(user_risk_management)
|
||||||
await session.commit()
|
await session.commit()
|
||||||
@@ -650,7 +692,7 @@ async def set_take_profit_percent(tg_id: int, take_profit_percent: float) -> boo
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_risk_management = UserRiskManagement(
|
user_risk_management = UserRiskManagement(
|
||||||
take_profit_percent=take_profit_percent,
|
take_profit_percent=take_profit_percent,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_risk_management)
|
session.add(user_risk_management)
|
||||||
|
|
||||||
@@ -691,7 +733,7 @@ async def set_stop_loss_percent(tg_id: int, stop_loss_percent: float) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_risk_management = UserRiskManagement(
|
user_risk_management = UserRiskManagement(
|
||||||
stop_loss_percent=stop_loss_percent,
|
stop_loss_percent=stop_loss_percent,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_risk_management)
|
session.add(user_risk_management)
|
||||||
|
|
||||||
@@ -732,7 +774,7 @@ async def set_commission_fee(tg_id: int, commission_fee: str) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_risk_management = UserRiskManagement(
|
user_risk_management = UserRiskManagement(
|
||||||
commission_fee=commission_fee,
|
commission_fee=commission_fee,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_risk_management)
|
session.add(user_risk_management)
|
||||||
|
|
||||||
@@ -749,6 +791,47 @@ async def set_commission_fee(tg_id: int, commission_fee: str) -> bool:
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def set_commission_place(tg_id: int, commission_place: str) -> bool:
|
||||||
|
"""
|
||||||
|
Set commission place for a user in the database.
|
||||||
|
:param tg_id: Telegram user ID
|
||||||
|
:param commission_place: Commission place
|
||||||
|
:return: True if successful, False otherwise
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
async with async_session() as session:
|
||||||
|
result = await session.execute(
|
||||||
|
select(User)
|
||||||
|
.options(joinedload(User.user_risk_management))
|
||||||
|
.filter_by(tg_id=tg_id)
|
||||||
|
)
|
||||||
|
user = result.scalars().first()
|
||||||
|
|
||||||
|
if user:
|
||||||
|
if user.user_risk_management:
|
||||||
|
# Updating existing record
|
||||||
|
user.user_risk_management.commission_place = commission_place
|
||||||
|
else:
|
||||||
|
# Creating new record
|
||||||
|
user_risk_management = UserRiskManagement(
|
||||||
|
commission_place=commission_place,
|
||||||
|
user_id=user.id,
|
||||||
|
)
|
||||||
|
session.add(user_risk_management)
|
||||||
|
|
||||||
|
await session.commit()
|
||||||
|
logger.info("User commission place updated for user: %s", tg_id)
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
logger.error("User not found with tg_id: %s", tg_id)
|
||||||
|
return False
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(
|
||||||
|
"Error adding/updating user commission place for user %s: %s", tg_id, e
|
||||||
|
)
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
# USER CONDITIONAL SETTINGS
|
# USER CONDITIONAL SETTINGS
|
||||||
|
|
||||||
|
|
||||||
@@ -772,7 +855,7 @@ async def create_user_conditional_settings(tg_id: int) -> None:
|
|||||||
|
|
||||||
# Create the user conditional settings
|
# Create the user conditional settings
|
||||||
user_conditional_settings = UserConditionalSettings(
|
user_conditional_settings = UserConditionalSettings(
|
||||||
user=user,
|
user_id=user.id,
|
||||||
timer_start=0,
|
timer_start=0,
|
||||||
timer_end=0,
|
timer_end=0,
|
||||||
)
|
)
|
||||||
@@ -837,7 +920,7 @@ async def set_start_timer(tg_id: int, timer_start: int) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_conditional_settings = UserConditionalSettings(
|
user_conditional_settings = UserConditionalSettings(
|
||||||
timer_start=timer_start,
|
timer_start=timer_start,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_conditional_settings)
|
session.add(user_conditional_settings)
|
||||||
|
|
||||||
@@ -876,7 +959,7 @@ async def set_stop_timer(tg_id: int, timer_end: int) -> bool:
|
|||||||
# Creating new record
|
# Creating new record
|
||||||
user_conditional_settings = UserConditionalSettings(
|
user_conditional_settings = UserConditionalSettings(
|
||||||
timer_end=timer_end,
|
timer_end=timer_end,
|
||||||
user=user,
|
user_id=user.id,
|
||||||
)
|
)
|
||||||
session.add(user_conditional_settings)
|
session.add(user_conditional_settings)
|
||||||
|
|
||||||
@@ -896,6 +979,7 @@ async def set_user_deal(
|
|||||||
tg_id: int,
|
tg_id: int,
|
||||||
symbol: str,
|
symbol: str,
|
||||||
current_step: int,
|
current_step: int,
|
||||||
|
current_series: int,
|
||||||
trade_mode: str,
|
trade_mode: str,
|
||||||
side_mode: str,
|
side_mode: str,
|
||||||
margin_type: str,
|
margin_type: str,
|
||||||
@@ -906,13 +990,17 @@ async def set_user_deal(
|
|||||||
max_bets_in_series: int,
|
max_bets_in_series: int,
|
||||||
take_profit_percent: int,
|
take_profit_percent: int,
|
||||||
stop_loss_percent: int,
|
stop_loss_percent: int,
|
||||||
base_quantity: float
|
base_quantity: float,
|
||||||
|
commission_fee: str,
|
||||||
|
commission_place: str,
|
||||||
|
pnl_series: float
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
Set the user deal in the database.
|
Set the user deal in the database.
|
||||||
:param tg_id: Telegram user ID
|
:param tg_id: Telegram user ID
|
||||||
:param symbol: Symbol
|
:param symbol: Symbol
|
||||||
:param current_step: Current step
|
:param current_step: Current step
|
||||||
|
:param current_series: Current series
|
||||||
:param trade_mode: Trade mode
|
:param trade_mode: Trade mode
|
||||||
:param side_mode: Side mode
|
:param side_mode: Side mode
|
||||||
:param margin_type: Margin type
|
:param margin_type: Margin type
|
||||||
@@ -924,6 +1012,9 @@ async def set_user_deal(
|
|||||||
:param take_profit_percent: Take profit percent
|
:param take_profit_percent: Take profit percent
|
||||||
:param stop_loss_percent: Stop loss percent
|
:param stop_loss_percent: Stop loss percent
|
||||||
:param base_quantity: Base quantity
|
:param base_quantity: Base quantity
|
||||||
|
:param commission_fee: Commission fee
|
||||||
|
:param commission_place: Commission place
|
||||||
|
:param pnl_series: PNL series
|
||||||
:return: bool
|
:return: bool
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
@@ -942,6 +1033,7 @@ async def set_user_deal(
|
|||||||
if deal:
|
if deal:
|
||||||
# Updating existing record
|
# Updating existing record
|
||||||
deal.current_step = current_step
|
deal.current_step = current_step
|
||||||
|
deal.current_series = current_series
|
||||||
deal.trade_mode = trade_mode
|
deal.trade_mode = trade_mode
|
||||||
deal.side_mode = side_mode
|
deal.side_mode = side_mode
|
||||||
deal.margin_type = margin_type
|
deal.margin_type = margin_type
|
||||||
@@ -953,12 +1045,16 @@ async def set_user_deal(
|
|||||||
deal.take_profit_percent = take_profit_percent
|
deal.take_profit_percent = take_profit_percent
|
||||||
deal.stop_loss_percent = stop_loss_percent
|
deal.stop_loss_percent = stop_loss_percent
|
||||||
deal.base_quantity = base_quantity
|
deal.base_quantity = base_quantity
|
||||||
|
deal.commission_fee = commission_fee
|
||||||
|
deal.commission_place = commission_place
|
||||||
|
deal.pnl_series = pnl_series
|
||||||
else:
|
else:
|
||||||
# Creating new record
|
# Creating new record
|
||||||
new_deal = UserDeals(
|
new_deal = UserDeals(
|
||||||
user=user,
|
user_id=user.id,
|
||||||
symbol=symbol,
|
symbol=symbol,
|
||||||
current_step=current_step,
|
current_step=current_step,
|
||||||
|
current_series=current_series,
|
||||||
trade_mode=trade_mode,
|
trade_mode=trade_mode,
|
||||||
side_mode=side_mode,
|
side_mode=side_mode,
|
||||||
margin_type=margin_type,
|
margin_type=margin_type,
|
||||||
@@ -969,7 +1065,10 @@ async def set_user_deal(
|
|||||||
max_bets_in_series=max_bets_in_series,
|
max_bets_in_series=max_bets_in_series,
|
||||||
take_profit_percent=take_profit_percent,
|
take_profit_percent=take_profit_percent,
|
||||||
stop_loss_percent=stop_loss_percent,
|
stop_loss_percent=stop_loss_percent,
|
||||||
base_quantity=base_quantity
|
base_quantity=base_quantity,
|
||||||
|
commission_fee=commission_fee,
|
||||||
|
commission_place=commission_place,
|
||||||
|
pnl_series=pnl_series
|
||||||
)
|
)
|
||||||
session.add(new_deal)
|
session.add(new_deal)
|
||||||
|
|
||||||
@@ -1078,6 +1177,91 @@ async def set_last_side_by_symbol(tg_id: int, symbol: str, last_side: str):
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def set_current_series(tg_id: int, symbol: str, current_series: int):
|
||||||
|
"""Set current series for a user deal by symbol in the database."""
|
||||||
|
try:
|
||||||
|
async with async_session() as session:
|
||||||
|
result = await session.execute(select(User).filter_by(tg_id=tg_id))
|
||||||
|
user = result.scalars().first()
|
||||||
|
if user is None:
|
||||||
|
logger.error(f"User with tg_id={tg_id} not found")
|
||||||
|
return False
|
||||||
|
|
||||||
|
result = await session.execute(
|
||||||
|
select(UserDeals).filter_by(user_id=user.id, symbol=symbol)
|
||||||
|
)
|
||||||
|
record = result.scalars().first()
|
||||||
|
|
||||||
|
if record:
|
||||||
|
record.current_series = current_series
|
||||||
|
else:
|
||||||
|
logger.error(f"User deal with user_id={user.id} and symbol={symbol} not found")
|
||||||
|
return False
|
||||||
|
await session.commit()
|
||||||
|
logger.info("Set current series for user %s and symbol %s", tg_id, symbol)
|
||||||
|
return True
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error setting user deal current series for user %s and symbol %s: %s", tg_id, symbol, e)
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def set_tp_sl_by_symbol(tg_id: int, symbol: str, tp: float, sl: float):
|
||||||
|
"""Set tp and sl for a user deal by symbol in the database."""
|
||||||
|
try:
|
||||||
|
async with async_session() as session:
|
||||||
|
result = await session.execute(select(User).filter_by(tg_id=tg_id))
|
||||||
|
user = result.scalars().first()
|
||||||
|
if user is None:
|
||||||
|
logger.error(f"User with tg_id={tg_id} not found")
|
||||||
|
return False
|
||||||
|
|
||||||
|
result = await session.execute(
|
||||||
|
select(UserDeals).filter_by(user_id=user.id, symbol=symbol)
|
||||||
|
)
|
||||||
|
record = result.scalars().first()
|
||||||
|
|
||||||
|
if record:
|
||||||
|
record.take_profit = tp
|
||||||
|
record.stop_loss = sl
|
||||||
|
else:
|
||||||
|
logger.error(f"User deal with user_id={user.id} and symbol={symbol} not found")
|
||||||
|
return False
|
||||||
|
await session.commit()
|
||||||
|
logger.info("Set tp and sl for user %s and symbol %s", tg_id, symbol)
|
||||||
|
return True
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error setting user deal tp and sl for user %s and symbol %s: %s", tg_id, symbol, e)
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def set_pnl_series_by_symbol(tg_id: int, symbol: str, pnl_series: float):
|
||||||
|
"""Set pnl series for a user deal by symbol in the database."""
|
||||||
|
try:
|
||||||
|
async with async_session() as session:
|
||||||
|
result = await session.execute(select(User).filter_by(tg_id=tg_id))
|
||||||
|
user = result.scalars().first()
|
||||||
|
if user is None:
|
||||||
|
logger.error(f"User with tg_id={tg_id} not found")
|
||||||
|
return False
|
||||||
|
|
||||||
|
result = await session.execute(
|
||||||
|
select(UserDeals).filter_by(user_id=user.id, symbol=symbol)
|
||||||
|
)
|
||||||
|
record = result.scalars().first()
|
||||||
|
|
||||||
|
if record:
|
||||||
|
record.pnl_series = pnl_series
|
||||||
|
else:
|
||||||
|
logger.error(f"User deal with user_id={user.id} and symbol={symbol} not found")
|
||||||
|
return False
|
||||||
|
await session.commit()
|
||||||
|
logger.info("Set pnl series for user %s and symbol %s", tg_id, symbol)
|
||||||
|
return True
|
||||||
|
except Exception as e:
|
||||||
|
logger.error("Error setting user deal pnl series for user %s and symbol %s: %s", tg_id, symbol, e)
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
# USER AUTO TRADING
|
# USER AUTO TRADING
|
||||||
|
|
||||||
async def get_all_user_auto_trading(tg_id: int):
|
async def get_all_user_auto_trading(tg_id: int):
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ After=syslog.target network-online.target
|
|||||||
ExecStart=sudo -u www-data /usr/bin/python3 /var/www/stcs/BybitBot_API.py
|
ExecStart=sudo -u www-data /usr/bin/python3 /var/www/stcs/BybitBot_API.py
|
||||||
PIDFile=/var/run/python/stcs.pid
|
PIDFile=/var/run/python/stcs.pid
|
||||||
RemainAfterExit=no
|
RemainAfterExit=no
|
||||||
RuntimeMaxSec=3600s
|
RuntimeMaxSec=604800s
|
||||||
Restart=always
|
Restart=always
|
||||||
RestartSec=5s
|
RestartSec=5s
|
||||||
|
|
||||||
|
|||||||
1
run.py
1
run.py
@@ -31,7 +31,6 @@ async def main():
|
|||||||
dp = Dispatcher(storage=storage)
|
dp = Dispatcher(storage=storage)
|
||||||
dp.include_router(router)
|
dp.include_router(router)
|
||||||
web_socket = WebSocketBot(telegram_bot=bot)
|
web_socket = WebSocketBot(telegram_bot=bot)
|
||||||
await web_socket.clear_user_sockets()
|
|
||||||
ws_task = asyncio.create_task(web_socket.run_user_check_loop())
|
ws_task = asyncio.create_task(web_socket.run_user_check_loop())
|
||||||
tg_task = asyncio.create_task(dp.start_polling(bot))
|
tg_task = asyncio.create_task(dp.start_polling(bot))
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user