Pavel_Durov/python_bot
Pavel_Durov/python_bot
1
1
Fork 1
Code Issues 2 Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: Pavel_Durov:e93d2c0ef79e1be468aa62f7d0ad48ed8b665d92
Branches Tags
Pavel_Durov:main
Pavel_Durov:0.1.0-dev
..
compare: Pavel_Durov:dea7805dff9a542e9c37b71e51dd28000add1372
Branches Tags
Pavel_Durov:main
Pavel_Durov:0.1.0-dev

No commits in common. "e93d2c0ef79e1be468aa62f7d0ad48ed8b665d92" and "dea7805dff9a542e9c37b71e51dd28000add1372" have entirely different histories.
e93d2c0ef7 ... dea7805dff

6 changed files with 166 additions and 183 deletions
Show Stats Expand all files Collapse all files

8
bot.py
View File

@ -1,12 +1,20 @@
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)

5
create_bot.py
View File

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

4
handlers/__init__.py
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

17
handlers/davinci.py
View File

@ -9,16 +9,15 @@ 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:],
temperature=0.7, temperature=0.7,
max_tokens=1000, max_tokens=1000,
top_p=1.0, top_p=1.0,
frequency_penalty=0.0, frequency_penalty=0.0,
presence_penalty=0.6, presence_penalty=0.6,
stop=["сброс"] stop=["сброс"]
) )
await message.reply(response['choices'][0]['text']) await message.reply(response['choices'][0]['text'])

311
handlers/pipisa.py
View File

@ -1,204 +1,181 @@
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
arango_client = ArangoClient(hosts='https://arango.vdk2ch.ru') dick_plus = ('<b>вырос</b>', '<b>подрос</b>', '<b>привстал</b>')
pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527') dick_minus = ('<b>уменьшился</b>', '<b>откусила злая гадюка</b>', '<b>продали евреи, он стал дешевле</b>')
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: else: size_change = random.choice(dick_minus) + '✂'
size_change = random.choice(dick_minus) + '✂'
# работа с таблицей
try: try:
#### Чекнуть есть ли юзер в базе b = session.query(user).filter(user.user_id == message.from_user.id).first()
candidate_cursor = dicks_collection.find({'user_id': message.from_user.id}, skip=0, limit=1488) if b:
last_time = b.datetimes
if candidate_cursor.count() > 0: now_time = datetime.datetime.now()
## Если есть то нарастить и отправить сообщение diff = now_time.day - last_time.day
print(11111)
user = candidate_cursor.pop() if abs(diff) <= 0:
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: try:
# metadata = dicks_collection.update_match( current_dick = b.dick_size
# {'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")}
# ) await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер <b>{current_dick}</b> см! ')
metadata = dicks_collection.insert( except:
{ print('ошибка в if diff <=0')
'_key': user['_key'], session.rollback()
'_id': user['_id'], finally:
'_rev': user['_rev'], session.close_all()
'user_id': message.from_user.id, print('пользователь уже крутил')
'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()
await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш <em>{randomword}</em> {size_change}\
# на <b>{abs(numb)}</b> см!\n\
# Теперь он равен <b>{updetedDick}</b> см! ')
else: else:
await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер <b>{user["dick_size"]}</b> см! ') try:
print(f'пользователь @{message.from_user.username} уже крутил') a = session.query(user).filter(user.user_id == message.from_user.id).first() # запрос на поиск в таблице
updetedDick = a.dick_size = (a.dick_size + numb)
else: a.datetimes = datetime.datetime.now()
## если нету, то создать session.flush()
session.commit() #обновляет запись
try: randomword = ForReadDict.RandomDick()
novenkiy = {
'user_id': message.from_user.id, await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш <em>{randomword}</em> {size_change}\
'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
}
#### TODO проверить как-то по-умному что мы таки его добавили
metadata = dicks_collection.insert(novenkiy)
if numb > 0:
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>{updetedDick}</b> см! ')
else: except:
await bot.send_message(message.chat.id, f'@{message.from_user.username}, добро пожаловать в игру!🚀\n\ session.rollback()
Ты стартуешь не с лучшей позиции, природа наградила тебя всего: <b>{abs(numb)}</b> см 😬') print('ошибка в обновлении пиписы')
finally:
session.close_all()
print('пиписа обновлена')
print(f'Успешно добавлен пользователь @{message.from_user.username}') else:
except Exception as e1: session.close()
try:
numb = randint(1,10)
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)
session.add(int_table)
session.flush()
session.commit()
session.close()
await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн показал головку 🚀\n\
на <b>{abs(numb)}</b> см!\n\
Теперь он равен <b>{numb}</b> см!')
except:
print('ошибка в добавлении нового пользователя') print('ошибка в добавлении нового пользователя')
print(e1) session.rollback()
finally:
session.close_all()
except Exception as e: print('Успешно добавлен пользователь')
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:
dicks = dicks_collection.find({'chat_id': message.chat.id}, skip=0, limit=1100) changechat_id = session.query(user).filter(user.user_id == message.from_user.id).first()
dicks = [d for d in dicks] chats = message.chat.id
except Exception as e: if changechat_id.chat_id != chats:
print('ошибка DB в /topdick') changechat_id.chat_id = chats
print(e) session.commit()
session.close()
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 = ': '
top_dicks = sorted(dicks, key=lambda dick: dick['dick_size'], reverse=True) # проверка на длину списка, если меньше limit, то:
dick_limit = min(len(top_dicks), 10)
top_dicks = top_dicks[:dick_limit]
dickos = ''
emo = ['🏆','🚀','🍆','🍌','🐍','🐎','🌭','🌶','👌','💩']
if len(top_dicks) == 0: sizelen = len(top)
await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌') if sizelen < 10:
else: lenminus = 10 - sizelen
for i in range(len(top_dicks)): top.extend(('-'*lenminus))
dickos += f' {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n' await bot.send_message(message.chat.id, f'🏆Топ 10 бубылд чата🏆\n\n\
i += 1 🚀 {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
await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos) 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:
dicks = dicks_collection.all( ) top_chat = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).limit(10).all()
dicks = [d for d in dicks]
except Exception as e: delimiter = ': '
print('ошибка DB в /glovaldick')
print(e)
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()
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: # def register_handlers_pipisa(dp: Dispatcher):
await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌') # dp.register_message_handler(up_dick)
else: # dp.register_message_handler(send_topchat)
for i in range(len(top_dicks)): # dp.register_message_handler(send_global_top)
dickos += f' {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n'
i += 1
await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + dickos)

4
requairements.txt
View File

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