forked from Pavel_Durov/python_bot
86 lines
3.4 KiB
Python
86 lines
3.4 KiB
Python
from db_logic import collections
|
|
import datetime, logging
|
|
|
|
|
|
def store_new_user(message, result ):
|
|
|
|
try:
|
|
novenkiy = {
|
|
'user_id': message.from_user.id,
|
|
'user_fullname': message.from_user.full_name,
|
|
'dick_size': abs(result),
|
|
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
|
'attempts': 1,
|
|
'chat_id': message.chat.id
|
|
}
|
|
metadata = collections.get_dicks_collection().insert(novenkiy, overwrite_mode='update')
|
|
|
|
logging.debug(f'Успешно добавлен нового пользователь @{message.from_user.username}')
|
|
except Exception as e2:
|
|
logging.error(f'ошибка DB в /dick :: добавление нового пользователя @{message.from_user.username}')
|
|
logging.error(e2)
|
|
|
|
|
|
def update_attempts(message, user ):
|
|
try:
|
|
metadata = collections.get_dicks_collection().insert(
|
|
{
|
|
'_key': user['_key'], ### этот нужен чтобы апдейт прилетал тому же юзеру
|
|
'user_id': user['user_id'] ,
|
|
'user_fullname': message.from_user.full_name,
|
|
'dick_size': user['dick_size'] ,
|
|
'datetimes': user['datetimes'] ,
|
|
'attempts': user['attempts'] + 1,
|
|
'chat_id': message.chat.id
|
|
},
|
|
overwrite_mode='update'
|
|
)
|
|
|
|
logging.debug(f'Успешно апдейтнули попытку крутить пипису @{message.from_user.username}')
|
|
except Exception as e2:
|
|
logging.error(f'ошибка DB в /dick :: обновление попытки крутить пипису @{message.from_user.username}')
|
|
logging.error(e2)
|
|
|
|
|
|
def update_dick_size(message, user, updatedDick ):
|
|
|
|
try:
|
|
metadata = collections.get_dicks_collection().insert(
|
|
{
|
|
'_key': user['_key'], ### этот нужен чтобы апдейт прилетал тому же юзеру
|
|
'user_id': user['user_id'],
|
|
'user_fullname': message.from_user.full_name,
|
|
'dick_size': updatedDick,
|
|
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
|
'attempts': user['attempts'] + 1,
|
|
'chat_id': message.chat.id
|
|
},
|
|
overwrite_mode='update'
|
|
)
|
|
|
|
logging.info(f'Успешно апдейтнули пипису @{message.from_user.username}')
|
|
except Exception as e2:
|
|
logging.error(f'ошибка DB в /dick :: обновление пиписы @{message.from_user.username}')
|
|
logging.error(e2)
|
|
|
|
|
|
def get_user(message):
|
|
try:
|
|
#### Чекнуть есть ли юзер в базе
|
|
candidate_cursor = collections.get_dicks_collection().find(
|
|
{
|
|
'user_id': message.from_user.id,
|
|
'chat_id': message.chat.id
|
|
},
|
|
skip = 0,
|
|
limit = 1488
|
|
)
|
|
if candidate_cursor.count() > 0:
|
|
user = candidate_cursor.pop()
|
|
else:
|
|
user = None
|
|
|
|
return user
|
|
except Exception as e:
|
|
logging.error('ошибка DB в /dick :: поиск юзера')
|
|
logging.error(e) |