Merge branch '0.1.0-dev' of https://git.vdk2ch.ru/Pavel_Durov/python_bot into 0.1.0-dev

.drone.yml

@@ -15,12 +15,11 @@ 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
-  - su -c "git pull" jack
-  #- chown -R jack:jack .
+  - git pull
+  - chown -R jack:jack .
   when:
     branch:
     - main

DBChat/user_operations.py

@@ -1,18 +0,0 @@
-import datetime
-
-def UserCreate(DBName, message, result, replyId): #create new user 
-
-        new_user = { #описание класса пользователя 
-                        '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, #уникальный идентификатор конфы, в которой этот пользователь крутит
-                        'reply_id':      replyId.id #идентификатор последнего ответа для этого пользователя
-                    }
-        try:
-            metadata = DBName.insert(new_user, overwrite_mode='update')
-        except Exception as add_err:
-            print('ошибка в добавлении нового пользователя')
-            print(add_err) 

README.md

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

TopMsg/topdicks.py

@@ -1,64 +0,0 @@
-async def topdicks_conf(chatId): # тело метода лучших результатов отдельной конфы
-        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': chatId}, 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)
-        
-        dick_limit = min(len(top_dicks), 10)
-        top_dicks  = top_dicks[:dick_limit]
-    
-        conf_top = ''
-        emo    = ['🏆','🚀','🍆','🍌','🐍','🐎','🌭','🌶','👌','💩']
-
-        if len(top_dicks) == 0:
-            await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌')
-        else:
-            for i in range(len(top_dicks)): 
-                conf_top += f'     {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n'
-                i += 1
-        
-
-        return conf_top
-
-
-
-async def topdicks_global(chatId): #тело метода глобальных лучших результатов
-    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.all( ) 
-        dicks = [d for d in dicks if d['user_id'] != d['chat_id']]
-    except Exception as e: 
-        print('ошибка DB в /glovaldick')
-        print(e)
-
-    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]
-
-    global_top = ''
-    emo    = ['🏆','🚀','💫','🍆','🍌','🐍','🐎','🌭','🌶','👌']
-
-    if len(top_dicks) == 0:
-        await bot.send_message(chatId, '🍆 Никто ничего не нарастил! 🍌')
-    else:
-        for i in range(len(top_dicks)): 
-            global_top += f'     {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n'
-            i += 1
-
-    return global_top
-
-
-    

alembic.ini

@@ -0,0 +1,110 @@
+# A generic, single database configuration.
+
+[alembic]
+# path to migration scripts
+script_location = alembic
+
+# template used to generate migration file names; The default value is %%(rev)s_%%(slug)s
+# Uncomment the line below if you want the files to be prepended with date and time
+# see https://alembic.sqlalchemy.org/en/latest/tutorial.html#editing-the-ini-file
+# for all available tokens
+# file_template = %%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d-%%(rev)s_%%(slug)s
+
+# sys.path path, will be prepended to sys.path if present.
+# defaults to the current working directory.
+prepend_sys_path = .
+
+# timezone to use when rendering the date within the migration file
+# as well as the filename.
+# If specified, requires the python-dateutil library that can be
+# installed by adding `alembic[tz]` to the pip requirements
+# string value is passed to dateutil.tz.gettz()
+# leave blank for localtime
+# timezone =
+
+# max length of characters to apply to the
+# "slug" field
+# truncate_slug_length = 40
+
+# set to 'true' to run the environment during
+# the 'revision' command, regardless of autogenerate
+# revision_environment = false
+
+# set to 'true' to allow .pyc and .pyo files without
+# a source .py file to be detected as revisions in the
+# versions/ directory
+# sourceless = false
+
+# version location specification; This defaults
+# to alembic/versions.  When using multiple version
+# directories, initial revisions must be specified with --version-path.
+# The path separator used here should be the separator specified by "version_path_separator" below.
+# version_locations = %(here)s/bar:%(here)s/bat:alembic/versions
+
+# version path separator; As mentioned above, this is the character used to split
+# version_locations. The default within new alembic.ini files is "os", which uses os.pathsep.
+# If this key is omitted entirely, it falls back to the legacy behavior of splitting on spaces and/or commas.
+# Valid values for version_path_separator are:
+#
+# version_path_separator = :
+# version_path_separator = ;
+# version_path_separator = space
+version_path_separator = os  # Use os.pathsep. Default configuration used for new projects.
+
+# set to 'true' to search source files recursively
+# in each "version_locations" directory
+# new in Alembic version 1.10
+# recursive_version_locations = false
+
+# the output encoding used when revision files
+# are written from script.py.mako
+# output_encoding = utf-8
+
+sqlalchemy.url = postgresql+psycopg2://postgres:postgres@postgres.vdk2ch.ru:5432/jack_bot
+
+
+[post_write_hooks]
+# post_write_hooks defines scripts or Python functions that are run
+# on newly generated revision scripts.  See the documentation for further
+# detail and examples
+
+# format using "black" - use the console_scripts runner, against the "black" entrypoint
+# hooks = black
+# black.type = console_scripts
+# black.entrypoint = black
+# black.options = -l 79 REVISION_SCRIPT_FILENAME
+
+# Logging configuration
+[loggers]
+keys = root,sqlalchemy,alembic
+
+[handlers]
+keys = console
+
+[formatters]
+keys = generic
+
+[logger_root]
+level = WARN
+handlers = console
+qualname =
+
+[logger_sqlalchemy]
+level = WARN
+handlers =
+qualname = sqlalchemy.engine
+
+[logger_alembic]
+level = INFO
+handlers =
+qualname = alembic
+
+[handler_console]
+class = StreamHandler
+args = (sys.stderr,)
+level = NOTSET
+formatter = generic
+
+[formatter_generic]
+format = %(levelname)-5.5s [%(name)s] %(message)s
+datefmt = %H:%M:%S

alembic/README

@@ -0,0 +1 @@
+Generic single-database configuration.

alembic/env.py

@@ -0,0 +1,78 @@
+from logging.config import fileConfig
+
+from sqlalchemy import engine_from_config
+from sqlalchemy import pool
+from sqdb import Base
+from alembic import context
+
+# this is the Alembic Config object, which provides
+# access to the values within the .ini file in use.
+config = context.config
+
+# Interpret the config file for Python logging.
+# This line sets up loggers basically.
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+# add your model's MetaData object here
+# for 'autogenerate' support
+# from myapp import mymodel
+# target_metadata = mymodel.Base.metadata
+target_metadata = Base.metadata
+
+# other values from the config, defined by the needs of env.py,
+# can be acquired:
+# my_important_option = config.get_main_option("my_important_option")
+# ... etc.
+
+
+def run_migrations_offline() -> None:
+    """Run migrations in 'offline' mode.
+
+    This configures the context with just a URL
+    and not an Engine, though an Engine is acceptable
+    here as well.  By skipping the Engine creation
+    we don't even need a DBAPI to be available.
+
+    Calls to context.execute() here emit the given string to the
+    script output.
+
+    """
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def run_migrations_online() -> None:
+    """Run migrations in 'online' mode.
+
+    In this scenario we need to create an Engine
+    and associate a connection with the context.
+
+    """
+    connectable = engine_from_config(
+        config.get_section(config.config_ini_section, {}),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    with connectable.connect() as connection:
+        context.configure(
+            connection=connection, target_metadata=target_metadata
+        )
+
+        with context.begin_transaction():
+            context.run_migrations()
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

alembic/script.py.mako

@@ -0,0 +1,24 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+
+"""
+from alembic import op
+import sqlalchemy as sa
+${imports if imports else ""}
+
+# revision identifiers, used by Alembic.
+revision = ${repr(up_revision)}
+down_revision = ${repr(down_revision)}
+branch_labels = ${repr(branch_labels)}
+depends_on = ${repr(depends_on)}
+
+
+def upgrade() -> None:
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    ${downgrades if downgrades else "pass"}

alembic/versions/7521d874503f_.py

@@ -0,0 +1,47 @@
+"""empty message
+
+Revision ID: 7521d874503f
+Revises: 
+Create Date: 2023-06-22 15:21:26.835612
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = '7521d874503f'
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('dick_requests',
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('size_change', sa.Integer(), nullable=True),
+    sa.Column('user_id', sa.Integer(), nullable=True),
+    sa.Column('date_changed', sa.DateTime(), nullable=True),
+    sa.ForeignKeyConstraint(['user_id'], ['users.user_id'], ),
+    sa.PrimaryKeyConstraint('id')
+    )
+    op.create_foreign_key(None, 'user_dicks', 'users', ['user_id'], ['user_id'])
+    op.add_column('users', sa.Column('date_created', sa.DateTime(), nullable=True))
+    op.add_column('users', sa.Column('date_updated', sa.DateTime(), nullable=True))
+    op.drop_column('users', 'chat_id')
+    op.drop_column('users', 'datetimes')
+    op.drop_column('users', 'dick_size')
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('users', sa.Column('dick_size', sa.INTEGER(), autoincrement=False, nullable=True))
+    op.add_column('users', sa.Column('datetimes', postgresql.TIMESTAMP(), autoincrement=False, nullable=True))
+    op.add_column('users', sa.Column('chat_id', sa.BIGINT(), autoincrement=False, nullable=True))
+    op.drop_column('users', 'date_updated')
+    op.drop_column('users', 'date_created')
+    op.drop_constraint(None, 'user_dicks', type_='foreignkey')
+    op.drop_table('dick_requests')
+    # ### end Alembic commands ###

bot.py

@@ -1,12 +1,20 @@
 import logging
+import graypy
+
 from aiogram import executor
+
 from create_bot import dp
+
 import handlers
 
 
 logging.basicConfig(level=logging.DEBUG)
 
 
+logging.basicConfig(
+    level=logging.DEBUG,
+    handlers=[graypy.GELFUDPHandler('localhost', 12201)]
+)
 
 # pipisa.register_handlers_pipisa(dp)
 # time_new_year.register_handlers_time(dp)
@@ -17,7 +25,3 @@ logging.basicConfig(level=logging.DEBUG)
 if __name__ == '__main__':
     executor.start_polling(dp, skip_updates=True)
 
-
-
-
-

create_bot.py

@@ -1,7 +1,4 @@
 from aiogram import Bot, Dispatcher
-
-
 API_TOKEN = '5724991559:AAEuLvpLsgP6LHRGMSyFtQLlR5qPQUO4b_w'
-
 bot = Bot(token=API_TOKEN, parse_mode="HTML")
 dp = Dispatcher(bot)

dicktxt/change_words.py

@@ -1,34 +0,0 @@
-import random       
-
-def ChangeWord(result): #выбирает рандомное слово для изменения размера пиписы
-
-        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) + '✂'
-
-
-        dick_inc  = (
-            '<b>вырос</b>',
-            '<b>подрос</b>',
-            '<b>привстал</b>',
-            '<b>увеличился</b>',
-            '<b>преисполнился</b>'
-        )
-        dick_static = (
-            '<b>продвинулся</b>',
-            '<b>наноуменьшился</b>',
-            '<b>познал стабильность</b>',
-            '<b>зафиксировал прибыль</b>'
-        )
-        dick_decr = (
-            '<b>неудачно потусил на сходке</b>',
-            '<b>уменьшился</b>',
-            '<b>откусила злая гадюка</b>',
-            '<b>продали евреи, он стал дешевле</b>',
-            '<b>схуднул</b>'
-            '<b>отрицательно вырос</b>'
-        )
-        return size_change

dicktxt/dick_sinonims.txt

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

handlers/__init__.py

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

handlers/davinci.py

@@ -1,49 +1,15 @@
 import openai
-from aiogram import types, Dispatcher
-from create_bot import dp, bot
-import logging
-
-
+from aiogram import types
+from create_bot import dp
 
 token = '5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc'
-#openai.api_key = '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'
-openai.api_key = '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' 
-ban_comands = ['/dick','/topdick']
+openai.api_key = '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'
 
-# 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 message.text.lower().startswith('тупица'):
+    if 'тупица' in message.text.lower():
 
-        ## TODO уточнить остатки токенов и денег
         response = openai.Completion.create(
             model="text-davinci-003",
             prompt=message.text[7:],
@@ -56,24 +22,5 @@ async def send(message: types.Message):
         )
         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,155 +1,201 @@
-from aiogram import types, Dispatcher
+from aiogram import types
+from sqlalchemy.orm.session import close_all_sessions
 from create_bot import bot, dp
+from sqdb import User, session, DickRequest, UserDick
 import random
+import traceback
 from random import randint
 import datetime
 from dicktxt import ForReadDict
-from dicktxt import change_words
-from pipisa_functions import pipisa_time
-from DBChat import user_operations
-from TopMsg import topdicks
-from handlers import post_cleaner
+
+size_change = None
+dick_plus = None
+dick_minus = None
 
 
-#### https://docs.python-arango.com/en/main/
-from arango import ArangoClient
+def get_dick(user_id, chat_id):
+    dick = session.query(UserDick).filter(UserDick.user_id == user_id, UserDick.chat_id == chat_id).first()
+    return dick
 
-admins = [317051301, 226765676]
+def top_dick_message(delimiter, top_list):
+    message = f'🏆Топ 10 бубылд чата🏆\n\n'
+    f'🚀 {delimiter.join(map(str, top_list[0]))}\n'
+    f'💫 {delimiter.join(map(str, top_list[1]))}\n'
+    f'🍆 {delimiter.join(map(str, top_list[2]))}\n'
+    f'🍌 {delimiter.join(map(str, top_list[3]))}\n'
+    f'🐍 {delimiter.join(map(str, top_list[4]))}\n'
+    f'🐎 {delimiter.join(map(str, top_list[5]))}\n'
+    f'🌭 {delimiter.join(map(str, top_list[6]))}\n'
+    f'🌶 {delimiter.join(map(str, top_list[7]))}\n'
+    f'👌 {delimiter.join(map(str, top_list[8]))}\n'
+    f'💩 {delimiter.join(map(str, top_list[9]))}\n'
+    return message
+
+
+@dp.message_handler(commands=["maintenance"])
+async def stop_for_maintenance(message: types.Message):  # остановить бота
+    if message.from_user.id == 226765676:
+        bot.send_message(message.from_user.id, "Блять. Как же больно. Мне холодно. Наверное это конец...")
+        close_all_sessions()
+        dp.stop_polling()
 
 
 @dp.message_handler(commands=["dick"])
-async def up_dick(message: types.Message):
-    #global size_change, dick_plus, dick_minus
+async def up_dick(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') 
-
-
-        
-        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 ## получение итогового результата
-
-                    pipisa_word  = ForReadDict.RandomDick()
-                    change_word = change_words.ChangeWord(result)
-
-                    enl_message = await bot.send_message(message.chat.id, f"@\
-                                        {message.from_user.username}, \
-                                        ваш <em>{pipisa_word}</em> ,{change_word(result)} на <b>{abs(result)}</b> см!\n\ \
-                                        Теперь он равен <b>{updetedDick}</b> см! ")
-                    
-                    post_cleaner.cleanup_sequence(user['user_id'], enl_message.id, message.chat.id)
+    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) + '✂'
 
 
+# работа с таблицей
     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 update_err:
-                        print(f'ошибка в обновлении пиписы @{message.from_user.username}')
-                        print(update_err) 
-                else:
-                    error_msg = await bot.send_message(message.chat.id, f'@\
-                                                    {message.from_user.username},🚫 вы уже крутили пипису, \
-                                                    ее размер <b>{user["dick_size"]}</b> см! ')
-                    
-                    post_cleaner.cleanup_sequence(user['user_id'], error_msg.id, error_msg.chat.id)
+        dick_request = DickRequest(size_change=numb, date_changed=datetime.datetime.now())
+        b = session.query(User).filter(User.user_id == message.from_user.id).first()
+        if b:
+            last_time = b.date_updated
+            now_time = datetime.datetime.now()
+            diff = now_time.day - last_time.day
             
+            if abs(diff) <= 0:
                 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 roll_err:
-                        print(f'ошибка в обновлении попытки крутить пипису @{message.from_user.username}')
-                        print(roll_err) 
+                    current_dick = b.dick_size
+                    await bot.send_message(message.chat.id, f'@{message.from_user.username},'
+                                                            f'🚫 вы уже крутили пипису, '
+                                                            f'ее размер <b>{current_dick}</b> см! ')
+                except Exception as e:
+                    print('ошибка в if diff <=0 ' + "\n".join(traceback.format_tb(e.__traceback__)))
+                    session.rollback()   
+                finally: 
+                    session.close_all()
+                    print('пользователь уже крутил')
                     
-                    print(f'пользователь @{message.from_user.username} уже крутил')
-
-            else: ## если пользователь не найден то создаём ему запись в БД
-                     
-                    if result > 0:
-                        new_reply = await bot.send_message(message.chat.id, f'@\
-                                                        {message.from_user.username}, ваш писюн показал головку 🚀\n\
-                                                        на <b>{abs(result)}</b> см!\n\
-                                                        Теперь он равен <b>{result}</b> см!') 
             else:
-                        new_reply =await bot.send_message(message.chat.id, f'@\
-                                                        {message.from_user.username}, добро пожаловать в игру!🚀\n\
-                                                        Ты стартуешь не с лучшей позиции, природа наградила тебя всего <b>{result}</b> см 😬')
+                try:
+                    usr = session.query(User)
+                    a = usr.filter(User.user_id == message.from_user.id).first()  # запрос на поиск в таблице
+                    dick = get_dick(User.user_id, message.chat.id)
+                    updated_dick = dick.dick_size = (dick.dick_size + numb)
+                    dick_request.user = a
+                    a.date_updated = datetime.datetime.now()
+                    session.add(dick_request)
+                    session.flush()
+                    session.commit()  # обновляет запись
+                    randomword = ForReadDict.RandomDick()
                 
-                    user_operations.UserCreate(dicks_collection, message, result, new_reply.id)
+                    await bot.send_message(message.chat.id, f'@{message.from_user.username}, '
+                                                            f'ваш <em>{randomword}</em> {size_change}'
+                                                            f'на <b>{abs(numb)}</b> см!\n'
+                                                            f'Теперь он равен <b>{updated_dick}</b> см! ')
+                except Exception as e:
+                    session.rollback()
+                    print('ошибка в обновлении пиписы, трейсбек-бек-бек: ' +
+                          "\n".join(traceback.format_tb(e.__traceback__)))
+                finally:
+                    session.close_all()
+                    print('пиписа обновлена')
 
-                    print(f'Успешно добавлен пользователь @{message.from_user.username}')
-                    
-
-
-        except Exception as crit_err: 
-            print('ошибка DB в /dick')
-            print(crit_err)
         else:
-        await message.reply('Растить елду можно только в общих чатах!')
-
+            session.close()
+            try:
+                numb = randint(1, 10)
+                int_table = User(user_id=message.from_user.id,
+                                 user_fullname=message.from_user.full_name,
+                                 date_updated=datetime.datetime.now(),
+                                 date_created=datetime.datetime.now())
+                new_dick = DickRequest(user_id=message.from_user.id,
+                                       chat_id=message.chat.id,
+                                       dick_size=numb)
 
+                dick_request.user = int_table
+                session.add(int_table)
+                session.add(new_dick)
+                session.add(dick_request)
+                session.flush()
+                session.commit()
+                session.close()
+                await bot.send_message(message.chat.id, f'@{message.from_user.username}, '
+                                                        f'ваш писюн показал головку 🚀 \n'
+                                                        f'на <b>{abs(numb)}</b> см!\n'
+                                                        f'Теперь он равен <b>{numb}</b> см!')
+            except Exception as e:
+                print('ошибка в добавлении нового пользователя, трейсбек: ' +
+                      '\n'.join(traceback.format_tb(e.__traceback__)))
+                session.rollback()
+            finally:
+                session.close_all()
+                print('Успешно добавлен пользователь')
         
+    except Exception as e:
+        session.rollback()
+        print('ошибка в основном коде, трейсбек:' + '\n'.join(traceback.format_tb(e.__traceback__)))
+    finally:
+        session.close_all()
         
     
 @dp.message_handler(commands=["topdick"])
-async def send_topchat(message: types.Message): # тело метода лучших результатов отдельной конфы
-    if message.from_user.id in admins or message.chat.type != 'private':
+async def send_top_chat(message: types.Message):
+    try:
+        top = []
+        chats = message.chat.id
+        changechat_id = session.query(DickRequest).filter(DickRequest.user_id == message.from_user.id,
+                                                          DickRequest.chat_id == chats).first()
+        dicks = session.query(DickRequest)\
+            .filter(DickRequest.chat_id == chats)\
+            .limit(10).all()\
+            .order_by(DickRequest.dick_size.desc())
+        for dick in dicks:
+            user = session.query(User.user_fullname).filter(User.user_id == dick.user_id)
+            top.append([user.full_name, dick.dick_size])
+        delimiter = ': '
+
+    # проверка на длину списка, если меньше limit, то:
+
+        size_len = len(top)
+        if size_len < 10:
+            len_minus = 10 - size_len
+            top.extend([('-'*len_minus),""])
+            await bot.send_message(message.chat.id, top_dick_message(delimiter, top))
+            session.close()
+        # если все нормально и участников минимум 10
 
-        dickos = topdicks.topdicks_conf(message.chat.id)
-        await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos)
         else:
-        await message.reply('Работает только в общих чатах!\n'\
-                            'Вы мсжете посмотреть топ по миру: /globaldick')
-
             
+            await bot.send_message(message.chat.id, top_dick_message(delimiter, top))
+    except Exception as e:
+        session.rollback()
+        print('ошибка в /topdick. Трейсбэк:\n' + '\n'.join(traceback.format_tb(e.__traceback__)))
+    finally:
+        session.close_all()
 
 
 @dp.message_handler(commands=["globaldick"])
-async def send_global_top(message: types.Message): # тело метода лучших результатов 
-    tops = topdicks.topdicks_global(message.chat.id,)
+async def send_global_top(message: types.Message):
+    try:
+        top = []
+        dicks = session.query(DickRequest) \
+            .limit(10).all() \
+            .order_by(DickRequest.dick_size.desc())
+        for dick in dicks:
+            user = session.query(User.user_fullname).filter(User.user_id == dick.user_id)
+            top.append([user.full_name, dick.dick_size])
+        delimiter = ': '
 
-    await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + tops)
+        await bot.send_message(message.chat.id, top_dick_message(delimiter, top))
+    except Exception as e:
+        session.rollback()
+        print('ошибка в /globaldick, трейсбэк:\n' + '\n'.join(traceback.format_tb(e.__traceback__)))
+    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)

handlers/post_cleaner.py

@@ -1,69 +0,0 @@
-import datetime
-from arango import ArangoClient
-from create_bot import bot
-
-def get_remove_collection(): 
-    arango_client = ArangoClient(hosts=CONFIG['databaso']['host'] ) 
-    pipisa_db     = arango_client.db(
-        CONFIG['databaso']['base'], 
-        username=CONFIG['databaso']['user'],
-        password=CONFIG['databaso']['pass']
-    ) 
-    remove_collection = pipisa_db.collection(CONFIG['databaso']['deleteCollection']) 
-
-    return remove_collection
-
-async def cleanup_sequence(userID: int, messageID:int, chatID:int):
-            
-    #remove last reply and check posts older than 5 minutes to delete
-    remove_collection = get_remove_collection
-    try:
-        clean_candidate = remove_collection.find(
-                    {
-                       '_key':    userID,
-                       'chat_id': chatID
-                    },
-                    skip=0, limit=1000)
-    
-        prev_enl_message_id = clean_candidate["messageID"]
-        await bot.delete_message(prev_enl_message_id)   #delete last reply to this user
-
-        current_time = datetime.datetime.now()
-        too_old = current_time-datetime.timedelta(min=5)
-        old_posts = remove_collection.find(
-                    {
-                      'chat_id': chatID,
-                    },
-                   skip=0, limit=1000)
-    
-        for old_Post in old_posts:
-          if old_Post["datetimes"] <= too_old:
-               to_delete_ID = old_Post["messageID"]
-               to_delete_UserId = old_Post['_key']
-
-               await bot.delete_message(to_delete_ID) #delete old message
-               old_posts.delete(to_delete_UserId)     #delete old message DB entry
-               
-
-    except Exception as autoCleanErr:
-        print(autoCleanErr)
-
-    #write last reply post ID to db
-    remove_collection = get_remove_collection
-    message_time = datetime.datetime.now()
-
-    try: 
-                    lastReply = {
-                        '_key':          userID,
-                        'datetimes':     message_time,
-                        'chat_id':       chatID,
-                        'messageID':     messageID
-                    }
-    
-                    metadata = remove_collection.insert(lastReply, overwrite_mode='update')     #write last reply time
-                     
-    except Exception as lastWriteErr:
-        print(lastWriteErr)
-
-    return 0
-

handlers/sendalarm.py

@@ -1,8 +1,8 @@
+import traceback
 
 from create_bot import dp, bot
 from aiogram import types, Dispatcher
-from sqdb import user, session
-
+from sqdb import User, session
 
 
 @dp.message_handler(commands=['sendall'])
@@ -12,17 +12,20 @@ async def sendall(message: types.Message):
         if message.from_user.id == 226765676:
             text = message.text[9:]
             try:
-                chats_id = -1001355646177
-                
+                chats_id = session.query(User.chat_id).distinct()
+                for row in chats_id:
                     try:
-                    await bot.send_message(chats_id, text)
-                except:
-                        print('вероятно бота нет в чате')
+                        await bot.send_message(row[0], text)
+                    except Exception as e:
+                        print('вероятно бота нет в чате, на всякий случай трейсбек:\n' +
+                              "\n".join(traceback.format_tb(e.__traceback__)))
                 await bot.send_message(message.from_user.id, 'Сообщение успешно отправлено во все чаты')
-            except: 
+            except Exception as e:
                 session.rollback()
-                print('eror sendall')
-            finally: session.close_all()
+                print('error sendall, traceback:\n' +
+                      "\n".join(traceback.format_tb(e.__traceback__)))
+            finally:
+                session.close_all()
 
 
 # def register_handlers_test(dp: Dispatcher):

handlers/start_help.py

@@ -1,27 +1,28 @@
-from aiogram import types, Dispatcher
-from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
+from aiogram import types
 from create_bot import dp, bot
 
+
 @dp.message_handler(commands=['start'])
 async def start_func(message: types.Message):
-    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('жмав') 
-    
+    await message.reply('похуй')
 
 
+@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

@@ -1,27 +1,28 @@
-from aiogram import types, Dispatcher
-from create_bot import dp, bot
+from aiogram import types
+from create_bot import dp
 import datetime
 
+
 @dp.message_handler(commands=["time"])
 async def send_time(message: types.Message):
-    new_year = datetime.datetime(datetime.now().year, 12, 31) #дата нового года
-    ct = datetime.datetime.now() #датавремя
+    new_year = datetime.datetime(2024, 12, 31)  # дата нового года
+    ct = datetime.datetime.now()  # дата/время
     cd = datetime.datetime.now().strftime("%d/%m/%Y")  # дата (д,м,г)
     ct1 = ct+datetime.timedelta(hours=15)  # +14 часов от сервера
-    ct2 = ct1.strftime('%H:%M') # форматирует датувремя просто во время(ч,м)
-    raznitsa = (new_year - ct).days #отнимает от нг текущее время и получаем разницу в днях
+    ct2 = ct1.strftime('%H:%M')  # форматирует дату/время просто во время(ч, м)
+    date_difference = (new_year - ct).days  # отнимает от нг текущее время и получаем разницу в днях
 
     days = ['день', 'дня', 'дней']
-    if raznitsa % 10 == 1 and raznitsa % 100 != 11:
+    if date_difference % 10 == 1 and date_difference % 100 != 11:
         p = 0
-    elif 2 <= raznitsa % 10 <= 4 and (raznitsa % 100 < 10 or raznitsa % 100 >= 20):
+    elif 2 <= date_difference % 10 <= 4 and (date_difference % 100 < 10 or date_difference % 100 >= 20):
         p = 1
     else:
         p = 2
-    num=(str(raznitsa) + ' ' + days[p])
+    num = (str(date_difference) + ' ' + days[p])
 
     await message.reply(f'Сегодня {cd} \nВремя: {ct2} \nДо Нового Года осталось {num}')
 
 
-def register_handlers_time(dp: Dispatcher):
-    dp.register_message_handler(send_time)
+# def register_handlers_time(dp: Dispatcher):
+#    dp.register_message_handler(send_time)

pipisa_functions/__init__.py

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

pipisa_functions/pipisa_time.py

@@ -1,88 +0,0 @@
-
-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,2 +0,0 @@
-aiogram
-python-arango

requirements.txt

@@ -0,0 +1,6 @@
+aiogram
+sqdb
+sqlalchemy
+graypy
+openai
+psycopg2

sqdb.py

@@ -1,36 +1,51 @@
 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.orm import sessionmaker, relationship
 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):
+class User(Base):
     __tablename__ = 'users'
 
     id = Column(Integer, primary_key=True)
     user_id = Column(Integer)
+    date_created = (Column(DateTime))
     user_fullname = Column(String)
+    date_updated = (Column(DateTime))
+    dick_requests = relationship("DickRequest", back_populates="user")
+    user_dicks = relationship("UserDick", back_populates="user")
+
+
+class UserDick(Base):
+    __tablename__ = 'user_dicks'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(Integer, ForeignKey("users.user_id"))
     dick_size = (Column(Integer))
-    datetimes = (Column(DateTime))
     chat_id = Column(BIGINT)
+    user = relationship("User", back_populates="user_dicks")
 
 
-
-
+class DickRequest(Base):
+    __tablename__ = 'dick_requests'
+    id = Column(Integer, primary_key=True)
+    size_change = Column(Integer)
+    user_id = Column(Integer, ForeignKey("users.user_id"))
+    user = relationship("User", back_populates="dick_requests")
+    date_changed = (Column(DateTime))
 
 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()
+
+# 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)