From 8fc70c267daeda6bc1c0d504d720d9b33d72056b Mon Sep 17 00:00:00 2001 From: Pavel_Durov Date: Mon, 17 Jul 2023 23:24:10 +0300 Subject: [PATCH] now commands work onnly in groups except for adms --- handlers/pipisa.py | 310 +++++++++++++++++++++++---------------------- 1 file changed, 160 insertions(+), 150 deletions(-) diff --git a/handlers/pipisa.py b/handlers/pipisa.py index 03ddfda..30d1706 100644 --- a/handlers/pipisa.py +++ b/handlers/pipisa.py @@ -11,187 +11,196 @@ from pipisa_functions import pipisa_time #### https://docs.python-arango.com/en/main/ from arango import ArangoClient - + +admins = [317051301, 226765676] @dp.message_handler(commands=["dick"]) async def up_dick(message: types.Message): #global size_change, dick_plus, dick_minus + if message.from_user.id in admins or message.chat.type != 'private': + + + arango_client = ArangoClient(hosts='https://arango.guaranteedstruggle.host') + pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527') + dicks_collection = pipisa_db.collection('dicks') + + + # рандомайзер + numb = randint(-9,10) + dick_plus = ( + 'вырос', + 'подрос', + 'привстал', + 'увеличился', + 'преисполнился' + ) + dick_minus = ( + 'неудачно потусил на сходке', + 'уменьшился', + 'откусила злая гадюка', + 'продали евреи, он стал дешевле', + 'схуднул' + ) + dick_stayed = ( + 'продвинулся', + 'наноуменьшился', + 'зафиксировал прибыль' + ) + + if numb > 0: + size_change = random.choice(dick_plus) + '🚀' + elif numb == 0: + size_change = random.choice(dick_stayed) + '🤨' + else: + size_change = random.choice(dick_minus) + '✂' + + + try: + #### Чекнуть есть ли юзер в базе + candidate_cursor = dicks_collection.find( + { + 'user_id': message.from_user.id, + 'chat_id': message.chat.id + }, + skip=0, limit=1488) - arango_client = ArangoClient(hosts='https://arango.guaranteedstruggle.host') - pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527') - dicks_collection = pipisa_db.collection('dicks') + if candidate_cursor.count() > 0: + ## Если есть то нарастить и отправить сообщение + + user = candidate_cursor.pop() + + + if pipisa_time.rolltime(check_datetime=user['datetimes'], curr_time=datetime.datetime.now()): + # пришло время крутить!! + + updetedDick = user["dick_size"] + numb + try: + dt = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") + 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) + + randomword = ForReadDict.RandomDick() - # рандомайзер - numb = randint(-9,10) - dick_plus = ( - 'вырос', - 'подрос', - 'привстал', - 'увеличился', - 'преисполнился' - ) - dick_minus = ( - 'неудачно потусил на сходке', - 'уменьшился', - 'откусила злая гадюка', - 'продали евреи, он стал дешевле', - 'схуднул' - ) - dick_stayed = ( - 'продвинулся', - 'наноуменьшился', - 'зафиксировал прибыль' - ) - - if numb > 0: - size_change = random.choice(dick_plus) + '🚀' - elif numb == 0: - size_change = random.choice(dick_stayed) + '🤨' - else: - size_change = random.choice(dick_minus) + '✂' - - - 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() - - - if pipisa_time.rolltime(check_datetime=user['datetimes'], curr_time=datetime.datetime.now()): - # пришло время крутить!! - - updetedDick = user["dick_size"] + numb - try: - dt = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") - 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) - - randomword = ForReadDict.RandomDick() - - - await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш {randomword} {size_change}\ + await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш {randomword} {size_change}\ на {abs(numb)} см!\n\ Теперь он равен {updetedDick} см! ') - else: - await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер {user["dick_size"]} см! ') + else: + await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер {user["dick_size"]} см! ') + + 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) + + print(f'пользователь @{message.from_user.username} уже крутил') + + else: + ## если нету, то создать + + try: + novenkiy = { + 'user_id': message.from_user.id, + 'user_fullname': message.from_user.full_name, + 'dick_size': numb, + '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') - 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) - print(f'пользователь @{message.from_user.username} уже крутил') - - else: - ## если нету, то создать - - try: - novenkiy = { - 'user_id': message.from_user.id, - 'user_fullname': message.from_user.full_name, - 'dick_size': numb, - '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') - - - if numb > 0: - await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн показал головку 🚀\n\ + if numb > 0: + await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн показал головку 🚀\n\ на {abs(numb)} см!\n\ Теперь он равен {numb} см!') - else: - await bot.send_message(message.chat.id, f'@{message.from_user.username}, добро пожаловать в игру!🚀\n\ + else: + await bot.send_message(message.chat.id, f'@{message.from_user.username}, добро пожаловать в игру!🚀\n\ Ты стартуешь не с лучшей позиции, природа наградила тебя всего {numb} см 😬') - - - print(f'Успешно добавлен пользователь @{message.from_user.username}') - except Exception as e1: - print('ошибка в добавлении нового пользователя') - print(e1) + + + 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) + else: + await message.reply('Растить елду можно только в общих чатах!') + @dp.message_handler(commands=["topdick"]) async def send_topchat(message: types.Message): + if message.from_user.id in admins or message.chat.type != 'private': + arango_client = ArangoClient(hosts='https://arango.guaranteedstruggle.host') + pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527') + dicks_collection = pipisa_db.collection('dicks') - arango_client = ArangoClient(hosts='https://arango.guaranteedstruggle.host') - pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527') - dicks_collection = pipisa_db.collection('dicks') + try: + dicks = dicks_collection.find({'chat_id': message.chat.id}, skip=0, limit=1100) + dicks = [d for d in dicks] + except Exception as e: + print('ошибка DB в /topdick') + print(e) - try: - dicks = dicks_collection.find({'chat_id': message.chat.id}, skip=0, limit=1100) - dicks = [d for d in dicks] - except Exception as e: - print('ошибка DB в /topdick') - print(e) - - top_dicks = sorted(dicks, key=lambda dick: dick['dick_size'], reverse=True) + top_dicks = sorted(dicks, key=lambda dick: dick['dick_size'], reverse=True) + + dick_limit = min(len(top_dicks), 10) + top_dicks = top_dicks[:dick_limit] - dick_limit = min(len(top_dicks), 10) - top_dicks = top_dicks[:dick_limit] - - dickos = '' - emo = ['🏆','🚀','🍆','🍌','🐍','🐎','🌭','🌶','👌','💩'] + dickos = '' + emo = ['🏆','🚀','🍆','🍌','🐍','🐎','🌭','🌶','👌','💩'] - if len(top_dicks) == 0: - await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌') + if len(top_dicks) == 0: + await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌') + else: + for i in range(len(top_dicks)): + dickos += f' {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n' + i += 1 + + await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos) else: - for i in range(len(top_dicks)): - dickos += f' {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n' - i += 1 + await message.reply('Работает только в общих чатах!\n'\ + 'Вы мсжете посмотреть топ по миру /globaldick') - await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos) - @dp.message_handler(commands=["globaldick"]) @@ -213,7 +222,7 @@ async def send_global_top(message: types.Message): dick_limit = min(len(top_dicks), 10) top_dicks = top_dicks[:dick_limit] - + dickos = '' emo = ['🏆','🚀','💫','🍆','🍌','🐍','🐎','🌭','🌶','👌'] @@ -226,3 +235,4 @@ async def send_global_top(message: types.Message): await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + dickos) +