меньше кода - лучше!

d

.drone.yml

@@ -15,11 +15,18 @@ steps:
 - name: test it first
   commands:
   - /usr/bin/python3 -m py_compile bot.py
+  - /usr/bin/python3 pipisa_functions/pipisa_time.py
 - name: pull into the folduh
   commands:
   - cd /usr/share/$DRONE_REPO_NAME
-  - git pull
-  - chown -R jack:jack .
+  - su -c "git pull" jack
+  #- chown -R jack:jack .
+  when:
+    branch:
+    - main
 - name: restart systemd unit
   commands:
   - systemctl restart jack_bot
+  when:
+    branch:
+    - main

README.md

@@ -0,0 +1,13 @@
+Пробуем проверять что-либо. 
+
+```mermaid
+journey
+  title Пиписа
+  section Наш Чат
+    Оче больше пипис: 5: ГЕГЕМОН
+    Обычно большепенис: 3: Догоняющие
+    Скромный дик: 1: Отставшие
+  section Не наш чат
+    Не запустил: -1: Отставшие
+    Не крутил: 1: Отставшие
+```

bot.py

@@ -1,28 +1,45 @@
 import logging
-import graypy
-
-from aiogram import executor
-
+from aiogram import executor, Bot, Dispatcher
 from create_bot import dp
-
 import handlers
+import json
+from global_conf import CONFIG
 
 
+
+def remake_field_generator(d, field='text'): 
+    for k, v in d.items():
+        if k == field: 
+            d[k] = '<blank!>' 
+            yield v
+        elif isinstance(v, list):
+            for k2 in v:
+                for id_val in remake_field_generator(k2):
+                    yield id_val
+        elif isinstance(v, dict):
+            for id_val in remake_field_generator(v):
+                yield id_val 
+
+def filter_grammar_messages(record):
+    if record.args and (not None in record.args):
+        j = json.loads(record.args[2]) 
+        for _ in remake_field_generator(j):
+            pass   
+        record.args = (record.args[0], record.args[1], json.dumps(j))
+    return True
+
 logging.basicConfig(level=logging.DEBUG)
+logging.getLogger("aiogram").addFilter(filter_grammar_messages)
 
 
-logging.basicConfig(
-    level    = logging.DEBUG,
-    handlers = [ graypy.GELFUDPHandler('localhost', 12201) ]
-)
-
 # pipisa.register_handlers_pipisa(dp)
 # time_new_year.register_handlers_time(dp)
 # sendalarm.register_handlers_test(dp)
 # start_help.register_handlers_StartHelp(dp)
 
-
 if __name__ == '__main__':
+    #print(CONFIG)
+    
     executor.start_polling(dp, skip_updates=True)
     
 

config.yaml

@@ -0,0 +1,17 @@
+---
+databaso:
+  host:       'https://arango.guaranteedstruggle.host'
+  user:        root
+  pass:        stolendick527
+  base:        pipisa
+  collection:  dicks
+
+telegram_token: 5724991559:AAEuLvpLsgP6LHRGMSyFtQLlR5qPQUO4b_w
+telegram_admins_ids:
+- 317051301 # hogweed1
+- 226765676 # blackjack
+- 94830892  # viktorRMI
+
+openai:
+  token:   5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc
+  api_key: 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  ## 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

create_bot.py

@@ -1,4 +1,8 @@
 from aiogram import Bot, Dispatcher
-API_TOKEN = '5724991559:AAEuLvpLsgP6LHRGMSyFtQLlR5qPQUO4b_w'
-bot = Bot(token=API_TOKEN, parse_mode="HTML")
-dp = Dispatcher(bot)
+from global_conf import CONFIG
+
+
+
+
+bot = Bot(token=CONFIG['telegram_token'], parse_mode="HTML")
+dp  = Dispatcher(bot)

dicktxt/ForReadDict.py

@@ -7,10 +7,6 @@ def RandomDick():
 # когда указываешь путь
     with open('dicktxt/dick_sinonims.txt', 'r', encoding='utf-8') as file:
         words = file.readlines()
+ 
+    return random.choice(words).strip()
 
-    random_line = random.choice(words).strip()
-    return random_line
-
-
-
-    

dicktxt/WordsChange.py

@@ -0,0 +1,21 @@
+import yaml
+import random
+
+
+def ChangeWord(result): #выбирает рандомное слово для изменения размера пиписы
+
+        with open("dicktxt/dick_changes.yaml", 'r', encoding='utf-8') as f:
+            words = yaml.load(f, Loader=yaml.Loader)
+
+        dick_inc     = [ f'<b>{w}</b>' for w in words['increment']  ]
+        dick_static  = [ f'<b>{w}</b>' for w in words['nothing']    ]
+        dick_decr    = [ f'<b>{w}</b>' for w in words['decrement']  ]
+
+        if result > 0:
+            size_change  = random.choice(dick_inc)    + '🚀'
+        elif result == 0:
+            size_change  = random.choice(dick_static) + '🤨' 
+        else: 
+            size_change  = random.choice(dick_decr)   + '✂'
+
+        return size_change

dicktxt/dick_changes.yaml

@@ -0,0 +1,30 @@
+---
+increment:
+- вырос
+- подрос
+- привстал
+- увеличился
+- преисполнился
+- приподнялся
+- изменил размер
+- облагородился
+- чрезвычайно преуспел
+decrement:
+- неудачно потусил на сходке
+- уменьшился
+- откусила злая гадюка
+- продали евреи, он стал дешевле
+- схуднул
+- отрицательно вырос
+- словил низкие вибрации
+- повредился после простуды
+- стал жертвой репрессий
+- был подло использован
+nothing:
+- продвинулся
+- наноуменьшился
+- познал стабильность
+- зафиксировал прибыль
+- подумал о судьбах Китая
+- прогулялся по лесу
+- самоутвердился

dicktxt/dick_sinonims.txt

@@ -1,4 +1,4 @@
-Кожаный бабах
+кожаный бабах
 пистон
 дуршлаг
 елдак
@@ -92,4 +92,5 @@
 маленький жиголо
 венерический суккуб
 боец невидимого фронта
-сморчок
+сморчок
+джечник

global_conf.py

@@ -0,0 +1,6 @@
+import yaml
+
+
+
+with open("config.yaml", 'r') as f:
+    CONFIG = yaml.load(f, Loader=yaml.Loader)

handlers/__init__.py

@@ -1,7 +1,6 @@
 
 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
+

handlers/davinci.py

@@ -1,25 +1,79 @@
 import openai
-from aiogram import types, Dispatcher
+from aiogram import types
 from create_bot import dp, bot
+import logging
+from global_conf import CONFIG
 
-token = '5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc'
-openai.api_key = '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'
+
+
+token          = CONFIG['openai']['token'] 
+openai.api_key = CONFIG['openai']['api_key']
+ban_comands    = ['/dick','/topdick']
+
+# Image_promt =['тупица сгенерируй','тупица сгенерируй картинку',\
+#                                          'тупица отправь картинку', 'тупица отправь картинку']
+
+
+@dp.message_handler(commands=['image'])
+async def send_image(message: types.Message):
+    try:
+        description = message.text.replace('/image', '').strip()
+        
+        if not description:
+            await message.reply('Ты даун, описание запроса напиши после пробела')
+            return
+    except Exception as e:
+        logging.error(f'Error in send_image: {e}')
+    try:
+
+        response = openai.Image.create(
+        prompt=message.text,
+        n=1,
+        size="1024x1024",
+        response_format="url",
+        
+    )
+        image_url = response['data'][0]['url']
+        await message.answer('Генерирую изображение...')
+        await message.reply_photo(image_url)
+    except Exception as e:
+        await message.reply('у меня не получилось')
 
 @dp.message_handler()
 async def send(message: types.Message):
-    if 'тупица' in message.text.lower():
+    if message.text.lower().startswith('тупица'):
 
+        ## TODO уточнить остатки токенов и денег
         response = openai.Completion.create(
-        model="text-davinci-003",
-        prompt=message.text[7:],
-        temperature=0.7,
-        max_tokens=1000,
-        top_p=1.0,
-        frequency_penalty=0.0,
-        presence_penalty=0.6,
-        stop=["сброс"]
+            model="text-davinci-003",
+            prompt=message.text[7:],
+            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'])
+
+    elif message.reply_to_message.from_user.is_bot:
+        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

@@ -1,179 +1,203 @@
-from aiogram import types, Dispatcher
-from sqlalchemy.orm.session import close_all_sessions
+from aiogram import types
 from create_bot import bot, dp
-from sqdb import user, session
 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
+
+#### https://docs.python-arango.com/en/main/
+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):
-# рандомайзер
-    numb = randint(-6,10)
-    global size_change, dick_plus, dick_minus
-    dick_plus  = ('<b>вырос</b>', '<b>подрос</b>', '<b>привстал</b>')
-    dick_minus = ('<b>уменьшился</b>', '<b>откусила злая гадюка</b>', '<b>продали евреи, он стал дешевле</b>')
-    if numb > 0:
-         size_change  = random.choice(dick_plus) + '🚀'
-	#if numb = 0 size_change = random.choice(const)  TODO
-    else: size_change = random.choice(dick_minus) + '✂'
+
+    if message.from_user.id in admins or message.chat.type != 'private':
+
+        dicks_collection = get_dicks_collection()
+
+        #### TODO уплатить нолог с 100 см для самого нищего
 
 
-# работа с таблицей
-    try:
-        b = session.query(user).filter(user.user_id == message.from_user.id).first()
-        if b:
-            last_time = b.datetimes
-            now_time = datetime.datetime.now()
-            diff = now_time.day - last_time.day
-            
-            if abs(diff) <= 0:
-                try:
-                    current_dick = b.dick_size
-                    
-                    
-                    await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер <b>{current_dick}</b> см! ')
-                except:
-                    print('ошибка в if diff <=0')
-                    session.rollback()   
-                finally: 
-                    session.close_all()
-                    print('пользователь уже крутил')
-                    
-            else:
-                try:
-                    a = session.query(user).filter(user.user_id == message.from_user.id).first() # запрос на поиск в таблице
-                    updetedDick = a.dick_size = (a.dick_size + numb)
-                    a.datetimes = datetime.datetime.now()
-                    session.flush()
-                    session.commit() #обновляет запись
-                    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\
+        # рандомайзер
+        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 = candidate_cursor.pop()
+    
+
+                if pipisa_time.rolltime(check_datetime=user['datetimes'], curr_time=datetime.datetime.now()):
+                    # пришло время крутить!!
+    
+                    updetedDick = user["dick_size"] + result 
+                    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() 
+                    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> см! ')
-                except:
-                    session.rollback()
-                    print('ошибка в обновлении пиписы')
-                finally:
-                    session.close_all()
-                    print('пиписа обновлена')
-                
+                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) 
 
-        else:
-            session.close()
-            try:
-                int_table = user(user_id = message.from_user.id, user_fullname = message.from_user.full_name, dick_size = numb, datetimes = datetime.datetime.now(),\
-            chat_id = message.chat.id)
+                    print(f'пользователь @{message.from_user.username} уже крутил')
 
-                session.add(int_table)
-                session.flush()
-                session.commit()
-                session.close()
-            
-                await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн {size_change}\n\
-на <b>{abs(numb)}</b> см!\n\
-Теперь он равен <b>{numb}</b> см!')
-            except:
-                print('ошибка в добавлении нового пользователя')
-                session.rollback()
-            finally:
-                session.close_all()
-                print('Успешно добавлен пользователь')
-        
-    except:
-        session.rollback()
-        print('ошибка в основном коде')
-    finally:
-        session.close_all()
+            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')
+                
+
+                    if result > 0:
+                        await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн показал головку 🚀\n\
+на <b>{abs(result)}</b> см!\n\
+Теперь он равен <b>{result}</b> см!') 
+                    else:
+                        await bot.send_message(message.chat.id, 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)
+
+
+        #### для всех отложенных сообщений
+            ## проверить если прошло ?пять минут? с публикации
+                ## попробовать удалить такой пост и затереть его с коллекции
+            ## оставлять последние в чате топ- и глобал-дик, сохранять последние сообщения о приросте, сохранять последний вызов /dick и иной команды
+    else:
+        await message.reply('Растить елду можно только в общих чатах!')
+
+ 
+ 
         
     
-@dp.message_handler(commands=["topdick"])
+@dp.message_handler(commands=['topdick','globaldick'])
 async def send_topchat(message: types.Message):
-    try:
-        changechat_id = session.query(user).filter(user.user_id == message.from_user.id).first()
-        chats = message.chat.id 
-        if changechat_id.chat_id != chats:
-            changechat_id.chat_id = chats
-            session.commit()
-            session.close()
+    if message.from_user.id in admins or message.chat.type != 'private':
+
+        top_  = False
+        glob_ = False
+        dicks_collection = get_dicks_collection()
+
+        if   message['text'] == '/topdick':
+            top_  = True
+            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)
+        elif message['text'] == '/globaldick':
+            glob_ = True
+            try:
+                dicks = dicks_collection.all( ) 
+                dicks = [d for d in dicks if d['user_id'] != d['chat_id']]
+            except Exception as e: 
+                print('ошибка DB в /globaldick')
+                print(e)
         else:
-            session.close()
-            top = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).filter(user.chat_id == chats).limit(10).all()
-            delimiter = ': '
-
-    # проверка на длину списка, если меньше limit, то:
-
-        sizelen = len(top)
-        if sizelen < 10:
-            lenminus = 10 - sizelen
-            top.extend(('-'*lenminus))
-            await bot.send_message(message.chat.id, f'🏆Топ 10 бубылд чата🏆\n\n\
-    🚀 {delimiter.join(map(str, top[0]))}\n\
-    💫 {delimiter.join(map(str, top[1]))}\n\
-    🍆 {delimiter.join(map(str, top[2]))}\n\
-    🍌 {delimiter.join(map(str, top[3]))}\n\
-    🐍 {delimiter.join(map(str, top[4]))}\n\
-    🐎 {delimiter.join(map(str, top[5]))}\n\
-    🌭 {delimiter.join(map(str, top[6]))}\n\
-    🌶 {delimiter.join(map(str, top[7]))}\n\
-    👌 {delimiter.join(map(str, top[8]))}\n\
-    💩 {delimiter.join(map(str, top[9]))}\n')
-            session.close()
-        # если все нормально и участников минимум 10
+            print('вызывают хз что!')  
+ 
+        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]
+    
+        dickos = ''
+        emo    = ['🏆','🚀','🍆','🍌','🐍','🐎','🌭','🌶','👌','💩']
 
+        if len(top_dicks) == 0:
+            await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌')
         else:
-            
-            await bot.send_message(message.chat.id, f'🏆Топ 10 бубылд чата🏆\n\n\
-    🚀 {delimiter.join(map(str, top[0]))}\n\
-    💫 {delimiter.join(map(str, top[1]))}\n\
-    🍆 {delimiter.join(map(str, top[2]))}\n\
-    🍌 {delimiter.join(map(str, top[3]))}\n\
-    🐍 {delimiter.join(map(str, top[4]))}\n\
-    🐎 {delimiter.join(map(str, top[5]))}\n\
-    🌭 {delimiter.join(map(str, top[6]))}\n\
-    🌶 {delimiter.join(map(str, top[7]))}\n\
-    👌 {delimiter.join(map(str, top[8]))}\n\
-    💩 {delimiter.join(map(str, top[9]))}\n')
-    except:
-        session.rollback()
-        print('ошибка в /topdick')
-    finally:
-        session.close_all()
-
-
-
-@dp.message_handler(commands=["globaldick"])
-async def send_global_top(message: types.Message):
-    try:
-        top_chat = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).limit(10).all()
-        
-        delimiter = ': '
-            
-        
-        await bot.send_message(message.chat.id, f'🏆Топ 10 пипис в мире🏆\n\n\
-    1. {delimiter.join(map(str, top_chat[0]))}\n\
-    2. {delimiter.join(map(str, top_chat[1]))}\n\
-    3. {delimiter.join(map(str, top_chat[2]))}\n\
-    4. {delimiter.join(map(str, top_chat[3]))}\n\
-    5. {delimiter.join(map(str, top_chat[4]))}\n\
-    6. {delimiter.join(map(str, top_chat[5]))}\n\
-    7. {delimiter.join(map(str, top_chat[6]))}\n\
-    8. {delimiter.join(map(str, top_chat[7]))}\n\
-    9. {delimiter.join(map(str, top_chat[8]))}\n\
-    10. {delimiter.join(map(str, top_chat[9]))}\n')
-    except:
-        session.rollback()
-        print('ошибка в /glovaldick')
-    finally:
-        session.close_all()
-
-
-# def register_handlers_pipisa(dp: Dispatcher):
-#     dp.register_message_handler(up_dick)
-#     dp.register_message_handler(send_topchat)
-#     dp.register_message_handler(send_global_top)
+            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 top_:
+                await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos)
+            elif glob_: 
+                await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + dickos)
+    else:
+        await message.reply('Работает только в общих чатах!\n'\
+                            'Вы можете посмотреть топ по миру /globaldick')

handlers/sendalarm.py

@@ -1,29 +0,0 @@
-
-from create_bot import dp, bot
-from aiogram import types, Dispatcher
-from sqdb import user, session
-
-
-
-@dp.message_handler(commands=['sendall'])
-async def sendall(message: types.Message):
-    print('команда прошла')
-    if message.chat.type == 'private':
-        if message.from_user.id == 226765676:
-            text = message.text[9:]
-            try:
-                chats_id = session.query(user.chat_id).distinct()
-                for row in chats_id:
-                    try:
-                        await bot.send_message(row[0], text)
-                    except:
-                        print('вероятно бота нет в чате')
-                await bot.send_message(message.from_user.id, 'Сообщение успешно отправлено во все чаты')
-            except: 
-                session.rollback()
-                print('eror sendall')
-            finally: session.close_all()
-
-
-# def register_handlers_test(dp: Dispatcher):
-#     dp.register_message_handler(send_welcome)

handlers/start_help.py

@@ -1,22 +1,27 @@
-from aiogram import types, Dispatcher
+from aiogram import types
+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)

handlers/time_new_year.py

@@ -4,7 +4,7 @@ import datetime
 
 @dp.message_handler(commands=["time"])
 async def send_time(message: types.Message):
-    new_year = datetime.datetime(2022, 12, 31) #дата нового года
+    new_year = datetime.datetime(datetime.now().year, 12, 31) #дата нового года
     ct = datetime.datetime.now() #датавремя
     cd = datetime.datetime.now().strftime("%d/%m/%Y") #дата (д,м,г)
     ct1 = ct+datetime.timedelta(hours=15) # +14 часов от сервера

pipisa_functions/__init__.py

@@ -0,0 +1 @@
+from . import pipisa_time

pipisa_functions/pipisa_time.py

@@ -0,0 +1,88 @@
+
+import unittest
+import datetime
+
+
+
+def rolltime(check_datetime , curr_time=datetime.datetime.now()):
+
+    last_time    = datetime.datetime.strptime(check_datetime, '%Y-%m-%d %H:%M:%S')
+    time_to_grow = curr_time.replace(hour=14, minute=8, second=8, microsecond=0) #+ datetime.timedelta(days=0)
+
+    timediff = (time_to_grow - last_time).total_seconds()
+
+    # print()
+    # print(f'\tlast_time : {last_time}')
+    # print(f'\ttime_to_grow : {time_to_grow}')
+    # print(f'\tcurr_time : {curr_time.strftime("%Y-%m-%d %H:%M:%S")}')
+    # print(f'\t\tsecs diff : {timediff}')
+
+
+    if timediff > 86400 or (curr_time > time_to_grow and last_time < time_to_grow):
+        return True
+    else:
+        return False
+ 
+
+
+class TestPipisa(unittest.TestCase):
+    def test_very_long_ago(self): 
+        self.assertEqual(rolltime('2021-10-11 23:44:00'), True, "Пиписа прокручена очень давно. Очевидный прокрут")
+
+    def test_just_after(self): 
+        self.assertEqual(rolltime(
+            datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0).strftime('%Y-%m-%d %H:%M:%S')
+            ), False, "Попытка крутить когда старый прокрут был через 1 минуту после сегодняшнего обновления роллов. Фейл.")
+
+    def test_yesterdayroll(self): 
+        self.assertEqual(
+            rolltime(
+                ( 
+                    datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0) - datetime.timedelta(days=1) 
+                ).strftime('%Y-%m-%d %H:%M:%S'),
+                datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0)
+            ), True, "Крутилась вчера через минуту после обновления роллов. Должно пропускать ")
+        
+        
+    def test_hour_before_roll(self): 
+        self.assertEqual(
+            rolltime(
+                ( 
+                    datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0) - datetime.timedelta(hours=1)
+                ).strftime('%Y-%m-%d %H:%M:%S'),
+                datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0)
+            ), True, "Крутилась последний раз за час до прокрута. Надо разрешать заново!")
+        
+    def test_after_midnight(self): 
+        self.assertEqual(
+            rolltime(
+                ( 
+                    datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0) - datetime.timedelta(hours=14) 
+                ).strftime('%Y-%m-%d %H:%M:%S'),
+                datetime.datetime.now().replace(hour=0, minute=11, second=8, microsecond=0)
+            ), False, "Где-то ночью прокручивается повторно. Фейл.")
+        
+    def test_before_midnight(self): 
+        self.assertEqual(
+            rolltime(
+                ( 
+                    datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0) - datetime.timedelta(hours=15) 
+                ).strftime('%Y-%m-%d %H:%M:%S'),
+                datetime.datetime.now().replace(hour=0, minute=11, second=8, microsecond=0) - datetime.timedelta(hours=15) 
+            ), False, "Где-то ночью прокручивается повторно. Фейл.")
+        
+    def test_somewhere_after(self): 
+        self.assertEqual(
+            rolltime(
+                ( 
+                    datetime.datetime.now().replace(hour=14, minute=26, second=8, microsecond=0)  
+                ).strftime('%Y-%m-%d %H:%M:%S'),
+                datetime.datetime.now().replace(hour=14, minute=27, second=8, microsecond=0)
+            ), False, "Сегодня уже открутили а хотят ещё. Фейл.")
+
+
+
+
+
+if __name__ == "__main__":
+    unittest.main(verbosity=2)

requairements.txt

@@ -1,4 +1,3 @@
 aiogram
-sqdb
-sqlalchemy
-graypy
+python-arango
+pyyaml

sqdb.py

@@ -1,56 +0,0 @@
-from sqlalchemy import create_engine, MetaData, Table, Integer, String, BIGINT,  \
-    Column, DateTime, ForeignKey, Numeric
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import sessionmaker
-from sqlalchemy import select, update
-
-engine = create_engine('postgresql+psycopg2://postgres:postgres@postgres.vdk2ch.ru:5432/jack_bot', echo=True)
-Session = sessionmaker(bind=engine)
-session = Session()
-
-
-
-Base = declarative_base()
-
-
-
-class user(Base):
-    __tablename__ = 'users'
-
-    id = Column(Integer, primary_key=True)
-    user_id = Column(Integer)
-    user_fullname = Column(String)
-    dick_size = (Column(Integer))
-    datetimes = (Column(DateTime))
-    chat_id = Column(BIGINT)
-
-
-
-
-    
-session.close()
-engine.dispose()
-# top = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).filter(user.chat_id == user.chat_id).limit(10).all()
-# print(top)
-
-
-#Base.metadata.create_all(engine)
-
-# int_table = user(user_id = 12345678, user_fullname = 'test', dick_size = 10) #импортирую юзер из sqdb
-# session.add(int_table)
-# session.commit()
-# select user_id, name from employees limit 10
-
-# def adding():
-        
-#     session.add()
-#     session.commit()
-        
-
-
-
-
-
-
-
-