def confirm_sub(mes: types.Message):
if mes.text == 'Подтвердить ✅':
price = db_acces.get_price_for_new_channel(mes.from_user.id)
count = db_acces.get_sub_for_new_channel(mes.from_user.id)
end_price = price * count
balance = db_acces.get_user_balance(mes.from_user.id)
if (balance - end_price) >= 0:
db_acces.set_user_balance(mes.from_user.id, balance - end_price)
db_acces.publish_channel_order(mes.from_user.id)
bot.send_message(mes.from_user.id,
'Ваш заказ успешно оформлен!',
reply_markup=get_main_menu_markup())
db_acces.set_user_state(mes.from_user.id, states.CHOOSE_OPTION)
else:
bot.send_message(
mes.from_user.id,
'Недостаточно средств! Пополните свой баланс и повторите заказ снова',
reply_markup=get_main_menu_markup())
db_acces.delete_post_order(mes.from_user.id)
db_acces.set_user_state(mes.from_user.id, states.CHOOSE_OPTION)
elif mes.text == 'Отмена':
bot.send_message(mes.from_user.id,
'Отменено',
reply_markup=get_main_menu_markup())
db_acces.delete_post_order(mes.from_user.id)
db_acces.set_user_state(mes.from_user.id, states.CHOOSE_OPTION)
else:
bot.send_message(mes.from_user.id,
'Какой-то странный ответ. Попробуйте еще раз')
def remove_sum_from_balance(mes: types.Message):
if mes.text == 'Отмена':
db_acces.set_user_state(mes.from_user.id, states.CHOOSE_OPTION)
bot.send_message(mes.from_user.id, 'Отменено', reply_markup=get_admin_markup())
return
try:
user_id, money = map(int, str(mes.text).split(' - '))
except ValueError:
bot.send_message(mes.from_user.id, 'Неверный формат, попробуйте снова')
else:
balance = db_acces.get_user_balance(user_id)
if balance is None:
bot.send_message(mes.from_user.id, 'Такого счета не существует', reply_markup=get_admin_markup())
db_acces.set_user_state(mes.from_user.id, states.CHOOSE_OPTION)
return
if (balance - money) <= 0:
result = db_acces.set_user_balance(user_id, 0)
else:
result = db_acces.set_user_balance(user_id, balance - money)
db_acces.set_user_state(mes.from_user.id, states.CHOOSE_OPTION)
if result:
gave_money = int(config['stat']['gave_money']) + money
config.set('stat', 'gave_money', str(gave_money))
save_data()
bot.send_message(user_id, 'C вашего балансв снято {}р.'.format(str(money)))
bot.send_message(mes.from_user.id, 'Баланс пользователя успешно уменьшен', reply_markup=get_admin_markup())
else:
bot.send_message(mes.from_user.id, 'Ошибка уменьшения баланса!', reply_markup=get_admin_markup())