forked from kodorvan/stcs
102 lines
3.1 KiB
Python
102 lines
3.1 KiB
Python
import logging.config
|
|
|
|
from app.bybit import get_bybit_client
|
|
from app.bybit.get_functions.get_positions import get_active_positions_by_symbol
|
|
from app.bybit.logger_bybit.logger_bybit import LOGGING_CONFIG
|
|
|
|
logging.config.dictConfig(LOGGING_CONFIG)
|
|
logger = logging.getLogger("close_positions")
|
|
|
|
|
|
async def close_position(tg_id: int, symbol: str) -> bool:
|
|
"""
|
|
Closes position
|
|
:param tg_id: Telegram user ID
|
|
:param symbol: symbol
|
|
:return: bool
|
|
"""
|
|
try:
|
|
client = await get_bybit_client(tg_id)
|
|
active_positions = await get_active_positions_by_symbol(tg_id, symbol)
|
|
side = active_positions.get("side")
|
|
size = active_positions.get("size")
|
|
position_idx = active_positions.get("positionIdx")
|
|
|
|
if side == "Buy":
|
|
side = "Sell"
|
|
elif side == "Sell":
|
|
side = "Buy"
|
|
|
|
response = client.place_order(
|
|
category="linear",
|
|
symbol=symbol,
|
|
side=side,
|
|
orderType="Market",
|
|
qty=size,
|
|
timeInForce="GTC",
|
|
positionIdx=position_idx,
|
|
)
|
|
if response["retCode"] == 0:
|
|
logger.info("Position closed for %s for user %s", symbol, tg_id)
|
|
return True
|
|
else:
|
|
logger.error("Error closing position for %s for user %s", symbol, tg_id)
|
|
return False
|
|
except Exception as e:
|
|
logger.error("Error closing position for %s for user %s: %s", symbol, tg_id, e)
|
|
return False
|
|
|
|
|
|
async def cancel_order(tg_id, symbol) -> bool:
|
|
"""
|
|
Cancel order by order id
|
|
"""
|
|
try:
|
|
client = await get_bybit_client(tg_id)
|
|
orders_resp = client.get_open_orders(category="linear", symbol=symbol)
|
|
orders = orders_resp.get("result", {}).get("list", [])
|
|
|
|
for order in orders:
|
|
order_id = order.get("orderId")
|
|
cancel_resp = client.cancel_order(
|
|
category="linear", symbol=symbol, orderId=order_id
|
|
)
|
|
|
|
if cancel_resp.get("retCode") == 0:
|
|
return True
|
|
else:
|
|
logger.error(
|
|
"Error canceling order for user %s: %s",
|
|
tg_id,
|
|
cancel_resp.get("retMsg"),
|
|
)
|
|
return False
|
|
return False
|
|
except Exception as e:
|
|
logger.error("Error canceling order for user %s: %s", tg_id, e)
|
|
return False
|
|
|
|
|
|
async def cancel_all_orders(tg_id: int) -> bool:
|
|
"""
|
|
Cancel all open orders
|
|
"""
|
|
try:
|
|
client = await get_bybit_client(tg_id)
|
|
cancel_resp = client.cancel_all_orders(category="linear", settleCoin="USDT")
|
|
|
|
if cancel_resp.get("retCode") == 0:
|
|
logger.info("All orders canceled for user %s", tg_id)
|
|
return True
|
|
else:
|
|
logger.error(
|
|
"Error canceling order for user %s: %s",
|
|
tg_id,
|
|
cancel_resp.get("retMsg"),
|
|
)
|
|
return False
|
|
|
|
except Exception as e:
|
|
logger.error("Error canceling order for user %s: %s", tg_id, e)
|
|
return False
|