Merge branch 'main' of https://git.guaranteedstruggle.host/Pavel_Durov/python_bot

handlers/__init__.py

@@ -1,7 +1,8 @@
 
 from handlers import pipisa
-from handlers import davinci 
-from handlers import time_new_year
-from handlers import sendalarm
 from handlers import start_help
 
+from handlers import time_new_year
+from handlers import sendalarm
+from handlers import davinci 
+

handlers/davinci.py

@@ -3,14 +3,15 @@ from aiogram import types, Dispatcher
 from create_bot import dp, bot
 
 
+
 token = '5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc'
 #openai.api_key = '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'
 openai.api_key = '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' 
-
+ban_comands = ['/dick','/topdick']
 
 @dp.message_handler()
 async def send(message: types.Message):
-    if 'тупица' in message.text.lower():
+    if message.text.lower().startswith('тупица'):
 
         ## TODO уточнить остатки токенов и денег
         response = openai.Completion.create(
@@ -25,5 +26,20 @@ async def send(message: types.Message):
         )
         await message.reply(response['choices'][0]['text'])
 
+    # elif message.reply_to_message.bot.id:
+    #     response = openai.Completion.create(
+    #         model="text-davinci-003",
+    #         prompt=message.text,
+    #         temperature=0.7,
+    #         max_tokens=1000,
+    #         top_p=1.0,
+    #         frequency_penalty=0.0,
+    #         presence_penalty=0.6,
+    #         stop=["сброс"]
+    #     )
+    #     await message.reply(response['choices'][0]['text'])
+        
+        
+
 
 

handlers/pipisa.py

@@ -12,184 +12,193 @@ 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
 
-    
-    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 message.from_user.id in admins or message.chat.type != 'private':
 
 
-    # рандомайзер
-    numb = randint(-9,10)
-    dick_plus  = (
-        '<b>вырос</b>',
-        '<b>подрос</b>',
-        '<b>привстал</b>',
-        '<b>увеличился</b>',
-        '<b>преисполнился</b>'
-    )
-    dick_minus = (
-        '<b>неудачно потусил на сходке</b>',
-        '<b>уменьшился</b>',
-        '<b>откусила злая гадюка</b>',
-        '<b>продали евреи, он стал дешевле</b>',
-        '<b>схуднул</b>'
-    )
-    dick_stayed = (
-        '<b>продвинулся</b>',
-        '<b>наноуменьшился</b>',
-        '<b>зафиксировал прибыль</b>'
-    )
-
-    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) + '✂'
+        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:
-        #### Чекнуть есть ли юзер в базе
-        candidate_cursor = dicks_collection.find(
-            {
-                'user_id': message.from_user.id,
-                'chat_id': message.chat.id
-            },
-            skip=0, limit=1488)
+        # рандомайзер
+        numb = randint(-9,10)
+        dick_plus  = (
+            '<b>вырос</b>',
+            '<b>подрос</b>',
+            '<b>привстал</b>',
+            '<b>увеличился</b>',
+            '<b>преисполнился</b>'
+        )
+        dick_minus = (
+            '<b>неудачно потусил на сходке</b>',
+            '<b>уменьшился</b>',
+            '<b>откусила злая гадюка</b>',
+            '<b>продали евреи, он стал дешевле</b>',
+            '<b>схуднул</b>'
+        )
+        dick_stayed = (
+            '<b>продвинулся</b>',
+            '<b>наноуменьшился</b>',
+            '<b>зафиксировал прибыль</b>'
+        )
 
-        if candidate_cursor.count() > 0: 
-            ## Если есть то нарастить и отправить сообщение
-
-            user = candidate_cursor.pop()
+        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) + '✂'
 
 
-            if pipisa_time.rolltime(check_datetime=user['datetimes'], curr_time=datetime.datetime.now()):
-                # пришло время крутить!!
+        try:
+            #### Чекнуть есть ли юзер в базе
+            candidate_cursor = dicks_collection.find(
+                {
+                    'user_id': message.from_user.id,
+                    'chat_id': message.chat.id
+                },
+                skip=0, limit=1488)
     
-                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) 
+            if candidate_cursor.count() > 0: 
+                ## Если есть то нарастить и отправить сообщение
 
-                randomword  = ForReadDict.RandomDick()
+                user = candidate_cursor.pop()
     
 
-                await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш <em>{randomword}</em> {size_change}\
+                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}, ваш <em>{randomword}</em> {size_change}\
 на <b>{abs(numb)}</b> см!\n\
 Теперь он равен <b>{updetedDick}</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) 
+
+                    print(f'пользователь @{message.from_user.username} уже крутил')
+
             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) 
+                    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
+                    }
     
-                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')
+                    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\
 на <b>{abs(numb)}</b> см!\n\
 Теперь он равен <b>{numb}</b> см!') 
-                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\
 Ты стартуешь не с лучшей позиции, природа наградила тебя всего <b>{numb}</b> см 😬')
                         
                         
-                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, '🍆 Никто ничего не нарастил! 🍌')
+        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
 
-    if len(top_dicks) == 0:
-        await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌')
+            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 bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos)
+        await message.reply('Работает только в общих чатах!\n'\
+                            'Вы мсжете посмотреть топ по миру /globaldick')
 
 
 
@@ -226,3 +235,4 @@ async def send_global_top(message: types.Message):
 
         await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + dickos)
 
+

handlers/start_help.py

@@ -1,22 +1,27 @@
 from aiogram import types, Dispatcher
+from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
 from create_bot import dp, bot
 
 @dp.message_handler(commands=['start'])
 async def start_func(message: types.Message):
-    await message.reply('похуй')
+    if message.chat.type == 'private':
+        
+        lkb = InlineKeyboardMarkup(row_width=1).add(InlineKeyboardButton(text = 'Добавить в группу', callback_data='add_2_group'))
+
+        await message.answer(f'<b>Добро пожаловать, {message.from_user.full_name}!</b>\n'
+                             'Это бот, который растит члены. Чтобы начать, добавь бота в чат'\
+                             , reply_markup=lkb
+                                                   
+                            )
+# TODO добавить кнопку, по которой смогут добавлять юзера в чат
+        
+@dp.callback_query_handler(text = 'add_2_group')
+async def addgroup(callback: types.CallbackQuery):
+    await callback.answer('жмав') 
+    
+
+    
 
-@dp.message_handler(commands=['photo'])
-async def send_image(message: types.Message):
-    await bot.send_photo(message.chat.id, photo='https://memepedia.ru/wp-content/uploads/2018/08/ya-pidoras.jpg', reply_to_message_id=message.message_id)
 
-@dp.message_handler(content_types=types.ContentTypes.TEXT)
-async def send_faggot(message: types.Message):
-    print(message.text) 
-    if message.text == 'хто я':
-     await bot.send_photo(message.chat.id, photo='https://www.meme-arsenal.com/memes/ecebd2339c7eab40e09874bd86a38a09.jpg')
 
 
-def register_handlers_StartHelp(dp: Dispatcher):
-    dp.register_message_handler(start_func)
-    dp.register_message_handler(send_faggot)
-    dp.register_message_handler(send_image)