делаем интереснее

handlers/pipisa.py

@@ -3,7 +3,7 @@ from create_bot import bot, dp
 import random
 from random import randint
 import datetime
-from dicktxt import ForReadDict
+from dicktxt import ForReadDict, WordsChange
 from pipisa_functions import pipisa_time
 
 from global_conf import CONFIG
@@ -14,54 +14,30 @@ from arango import ArangoClient
 admins = CONFIG['telegram_admins_ids']
 
 
+def get_dicks_collection(): 
+    arango_client = ArangoClient(hosts=CONFIG['databaso']['host'] ) 
+    pipisa_db     = arango_client.db(
+        CONFIG['databaso']['base'], 
+        username=CONFIG['databaso']['user'],
+        password=CONFIG['databaso']['pass']
+    ) 
+    dicks_collection = pipisa_db.collection(CONFIG['databaso']['collection']) 
+
+    return dicks_collection
+
+
 @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=CONFIG['databaso']['host'] ) 
-        pipisa_db     = arango_client.db(
-            CONFIG['databaso']['base'], 
-            username=CONFIG['databaso']['user'],
-            password=CONFIG['databaso']['pass']
-        ) 
-        dicks_collection = pipisa_db.collection(CONFIG['databaso']['collection']) 
-
+        dicks_collection = get_dicks_collection()
 
         #### TODO уплатить нолог с 100 см для самого нищего
 
 
         # рандомайзер
-        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) + '✂'
-
+        result = randint(-9,10)
 
         try:
             #### Чекнуть есть ли юзер в базе
@@ -81,7 +57,7 @@ async def up_dick(message: types.Message):
                 if pipisa_time.rolltime(check_datetime=user['datetimes'], curr_time=datetime.datetime.now()):
                     # пришло время крутить!!
     
-                    updetedDick = user["dick_size"] + numb 
+                    updetedDick = user["dick_size"] + result 
                     try:
                         dt = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
                         metadata = dicks_collection.insert(
@@ -101,11 +77,11 @@ async def up_dick(message: types.Message):
                         print(f'ошибка в обновлении пиписы @{message.from_user.username}')
                         print(e2) 
 
-                    randomword  = ForReadDict.RandomDick()
+                    randomword    = ForReadDict.RandomDick() 
+                    change_phrase = WordsChange.ChangeWord(result)
 
-
-                    await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш <em>{randomword}</em> {size_change}\
-на <b>{abs(numb)}</b> см!\n\
+                    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> см! ')
                 else:
                     await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер <b>{user["dick_size"]}</b> см! ')
@@ -137,7 +113,7 @@ async def up_dick(message: types.Message):
                     novenkiy = {
                         'user_id':       message.from_user.id,
                         'user_fullname': message.from_user.full_name,
-                        'dick_size':     numb,
+                        'dick_size':     result,
                         'datetimes':     datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
                         'attempts':      1,
                         'chat_id':       message.chat.id
@@ -146,13 +122,13 @@ async def up_dick(message: types.Message):
                     metadata = dicks_collection.insert(novenkiy, overwrite_mode='update')
                 
 
-                    if numb > 0:
+                    if result > 0:
                         await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн показал головку 🚀\n\
-на <b>{abs(numb)}</b> см!\n\
-Теперь он равен <b>{numb}</b> см!') 
+на <b>{abs(result)}</b> см!\n\
+Теперь он равен <b>{result}</b> см!') 
                     else:
                         await bot.send_message(message.chat.id, f'@{message.from_user.username}, добро пожаловать в игру!🚀\n\
-Ты стартуешь не с лучшей позиции, природа наградила тебя всего <b>{numb}</b> см 😬')
+Ты стартуешь не с лучшей позиции, природа наградила тебя всего <b>{result}</b> см 😬')
                         
                         
                     print(f'Успешно добавлен пользователь @{message.from_user.username}')
@@ -164,6 +140,12 @@ async def up_dick(message: types.Message):
         except Exception as e: 
             print('ошибка DB в /dick')
             print(e)
+
+
+        #### для всех отложенных сообщений
+            ## проверить если прошло ?пять минут? с публикации
+                ## попробовать удалить такой пост и затереть его с коллекции
+            ## оставлять последние в чате топ- и глобал-дик, сохранять последние сообщения о приросте, сохранять последний вызов /dick и иной команды
     else:
         await message.reply('Растить елду можно только в общих чатах!')
 
@@ -174,15 +156,8 @@ async def up_dick(message: types.Message):
 @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=CONFIG['databaso']['host'] ) 
-        pipisa_db     = arango_client.db(
-            CONFIG['databaso']['base'], 
-            username=CONFIG['databaso']['user'],
-            password=CONFIG['databaso']['pass']
-        ) 
-        dicks_collection = pipisa_db.collection(CONFIG['databaso']['collection']) 
-
+ 
+        dicks_collection = get_dicks_collection()
         try:
             dicks = dicks_collection.find({'chat_id': message.chat.id}, skip=0, limit=1100)
             dicks = [d for d in dicks]
@@ -214,16 +189,8 @@ async def send_topchat(message: types.Message):
 
 
 @dp.message_handler(commands=["globaldick"])
-async def send_global_top(message: types.Message):
-
-
-    arango_client = ArangoClient(hosts=CONFIG['databaso']['host'] ) 
-    pipisa_db     = arango_client.db(
-        CONFIG['databaso']['base'], 
-        username=CONFIG['databaso']['user'],
-        password=CONFIG['databaso']['pass']
-    ) 
-    dicks_collection = pipisa_db.collection(CONFIG['databaso']['collection']) 
+async def send_global_top(message: types.Message): 
+    dicks_collection = get_dicks_collection()
 
     try:
         dicks = dicks_collection.all( )