forked from Pavel_Durov/python_bot
fx
This commit is contained in:
Simple_Not
parent
1b2410afc4
commit
f20c90ea34
@ -1,238 +1,121 @@
|
||||
from aiogram import types, Dispatcher
|
||||
#from sqlalchemy.orm.session import close_all_sessions
|
||||
from aiogram import types
|
||||
from create_bot import bot, dp
|
||||
#from sqdb import user, session
|
||||
import random
|
||||
from random import randint
|
||||
import datetime
|
||||
from dicktxt import ForReadDict
|
||||
import datetime, logging
|
||||
from dicktxt import ForReadDict, WordsChange
|
||||
from pipisa_functions import pipisa_time
|
||||
|
||||
from global_conf import CONFIG
|
||||
|
||||
#### https://docs.python-arango.com/en/main/
|
||||
from arango import ArangoClient
|
||||
from db_logic import tops, user_stuff, postcleaner
|
||||
|
||||
|
||||
admins = CONFIG['telegram_admins_ids']
|
||||
|
||||
admins = [317051301, 226765676]
|
||||
|
||||
|
||||
@dp.message_handler(commands=["dick"])
|
||||
async def up_dick(message: types.Message):
|
||||
#global size_change, dick_plus, dick_minus
|
||||
|
||||
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')
|
||||
postcleaner.append_post_to_cleaning_sequence(message=message, type=f'COMMAND_CALL__DICK')
|
||||
await clean_posts(chat_id=message.chat.id, type='COMMAND_CALL__DICK', max_id=message.message_id)
|
||||
|
||||
#### TODO уплатить нолог с 100 см для самого нищего
|
||||
|
||||
# рандомайзер
|
||||
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>',
|
||||
'<b>схуднул</b>'
|
||||
)
|
||||
dick_stayed = (
|
||||
'<b>продвинулся</b>',
|
||||
'<b>наноуменьшился</b>',
|
||||
'<b>зафиксировал прибыль</b>'
|
||||
)
|
||||
result = randint(-9,10)
|
||||
|
||||
#### Чекнуть есть ли юзер в базе
|
||||
user = user_stuff.get_user(message)
|
||||
|
||||
if user:
|
||||
## Если есть то нарастить и отправить сообщение
|
||||
|
||||
if numb > 0:
|
||||
size_change = random.choice(dick_plus) + '🚀'
|
||||
elif numb == 0:
|
||||
size_change = random.choice(dick_stayed) + '🤨'
|
||||
else:
|
||||
size_change = random.choice(dick_minus) + '✂'
|
||||
if pipisa_time.rolltime(check_datetime=user['datetimes'], curr_time=datetime.datetime.now()):
|
||||
# пришло время крутить!!
|
||||
|
||||
updatedDick = user["dick_size"] + result
|
||||
user_stuff.update_dick_size(message, user, updatedDick)
|
||||
|
||||
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"] + numb
|
||||
try:
|
||||
dt = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||
metadata = dicks_collection.insert(
|
||||
{
|
||||
'_key': user['_key'], ### этот нужен чтобы апдейт прилетал тому же юзеру
|
||||
'user_id': user['user_id'],
|
||||
'user_fullname': message.from_user.full_name,
|
||||
'dick_size': updetedDick,
|
||||
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||
'attempts': user['attempts'] + 1,
|
||||
'chat_id': message.chat.id
|
||||
},
|
||||
overwrite_mode='update'
|
||||
)
|
||||
print(f'Успешно апдейтнули пипису @{message.from_user.username}')
|
||||
except Exception as e2:
|
||||
print(f'ошибка в обновлении пиписы @{message.from_user.username}')
|
||||
print(e2)
|
||||
|
||||
randomword = ForReadDict.RandomDick()
|
||||
|
||||
|
||||
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:
|
||||
await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер <b>{user["dick_size"]}</b> см! ')
|
||||
|
||||
try:
|
||||
metadata = dicks_collection.insert(
|
||||
{
|
||||
'_key': user['_key'], ### этот нужен чтобы апдейт прилетал тому же юзеру
|
||||
'user_id': user['user_id'] ,
|
||||
'user_fullname': message.from_user.full_name,
|
||||
'dick_size': user['dick_size'] ,
|
||||
'datetimes': user['datetimes'] ,
|
||||
'attempts': user['attempts'] + 1,
|
||||
'chat_id': message.chat.id
|
||||
},
|
||||
overwrite_mode='update'
|
||||
)
|
||||
print(f'Успешно апдейтнули попытку крутить пипису @{message.from_user.username}')
|
||||
except Exception as e2:
|
||||
print(f'ошибка в обновлении попытки крутить пипису @{message.from_user.username}')
|
||||
print(e2)
|
||||
|
||||
print(f'пользователь @{message.from_user.username} уже крутил')
|
||||
randomdickword = ForReadDict.RandomDick()
|
||||
change_phrase = WordsChange.ChangeWord(result)
|
||||
|
||||
await bot.send_message( message.chat.id,
|
||||
f'@{message.from_user.username}, ваш <em>{randomdickword}</em> {change_phrase} на <b>{abs(result)}</b> см!\nТеперь он равен <b>{updatedDick}</b> см! '
|
||||
)
|
||||
else:
|
||||
## если нету, то создать
|
||||
mmm = await bot.send_message( message.chat.id,
|
||||
f'@{message.from_user.username}, 🚫 вы уже крутили пипису, её размер <b>{user["dick_size"]}</b> см! '
|
||||
)
|
||||
user_stuff.update_attempts( message, user)
|
||||
|
||||
try:
|
||||
novenkiy = {
|
||||
'user_id': message.from_user.id,
|
||||
'user_fullname': message.from_user.full_name,
|
||||
'dick_size': abs(numb),
|
||||
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||
'attempts': 1,
|
||||
'chat_id': message.chat.id
|
||||
}
|
||||
postcleaner.append_post_to_cleaning_sequence(message=mmm, type='ALREADY_ROLLED')
|
||||
await clean_posts(chat_id=mmm.chat.id, type='ALREADY_ROLLED', max_id=mmm.message_id)
|
||||
else:
|
||||
## если нету, то создать
|
||||
user_stuff.store_new_user(message, result)
|
||||
|
||||
metadata = dicks_collection.insert(novenkiy, overwrite_mode='update')
|
||||
if result > 0:
|
||||
await bot.send_message( message.chat.id,
|
||||
f'@{message.from_user.username}, ваш писюн показал головку 🚀\nна <b>{abs(result)}</b> см!\nТеперь он равен <b>{result}</b> см!'
|
||||
)
|
||||
else:
|
||||
await bot.send_message( message.chat.id,
|
||||
f'@{message.from_user.username}, добро пожаловать в игру!🚀\nТы стартуешь не с лучшей позиции, природа наградила тебя всего <b>{result}</b> см 😬'
|
||||
)
|
||||
|
||||
|
||||
await bot.send_message(message.chat.id, f'@{message.from_user.username}, Добро пожаловать в игру! Ваш писюн показал головку 🚀\n\
|
||||
на <b>{abs(numb)}</b> см!\n\
|
||||
Теперь он равен <b>{abs(numb)}</b> см!')
|
||||
|
||||
|
||||
|
||||
print(f'Успешно добавлен пользователь @{message.from_user.username}')
|
||||
except Exception as e1:
|
||||
print('ошибка в добавлении нового пользователя')
|
||||
print(e1)
|
||||
|
||||
|
||||
except Exception as e:
|
||||
print('ошибка DB в /dick')
|
||||
print(e)
|
||||
else:
|
||||
await message.reply('Растить елду можно только в общих чатах!')
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@dp.message_handler(commands=["topdick"])
|
||||
@dp.message_handler(commands = ['topdick', 'globaldick'])
|
||||
async def send_topchat(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')
|
||||
top_ = message['text'].startswith('/topdick')
|
||||
glob_ = message['text'].startswith('/globaldick')
|
||||
|
||||
try:
|
||||
dicks = dicks_collection.find({'chat_id': message.chat.id}, skip=0, limit=1100)
|
||||
dicks = [d for d in dicks]
|
||||
except Exception as e:
|
||||
print('ошибка DB в /topdick')
|
||||
print(e)
|
||||
if top_:
|
||||
dickos = tops.get_tops( top_ = True, chat_id=message.chat.id )
|
||||
postcleaner.append_post_to_cleaning_sequence(message=message, type=f'COMMAND_CALL__TOPDICK')
|
||||
await clean_posts(chat_id=message.chat.id, type='COMMAND_CALL__TOPDICK', max_id=message.message_id)
|
||||
elif glob_:
|
||||
dickos = tops.get_tops( glob_ = True )
|
||||
postcleaner.append_post_to_cleaning_sequence(message=message, type=f'COMMAND_CALL__GLOBALDICK')
|
||||
await clean_posts(chat_id=message.chat.id, type='COMMAND_CALL__GLOBALDICK', max_id=message.message_id)
|
||||
else:
|
||||
logging.error('вызывают хз что!')
|
||||
|
||||
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:
|
||||
if not dickos:
|
||||
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
|
||||
|
||||
await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos)
|
||||
if top_:
|
||||
await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos)
|
||||
elif glob_:
|
||||
await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + dickos)
|
||||
else:
|
||||
await message.reply('Работает только в общих чатах!\n'\
|
||||
'Вы мсжете посмотреть топ по миру /globaldick')
|
||||
'Вы можете посмотреть топ по миру /globaldick')
|
||||
|
||||
|
||||
|
||||
|
||||
@dp.message_handler(commands=["globaldick"])
|
||||
async def send_global_top(message: types.Message):
|
||||
|
||||
|
||||
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]
|
||||
|
||||
dickos = ''
|
||||
emo = ['🏆','🚀','💫','🍆','🍌','🐍','🐎','🌭','🌶','👌']
|
||||
|
||||
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
|
||||
|
||||
await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + dickos)
|
||||
async def clean_posts(chat_id, type=None, max_id=None):
|
||||
psts = postcleaner.get_posts_to_be_removed(chat_id, type, max_id)
|
||||
|
||||
for p in psts:
|
||||
postcleaner.del_post(chat_id=p['chat_id'], msg_id=p['msg_id'])
|
||||
|
||||
#### TODO проверить админит ли бот
|
||||
|
||||
try:
|
||||
await bot.delete_message( chat_id=p['chat_id'], message_id=p['msg_id'], )
|
||||
except Exception as e:
|
||||
logging.error('ошибка удаления поста-на-удаление')
|
||||
logging.error(e)
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user