Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def firewall(e):
- sql = DbProcedures()
- try:
- sq_user = DbProcedures(user_id=e.from_id)
- if sq_user.user_info.get('banned') == 1:
- bot.send_message(text="Вы забанены администратором u1", peer_id=e.peer_id)
- return
- except: pass
- if sql.config['total_connections'] >= sql.config['max_connections']:
- bot.send_message(
- text=f"Файрволл: достигнуто число максимального количества подключений к базе данных. Сейчас: {sql.config['total_connections']}. Максимум: {sql.config['max_connections']}",
- peer_id=e.peer_id
- )
- raise Exception("Слишком много подключений")
- else:
- sql.set_config(column="total_connections", value=sql.config['total_connections'] + 1)
- bot.before_function = firewall
- def unset_connection(e):
- sql = DbProcedures()
- sql.set_config(column="total_connections", value=sql.config['total_connections'] - 1)
- bot.after_function = unset_connection
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement