forked from Pavel_Durov/python_bot
меньше кода - лучше!
This commit is contained in:
Simple_Not
parent
505d90aea8
commit
48d53fd1b9
86
db_logic/user_stuff.py
Normal file
86
db_logic/user_stuff.py
Normal file
@ -0,0 +1,86 @@
|
||||
from global_conf import CONFIG
|
||||
from db_logic import collections
|
||||
|
||||
import datetime
|
||||
|
||||
|
||||
def store_new_user(dicks_collection, message, result ):
|
||||
|
||||
try:
|
||||
novenkiy = {
|
||||
'user_id': message.from_user.id,
|
||||
'user_fullname': message.from_user.full_name,
|
||||
'dick_size': result,
|
||||
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||
'attempts': 1,
|
||||
'chat_id': message.chat.id
|
||||
}
|
||||
metadata = dicks_collection.insert(novenkiy, overwrite_mode='update')
|
||||
|
||||
print(f'Успешно добавлен нового пользователь @{message.from_user.username}')
|
||||
except Exception as e2:
|
||||
print(f'ошибка DB в /dick :: добавление нового пользователя @{message.from_user.username}')
|
||||
print(e2)
|
||||
|
||||
|
||||
def update_attempts(dicks_collection, message, user ):
|
||||
try:
|
||||
metadata = 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'
|
||||
)
|
||||
print(f'Успешно апдейтнули попытку крутить пипису @{message.from_user.username}')
|
||||
except Exception as e2:
|
||||
print(f'ошибка DB в /dick :: обновление попытки крутить пипису @{message.from_user.username}')
|
||||
print(e2)
|
||||
|
||||
|
||||
def update_dick_size(dicks_collection, message, user, updatedDick ):
|
||||
|
||||
try:
|
||||
metadata = 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'
|
||||
)
|
||||
print(f'Успешно апдейтнули пипису @{message.from_user.username}')
|
||||
except Exception as e2:
|
||||
print(f'ошибка DB в /dick :: обновление пиписы @{message.from_user.username}')
|
||||
print(e2)
|
||||
|
||||
|
||||
def get_user(dicks_collection, message):
|
||||
try:
|
||||
#### Чекнуть есть ли юзер в базе
|
||||
candidate_cursor = 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:
|
||||
print('ошибка DB в /dick :: поиск юзера')
|
||||
print(e)
|
||||
@ -8,8 +8,7 @@ from pipisa_functions import pipisa_time
|
||||
|
||||
from global_conf import CONFIG
|
||||
|
||||
from db_logic import tops
|
||||
from db_logic import collections
|
||||
from db_logic import collections, tops, user_stuff
|
||||
|
||||
|
||||
admins = CONFIG['telegram_admins_ids']
|
||||
@ -28,86 +27,37 @@ async def up_dick(message: types.Message):
|
||||
|
||||
# рандомайзер
|
||||
result = randint(-9,10)
|
||||
try:
|
||||
#### Чекнуть есть ли юзер в базе
|
||||
candidate_cursor = dicks_collection.find(
|
||||
{
|
||||
'user_id': message.from_user.id,
|
||||
'chat_id': message.chat.id
|
||||
},
|
||||
skip = 0,
|
||||
limit = 1488
|
||||
)
|
||||
|
||||
if candidate_cursor.count() > 0:
|
||||
#### Чекнуть есть ли юзер в базе
|
||||
user = user_stuff.get_user(dicks_collection, message)
|
||||
|
||||
if user:
|
||||
## Если есть то нарастить и отправить сообщение
|
||||
user = candidate_cursor.pop()
|
||||
|
||||
if pipisa_time.rolltime(check_datetime=user['datetimes'], curr_time=datetime.datetime.now()):
|
||||
# пришло время крутить!!
|
||||
|
||||
updetedDick = user["dick_size"] + result
|
||||
try:
|
||||
metadata = dicks_collection.insert(
|
||||
{
|
||||
'_key': user['_key'], ### этот нужен чтобы апдейт прилетал тому же юзеру
|
||||
'user_id': user['user_id'],
|
||||
'user_fullname': message.from_user.full_name,
|
||||
'dick_size': updetedDick,
|
||||
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||
'attempts': user['attempts'] + 1,
|
||||
'chat_id': message.chat.id
|
||||
},
|
||||
overwrite_mode='update'
|
||||
)
|
||||
print(f'Успешно апдейтнули пипису @{message.from_user.username}')
|
||||
except Exception as e2:
|
||||
print(f'ошибка в обновлении пиписы @{message.from_user.username}')
|
||||
print(e2)
|
||||
updatedDick = user["dick_size"] + result
|
||||
user_stuff.store_new_user(dicks_collection, message, user, updatedDick)
|
||||
|
||||
randomword = ForReadDict.RandomDick()
|
||||
change_phrase = WordsChange.ChangeWord(result)
|
||||
|
||||
await bot.send_message( message.chat.id,
|
||||
f'@{message.from_user.username}, ваш <em>{randomword}</em> {change_phrase} на <b>{abs(result)}</b> см!\nТеперь он равен <b>{updetedDick}</b> см! '
|
||||
f'@{message.from_user.username}, ваш <em>{randomword}</em> {change_phrase} на <b>{abs(result)}</b> см!\nТеперь он равен <b>{updatedDick}</b> см! '
|
||||
)
|
||||
else:
|
||||
await bot.send_message( message.chat.id,
|
||||
f'@{message.from_user.username}, 🚫 вы уже крутили пипису, ее размер <b>{user["dick_size"]}</b> см! '
|
||||
)
|
||||
|
||||
try:
|
||||
metadata = 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'
|
||||
)
|
||||
print(f'Успешно апдейтнули попытку крутить пипису @{message.from_user.username}')
|
||||
except Exception as e2:
|
||||
print(f'ошибка в обновлении попытки крутить пипису @{message.from_user.username}')
|
||||
print(e2)
|
||||
user_stuff.update_attempts(dicks_collection, message, user)
|
||||
|
||||
print(f'пользователь @{message.from_user.username} уже крутил')
|
||||
|
||||
else:
|
||||
## если нету, то создать
|
||||
|
||||
try:
|
||||
novenkiy = {
|
||||
'user_id': message.from_user.id,
|
||||
'user_fullname': message.from_user.full_name,
|
||||
'dick_size': result,
|
||||
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||
'attempts': 1,
|
||||
'chat_id': message.chat.id
|
||||
}
|
||||
metadata = dicks_collection.insert(novenkiy, overwrite_mode='update')
|
||||
user_stuff.store_new_user(dicks_collection, message, result)
|
||||
|
||||
if result > 0:
|
||||
await bot.send_message( message.chat.id,
|
||||
@ -118,14 +68,10 @@ async def up_dick(message: types.Message):
|
||||
f'@{message.from_user.username}, добро пожаловать в игру!🚀\nТы стартуешь не с лучшей позиции, природа наградила тебя всего <b>{result}</b> см 😬'
|
||||
)
|
||||
|
||||
print(f'Успешно добавлен пользователь @{message.from_user.username}')
|
||||
except Exception as e1:
|
||||
print('ошибка в добавлении нового пользователя')
|
||||
print(e1)
|
||||
|
||||
except Exception as e:
|
||||
print('ошибка DB в /dick')
|
||||
print(e)
|
||||
# except Exception as e:
|
||||
# print('ошибка DB в /dick')
|
||||
# print(e)
|
||||
|
||||
#### для всех отложенных сообщений
|
||||
## проверить если прошло ?пять минут? с публикации
|
||||
|
||||
Loading…
Reference in New Issue
Block a user