forked from kodorvan/stcs
Added user verification
This commit is contained in:
@@ -85,7 +85,19 @@ async def cmd_to_main(message: Message, state: FSMContext) -> None:
|
||||
None: Exceptions are caught and logged internally.
|
||||
"""
|
||||
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(
|
||||
"Command to_profile_tg processed successfully for user: %s",
|
||||
message.from_user.id,
|
||||
@@ -117,9 +129,21 @@ async def profile_bybit(message: Message, state: FSMContext) -> None:
|
||||
"""
|
||||
try:
|
||||
await state.clear()
|
||||
await user_profile_bybit(
|
||||
tg_id=message.from_user.id, message=message, state=state
|
||||
)
|
||||
user = await rq.get_user(tg_id=message.from_user.id)
|
||||
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(
|
||||
"Command to_profile_bybit processed successfully for user: %s",
|
||||
message.from_user.id,
|
||||
@@ -150,15 +174,31 @@ async def profile_bybit_callback(
|
||||
"""
|
||||
try:
|
||||
await state.clear()
|
||||
await user_profile_bybit(
|
||||
tg_id=callback_query.from_user.id,
|
||||
message=callback_query.message,
|
||||
state=state,
|
||||
)
|
||||
logger.debug(
|
||||
"Callback profile_bybit processed successfully for user: %s",
|
||||
callback_query.from_user.id,
|
||||
)
|
||||
user = await rq.get_user(tg_id=callback_query.from_user.id)
|
||||
|
||||
if user:
|
||||
await user_profile_bybit(
|
||||
tg_id=callback_query.from_user.id,
|
||||
message=callback_query.message,
|
||||
state=state,
|
||||
)
|
||||
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()
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
|
||||
Reference in New Issue
Block a user