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

.drone.yml

@@ -20,6 +20,12 @@ steps:
   - cd /usr/share/$DRONE_REPO_NAME
   - git pull
   - chown -R jack:jack .
+  when:
+    branch:
+    - main
 - name: restart systemd unit
   commands:
   - systemctl restart jack_bot
+  when:
+    branch:
+    - main

bot.py

@@ -12,8 +12,8 @@ logging.basicConfig(level=logging.DEBUG)
 
 
 logging.basicConfig(
-    level    = logging.DEBUG,
+    level=logging.DEBUG,
-    handlers = [ graypy.GELFUDPHandler('localhost', 12201) ]
+    handlers=[graypy.GELFUDPHandler('localhost', 12201)]
 )
 
 # pipisa.register_handlers_pipisa(dp)
@@ -25,7 +25,3 @@ logging.basicConfig(
 if __name__ == '__main__':
     executor.start_polling(dp, skip_updates=True)
 
-
-
-
-

handlers/davinci.py

@@ -1,10 +1,11 @@
 import openai
-from aiogram import types, Dispatcher
+from aiogram import types
-from create_bot import dp, bot
+from create_bot import dp
 
 token = '5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc'
 openai.api_key = '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'
 
+
 @dp.message_handler()
 async def send(message: types.Message):
     if 'тупица' in message.text.lower():

handlers/pipisa.py

@@ -1,42 +1,76 @@
-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
+from sqdb import User, session, DickRequest, UserDick
 import random
+import traceback
 from random import randint
 import datetime
 from dicktxt import ForReadDict
 
+size_change = None
+dick_plus = None
+dick_minus = None
+
+
+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
+
+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):
+async def up_dick(message: types.Message):  # рандомайзер
-# рандомайзер
+
-    numb = randint(-6,10)
+    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
+    # if numb = 0 size_change = random.choice(const)  TODO
-    else: size_change = random.choice(dick_minus) + '✂'
+    else:
+        size_change = random.choice(dick_minus) + '✂'
 
 
 # работа с таблицей
     try:
-        b = session.query(user).filter(user.user_id == message.from_user.id).first()
+        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.datetimes
+            last_time = b.date_updated
             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},'
-                    
+                                                            f'🚫 вы уже крутили пипису, '
-                    await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер <b>{current_dick}</b> см! ')
+                                                            f'ее размер <b>{current_dick}</b> см! ')
-                except:
+                except Exception as e:
-                    print('ошибка в if diff <=0')
+                    print('ошибка в if diff <=0 ' + "\n".join(traceback.format_tb(e.__traceback__)))
                     session.rollback()   
                 finally: 
                     session.close_all()
@@ -44,130 +78,118 @@ async def up_dick(message: types.Message):
                     
             else:
                 try:
-                    a = session.query(user).filter(user.user_id == message.from_user.id).first() # запрос на поиск в таблице
+                    usr = session.query(User)
-                    updetedDick = a.dick_size = (a.dick_size + numb)
+                    a = usr.filter(User.user_id == message.from_user.id).first()  # запрос на поиск в таблице
-                    a.datetimes = datetime.datetime.now()
+                    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() #обновляет запись
+                    session.commit()  # обновляет запись
                     randomword = ForReadDict.RandomDick()
                 
-                    await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш <em>{randomword}</em> {size_change}\
+                    await bot.send_message(message.chat.id, f'@{message.from_user.username}, '
-на <b>{abs(numb)}</b> см!\n\
+                                                            f'ваш <em>{randomword}</em> {size_change}'
-Теперь он равен <b>{updetedDick}</b> см! ')
+                                                            f'на <b>{abs(numb)}</b> см!\n'
-                except:
+                                                            f'Теперь он равен <b>{updated_dick}</b> см! ')
+                except Exception as e:
                     session.rollback()
-                    print('ошибка в обновлении пиписы')
+                    print('ошибка в обновлении пиписы, трейсбек-бек-бек: ' +
+                          "\n".join(traceback.format_tb(e.__traceback__)))
                 finally:
                     session.close_all()
                     print('пиписа обновлена')
 
-                    
-
         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(),\
+                int_table = User(user_id=message.from_user.id,
-            chat_id = message.chat.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}, '
-                await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн {size_change}\n\
+                                                        f'ваш писюн {size_change}\n'
-на <b>{abs(numb)}</b> см!\n\
+                                                        f'на <b>{abs(numb)}</b> см!\n'
-Теперь он равен <b>{numb}</b> см!')
+                                                        f'Теперь он равен <b>{numb}</b> см!')
-            except:
+            except Exception as e:
-                print('ошибка в добавлении нового пользователя')
+                print('ошибка в добавлении нового пользователя, трейсбек: ' +
+                      '\n'.join(traceback.format_tb(e.__traceback__)))
                 session.rollback()
             finally:
                 session.close_all()
                 print('Успешно добавлен пользователь')
         
-    except:
+    except Exception as e:
         session.rollback()
-        print('ошибка в основном коде')
+        print('ошибка в основном коде, трейсбек:' + '\n'.join(traceback.format_tb(e.__traceback__)))
     finally:
         session.close_all()
         
     
 @dp.message_handler(commands=["topdick"])
-async def send_topchat(message: types.Message):
+async def send_top_chat(message: types.Message):
     try:
-        changechat_id = session.query(user).filter(user.user_id == message.from_user.id).first()
+        top = []
         chats = message.chat.id
-        if changechat_id.chat_id != chats:
+        changechat_id = session.query(DickRequest).filter(DickRequest.user_id == message.from_user.id,
-            changechat_id.chat_id = chats
+                                                          DickRequest.chat_id == chats).first()
-            session.commit()
+        dicks = session.query(DickRequest)\
-            session.close()
+            .filter(DickRequest.chat_id == chats)\
-        else:
+            .limit(10).all()\
-            session.close()
+            .order_by(DickRequest.dick_size.desc())
-            top = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).filter(user.chat_id == chats).limit(10).all()
+        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, то:
 
-        sizelen = len(top)
+        size_len = len(top)
-        if sizelen < 10:
+        if size_len < 10:
-            lenminus = 10 - sizelen
+            len_minus = 10 - size_len
-            top.extend(('-'*lenminus))
+            top.extend([('-'*len_minus),""])
-            await bot.send_message(message.chat.id, f'🏆Топ 10 бубылд чата🏆\n\n\
+            await bot.send_message(message.chat.id, top_dick_message(delimiter, top))
-    🚀 {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
 
         else:
             
-            await bot.send_message(message.chat.id, f'🏆Топ 10 бубылд чата🏆\n\n\
+            await bot.send_message(message.chat.id, top_dick_message(delimiter, top))
-    🚀 {delimiter.join(map(str, top[0]))}\n\
+    except Exception as e:
-    💫 {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')
+        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):
     try:
-        top_chat = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).limit(10).all()
+        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, top_dick_message(delimiter, top))
-        await bot.send_message(message.chat.id, f'🏆Топ 10 пипис в мире🏆\n\n\
+    except Exception as e:
-    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')
+        print('ошибка в /globaldick, трейсбэк:\n' + '\n'.join(traceback.format_tb(e.__traceback__)))
     finally:
         session.close_all()
 
@@ -176,4 +198,3 @@ async def send_global_top(message: types.Message):
 #     dp.register_message_handler(up_dick)
 #     dp.register_message_handler(send_topchat)
 #     dp.register_message_handler(send_global_top)
-

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 = session.query(user.chat_id).distinct()
+                chats_id = session.query(User.chat_id).distinct()
                 for row in chats_id:
                     try:
                         await bot.send_message(row[0], text)
-                    except:
+                    except Exception as e:
-                        print('вероятно бота нет в чате')
+                        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')
+                print('error sendall, traceback:\n' +
-            finally: session.close_all()
+                      "\n".join(traceback.format_tb(e.__traceback__)))
+            finally:
+                session.close_all()
 
 
 # def register_handlers_test(dp: Dispatcher):

handlers/start_help.py

@@ -1,22 +1,28 @@
-from aiogram import types, Dispatcher
+from aiogram import types
 from create_bot import dp, bot
 
+
 @dp.message_handler(commands=['start'])
 async def start_func(message: types.Message):
     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)
+    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')
+        await bot.send_photo(message.chat.id,
+                             photo='https://www.meme-arsenal.com/memes/ecebd2339c7eab40e09874bd86a38a09.jpg')
 
 
-def register_handlers_StartHelp(dp: Dispatcher):
+# def register_handlers_StartHelp(dp: Dispatcher):
-    dp.register_message_handler(start_func)
+#     dp.register_message_handler(start_func)
-    dp.register_message_handler(send_faggot)
+#     dp.register_message_handler(send_faggot)
-    dp.register_message_handler(send_image)
+#     dp.register_message_handler(send_image)

handlers/time_new_year.py

@@ -1,27 +1,28 @@
-from aiogram import types, Dispatcher
+from aiogram import types
-from create_bot import dp, bot
+from create_bot import dp
 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(2022, 12, 31)  # дата нового года
-    ct = datetime.datetime.now() #датавремя
+    ct = datetime.datetime.now()  # дата/время
-    cd = datetime.datetime.now().strftime("%d/%m/%Y") #дата (д,м,г)
+    cd = datetime.datetime.now().strftime("%d/%m/%Y")  # дата (д,м,г)
     ct1 = ct+datetime.timedelta(hours=15)  # +14 часов от сервера
-    ct2 = ct1.strftime('%H:%M') # форматирует датувремя просто во время(ч,м)
+    ct2 = ct1.strftime('%H:%M')  # форматирует дату/время просто во время(ч, м)
-    raznitsa = (new_year - ct).days #отнимает от нг текущее время и получаем разницу в днях
+    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):
+# def register_handlers_time(dp: Dispatcher):
-    dp.register_message_handler(send_time)
+#    dp.register_message_handler(send_time)

requirements.txt

@@ -2,3 +2,4 @@ aiogram
 sqdb
 sqlalchemy
 graypy
+openai

sqdb.py

@@ -1,36 +1,52 @@
 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")
+    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="dick_requests")
 
 
-
+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)