bring dick back.
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Simple_Not 2023-07-12 21:54:24 +10:00
parent b673b43c93
commit e93d2c0ef7
6 changed files with 183 additions and 166 deletions

8
bot.py
View File

@ -1,20 +1,12 @@
import logging import logging
import graypy
from aiogram import executor from aiogram import executor
from create_bot import dp from create_bot import dp
import handlers import handlers
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
logging.basicConfig(
level = logging.DEBUG,
handlers = [ graypy.GELFUDPHandler('localhost', 12201) ]
)
# pipisa.register_handlers_pipisa(dp) # pipisa.register_handlers_pipisa(dp)
# time_new_year.register_handlers_time(dp) # time_new_year.register_handlers_time(dp)

View File

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

View File

@ -1,7 +1,7 @@
from handlers import pipisa from handlers import pipisa
from handlers import davinci #from handlers import davinci #### токены рип
from handlers import time_new_year from handlers import time_new_year
from handlers import sendalarm #from handlers import sendalarm
from handlers import start_help from handlers import start_help

View File

@ -9,6 +9,7 @@ openai.api_key = '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'
async def send(message: types.Message): async def send(message: types.Message):
if 'тупица' in message.text.lower(): if 'тупица' in message.text.lower():
## TODO уточнить остатки токенов и денег
response = openai.Completion.create( response = openai.Completion.create(
model="text-davinci-003", model="text-davinci-003",
prompt=message.text[7:], prompt=message.text[7:],

View File

@ -1,181 +1,204 @@
from aiogram import types, Dispatcher from aiogram import types, Dispatcher
from sqlalchemy.orm.session import close_all_sessions #from sqlalchemy.orm.session import close_all_sessions
from create_bot import bot, dp from create_bot import bot, dp
from sqdb import user, session #from sqdb import user, session
import random import random
from random import randint from random import randint
import datetime import datetime
from dicktxt import ForReadDict from dicktxt import ForReadDict
from arango import ArangoClient
#from arangodb_base import dicks_collection
@dp.message_handler(commands=["dick"]) @dp.message_handler(commands=["dick"])
async def up_dick(message: types.Message): async def up_dick(message: types.Message):
# рандомайзер #global size_change, dick_plus, dick_minus
numb = randint(-6,10)
global size_change, dick_plus, dick_minus
dick_plus = ('<b>вырос</b>', '<b>подрос</b>', '<b>привстал</b>') arango_client = ArangoClient(hosts='https://arango.vdk2ch.ru')
dick_minus = ('<b>уменьшился</b>', '<b>откусила злая гадюка</b>', '<b>продали евреи, он стал дешевле</b>') pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527')
dicks_collection = pipisa_db.collection('dicks')
# рандомайзер
numb = randint(-9,10)
dick_plus = (
'<b>вырос</b>',
'<b>подрос</b>',
'<b>привстал</b>',
'<b>увеличился</b>',
'<b>преисполнился</b>'
)
dick_minus = (
'<b>неудачно потусил на сходке</b>',
'<b>уменьшился</b>',
'<b>откусила злая гадюка</b>',
'<b>продали евреи, он стал дешевле</b>'
)
if numb > 0: if numb > 0:
size_change = random.choice(dick_plus) + '🚀' 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) + ''
# работа с таблицей
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: else:
size_change = random.choice(dick_minus) + ''
try: try:
a = session.query(user).filter(user.user_id == message.from_user.id).first() # запрос на поиск в таблице #### Чекнуть есть ли юзер в базе
updetedDick = a.dick_size = (a.dick_size + numb) candidate_cursor = dicks_collection.find({'user_id': message.from_user.id}, skip=0, limit=1488)
a.datetimes = datetime.datetime.now()
session.flush() if candidate_cursor.count() > 0:
session.commit() #обновляет запись ## Если есть то нарастить и отправить сообщение
print(11111)
user = candidate_cursor.pop()
print(user)
print(user['datetimes'])
last_time = datetime.datetime.strptime(user['datetimes'], '%Y-%m-%d %H:%M:%S')
time_to_grow = datetime.datetime.now().replace(hour=14, minute=8, second=8, microsecond=0)
if (time_to_grow - last_time).days > 0:
# прошло больше суток
updetedDick = user["dick_size"] + numb
try:
# metadata = dicks_collection.update_match(
# {'user_id': message.from_user.id, 'chat_it': message.chat.id},
# {'dick_size': updetedDick, 'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
# )
metadata = dicks_collection.insert(
{
'_key': user['_key'],
'_id': user['_id'],
'_rev': user['_rev'],
'user_id': message.from_user.id,
'user_fullname': user['user_fullname'],
'dick_size': updetedDick,
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
'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() 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}, ваш <em>{randomword}</em> {size_change}\
на <b>{abs(numb)}</b> см!\n\ # на <b>{abs(numb)}</b> см!\n\
Теперь он равен <b>{updetedDick}</b> см! ') # Теперь он равен <b>{updetedDick}</b> см! ')
except: else:
session.rollback() await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер <b>{user["dick_size"]}</b> см! ')
print('ошибка в обновлении пиписы') print(f'пользователь @{message.from_user.username} уже крутил')
finally:
session.close_all()
print('пиписа обновлена')
else: else:
session.close() ## если нету, то создать
try: try:
numb = randint(1,10) novenkiy = {
'user_id': message.from_user.id,
'user_fullname': message.from_user.full_name,
'dick_size': numb,
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
'chat_id': message.chat.id
}
int_table = user(user_id = message.from_user.id, user_fullname = message.from_user.full_name, dick_size = numb, datetimes = datetime.datetime.now(),\ #### TODO проверить как-то по-умному что мы таки его добавили
chat_id = message.chat.id) metadata = dicks_collection.insert(novenkiy)
session.add(int_table)
session.flush()
session.commit()
session.close()
if numb > 0:
await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн показал головку 🚀\n\ await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн показал головку 🚀\n\
на <b>{abs(numb)}</b> см!\n\ на <b>{abs(numb)}</b> см!\n\
Теперь он равен <b>{numb}</b> см!') Теперь он равен <b>{numb}</b> см!')
except: else:
print('ошибка в добавлении нового пользователя') await bot.send_message(message.chat.id, f'@{message.from_user.username}, добро пожаловать в игру!🚀\n\
session.rollback() Ты стартуешь не с лучшей позиции, природа наградила тебя всего: <b>{abs(numb)}</b> см 😬')
finally:
session.close_all()
print('Успешно добавлен пользователь') print(f'Успешно добавлен пользователь @{message.from_user.username}')
except Exception as e1:
print('ошибка в добавлении нового пользователя')
print(e1)
except Exception as e:
print('ошибка DB в /dick')
print(e)
except:
session.rollback()
print('ошибка в основном коде')
finally:
session.close_all()
@dp.message_handler(commands=["topdick"]) @dp.message_handler(commands=["topdick"])
async def send_topchat(message: types.Message): async def send_topchat(message: types.Message):
arango_client = ArangoClient(hosts='https://arango.vdk2ch.ru')
pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527')
dicks_collection = pipisa_db.collection('dicks')
try: try:
changechat_id = session.query(user).filter(user.user_id == message.from_user.id).first() dicks = dicks_collection.find({'chat_id': message.chat.id}, skip=0, limit=1100)
chats = message.chat.id dicks = [d for d in dicks]
if changechat_id.chat_id != chats: except Exception as e:
changechat_id.chat_id = chats print('ошибка DB в /topdick')
session.commit() print(e)
session.close()
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: else:
session.close() for i in range(len(top_dicks)):
top = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).filter(user.chat_id == chats).limit(10).all() dickos += f' {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n'
delimiter = ': ' i += 1
# проверка на длину списка, если меньше limit, то: await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos)
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
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"]) @dp.message_handler(commands=["globaldick"])
async def send_global_top(message: types.Message): async def send_global_top(message: types.Message):
arango_client = ArangoClient(hosts='https://arango.vdk2ch.ru')
pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527')
dicks_collection = pipisa_db.collection('dicks')
try: try:
top_chat = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).limit(10).all() dicks = dicks_collection.all( )
dicks = [d for d in dicks]
except Exception as e:
print('ошибка DB в /glovaldick')
print(e)
delimiter = ': ' 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]
await bot.send_message(message.chat.id, f'🏆Топ 10 пипис в мире🏆\n\n\ dickos = ''
1. {delimiter.join(map(str, top_chat[0]))}\n\ emo = ['🏆','🚀','💫','🍆','🍌','🐍','🐎','🌭','🌶','👌']
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()
if len(top_dicks) == 0:
await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌')
else:
for i in range(len(top_dicks)):
dickos += f' {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n'
i += 1
# def register_handlers_pipisa(dp: Dispatcher): await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + dickos)
# dp.register_message_handler(up_dick)
# dp.register_message_handler(send_topchat)
# dp.register_message_handler(send_global_top)

View File

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