Advertisement
Guest User

Untitled

a guest
Mar 24th, 2019
69
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.00 KB | None | 0 0
  1. def firewall(e):
  2.     sql = DbProcedures()
  3.     try:
  4.         sq_user = DbProcedures(user_id=e.from_id)
  5.         if sq_user.user_info.get('banned') == 1:
  6.             bot.send_message(text="Вы забанены администратором u1", peer_id=e.peer_id)
  7.             return
  8.     except: pass
  9.  
  10.     if sql.config['total_connections'] >= sql.config['max_connections']:
  11.         bot.send_message(
  12.             text=f"Файрволл: достигнуто число максимального количества подключений к базе данных. Сейчас: {sql.config['total_connections']}. Максимум: {sql.config['max_connections']}",
  13.             peer_id=e.peer_id
  14.         )
  15.         raise Exception("Слишком много подключений")
  16.     else:
  17.         sql.set_config(column="total_connections", value=sql.config['total_connections'] + 1)
  18.  
  19. bot.before_function = firewall
  20.  
  21. def unset_connection(e):
  22.     sql = DbProcedures()
  23.     sql.set_config(column="total_connections", value=sql.config['total_connections'] - 1)
  24.  
  25. bot.after_function = unset_connection
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement