This commit is contained in:
parent
14c238ec2f
commit
ee27dbfd7b
@ -5,6 +5,7 @@ databaso:
|
|||||||
pass: stolendick527
|
pass: stolendick527
|
||||||
base: pipisa
|
base: pipisa
|
||||||
collection: dicks
|
collection: dicks
|
||||||
|
posts_removal_collection: posts_removal
|
||||||
|
|
||||||
telegram_token: 5724991559:AAEuLvpLsgP6LHRGMSyFtQLlR5qPQUO4b_w
|
telegram_token: 5724991559:AAEuLvpLsgP6LHRGMSyFtQLlR5qPQUO4b_w
|
||||||
telegram_admins_ids:
|
telegram_admins_ids:
|
||||||
|
@ -7,6 +7,7 @@ from arango import ArangoClient
|
|||||||
|
|
||||||
|
|
||||||
def get_dicks_collection():
|
def get_dicks_collection():
|
||||||
|
try:
|
||||||
arango_client = ArangoClient(hosts=CONFIG['databaso']['host'] )
|
arango_client = ArangoClient(hosts=CONFIG['databaso']['host'] )
|
||||||
pipisa_db = arango_client.db(
|
pipisa_db = arango_client.db(
|
||||||
CONFIG['databaso']['base'],
|
CONFIG['databaso']['base'],
|
||||||
@ -16,3 +17,22 @@ def get_dicks_collection():
|
|||||||
dicks_collection = pipisa_db.collection(CONFIG['databaso']['collection'])
|
dicks_collection = pipisa_db.collection(CONFIG['databaso']['collection'])
|
||||||
|
|
||||||
return dicks_collection
|
return dicks_collection
|
||||||
|
except Exception as e:
|
||||||
|
print('ошибка DB при взятии коллекции пипис')
|
||||||
|
print(e)
|
||||||
|
|
||||||
|
|
||||||
|
def get_posts_removal_collection():
|
||||||
|
try:
|
||||||
|
arango_client = ArangoClient(hosts=CONFIG['databaso']['host'] )
|
||||||
|
pipisa_db = arango_client.db(
|
||||||
|
CONFIG['databaso']['base'],
|
||||||
|
username=CONFIG['databaso']['user'],
|
||||||
|
password=CONFIG['databaso']['pass']
|
||||||
|
)
|
||||||
|
posts_removal_collection = pipisa_db.collection(CONFIG['databaso']['posts_removal_collection'])
|
||||||
|
|
||||||
|
return posts_removal_collection
|
||||||
|
except Exception as e:
|
||||||
|
print('ошибка DB при взятии коллекции постов-на-удаление')
|
||||||
|
print(e)
|
||||||
|
71
db_logic/postcleaner.py
Normal file
71
db_logic/postcleaner.py
Normal file
@ -0,0 +1,71 @@
|
|||||||
|
from db_logic import collections
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
|
||||||
|
def append_post_to_cleaning_sequence(message, type=None):
|
||||||
|
try:
|
||||||
|
novenkiy = {
|
||||||
|
'msg_id': message.message_id,
|
||||||
|
'chat_id': message.chat.id,
|
||||||
|
'type': type,
|
||||||
|
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||||
|
}
|
||||||
|
metadata = collections.get_posts_removal_collection().insert(novenkiy, overwrite_mode='update')
|
||||||
|
|
||||||
|
print(f'Успешно добавлен пост на удаление')
|
||||||
|
except Exception as e2:
|
||||||
|
print(f'ошибка DB :: добавление нового поста на удаление')
|
||||||
|
print(e2)
|
||||||
|
|
||||||
|
|
||||||
|
def get_posts_to_be_removed(chat_id, type=None):
|
||||||
|
# собираем
|
||||||
|
try:
|
||||||
|
#posts = [p for p in collections.get_posts_removal_collection().all( ) ]
|
||||||
|
|
||||||
|
if type:
|
||||||
|
posts = [p for p in collections.get_posts_removal_collection().find({'chat_id': chat_id, 'type': type}, skip=0, limit=1100) ]
|
||||||
|
else:
|
||||||
|
posts = [p for p in collections.get_posts_removal_collection().find({'chat_id': chat_id}, skip=0, limit=1100) ]
|
||||||
|
|
||||||
|
posts_ret = []
|
||||||
|
for p in posts:
|
||||||
|
last_time = datetime.datetime.strptime(p['datetimes'], '%Y-%m-%d %H:%M:%S')
|
||||||
|
timediff = (datetime.datetime.now() - last_time).total_seconds()
|
||||||
|
|
||||||
|
if timediff > 60:
|
||||||
|
posts_ret.append(p)
|
||||||
|
|
||||||
|
print(posts_ret)
|
||||||
|
posts_ret = [ p for p in posts_ret if p['msg_id'] != max([pp['msg_id'] for pp in posts_ret])]
|
||||||
|
### оставлять последний пост
|
||||||
|
|
||||||
|
|
||||||
|
return posts_ret
|
||||||
|
except Exception as e:
|
||||||
|
print('ошибка DB :: получение постов на удаление')
|
||||||
|
print(e)
|
||||||
|
|
||||||
|
|
||||||
|
def del_post(msg_id, chat_id):
|
||||||
|
# удаляем из базы
|
||||||
|
try:
|
||||||
|
|
||||||
|
candidate_cursor = collections.get_posts_removal_collection().find(
|
||||||
|
{
|
||||||
|
'msg_id': msg_id,
|
||||||
|
'chat_id': chat_id,
|
||||||
|
},
|
||||||
|
skip = 0,
|
||||||
|
limit = 1488
|
||||||
|
)
|
||||||
|
if candidate_cursor.count() > 0:
|
||||||
|
pp = candidate_cursor.pop()
|
||||||
|
else:
|
||||||
|
pp = None
|
||||||
|
|
||||||
|
collections.get_posts_removal_collection().delete(pp)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print('ошибка DB :: удаление поста на удаление')
|
||||||
|
print(e)
|
@ -1,6 +1,4 @@
|
|||||||
from global_conf import CONFIG
|
|
||||||
from db_logic import collections
|
from db_logic import collections
|
||||||
|
|
||||||
import datetime
|
import datetime
|
||||||
|
|
||||||
|
|
||||||
|
@ -8,7 +8,7 @@ from pipisa_functions import pipisa_time
|
|||||||
|
|
||||||
from global_conf import CONFIG
|
from global_conf import CONFIG
|
||||||
|
|
||||||
from db_logic import collections, tops, user_stuff
|
from db_logic import tops, user_stuff, postcleaner
|
||||||
|
|
||||||
|
|
||||||
admins = CONFIG['telegram_admins_ids']
|
admins = CONFIG['telegram_admins_ids']
|
||||||
@ -20,6 +20,10 @@ async def up_dick(message: types.Message):
|
|||||||
|
|
||||||
if message.from_user.id in admins or message.chat.type != 'private':
|
if message.from_user.id in admins or message.chat.type != 'private':
|
||||||
|
|
||||||
|
|
||||||
|
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')
|
||||||
|
|
||||||
#### TODO уплатить нолог с 100 см для самого нищего
|
#### TODO уплатить нолог с 100 см для самого нищего
|
||||||
|
|
||||||
# рандомайзер
|
# рандомайзер
|
||||||
@ -44,10 +48,13 @@ async def up_dick(message: types.Message):
|
|||||||
f'@{message.from_user.username}, ваш <em>{randomdickword}</em> {change_phrase} на <b>{abs(result)}</b> см!\nТеперь он равен <b>{updatedDick}</b> см! '
|
f'@{message.from_user.username}, ваш <em>{randomdickword}</em> {change_phrase} на <b>{abs(result)}</b> см!\nТеперь он равен <b>{updatedDick}</b> см! '
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
await bot.send_message( message.chat.id,
|
mmm = await bot.send_message( message.chat.id,
|
||||||
f'@{message.from_user.username}, 🚫 вы уже крутили пипису, её размер <b>{user["dick_size"]}</b> см! '
|
f'@{message.from_user.username}, 🚫 вы уже крутили пипису, её размер <b>{user["dick_size"]}</b> см! '
|
||||||
)
|
)
|
||||||
user_stuff.update_attempts( message, user)
|
user_stuff.update_attempts( message, user)
|
||||||
|
|
||||||
|
postcleaner.append_post_to_cleaning_sequence(message=mmm, type='ALREADY_ROLLED')
|
||||||
|
await clean_posts(chat_id=mmm.chat.id, type='ALREADY_ROLLED')
|
||||||
else:
|
else:
|
||||||
## если нету, то создать
|
## если нету, то создать
|
||||||
user_stuff.store_new_user(message, result)
|
user_stuff.store_new_user(message, result)
|
||||||
@ -61,10 +68,6 @@ async def up_dick(message: types.Message):
|
|||||||
f'@{message.from_user.username}, добро пожаловать в игру!🚀\nТы стартуешь не с лучшей позиции, природа наградила тебя всего <b>{result}</b> см 😬'
|
f'@{message.from_user.username}, добро пожаловать в игру!🚀\nТы стартуешь не с лучшей позиции, природа наградила тебя всего <b>{result}</b> см 😬'
|
||||||
)
|
)
|
||||||
|
|
||||||
#### для всех отложенных сообщений
|
|
||||||
## проверить если прошло ?пять минут? с публикации
|
|
||||||
## попробовать удалить такой пост и затереть его с коллекции
|
|
||||||
## оставлять последние в чате топ- и глобал-дик, сохранять последние сообщения о приросте, сохранять последний вызов /dick и иной команды
|
|
||||||
else:
|
else:
|
||||||
await message.reply('Растить елду можно только в общих чатах!')
|
await message.reply('Растить елду можно только в общих чатах!')
|
||||||
|
|
||||||
@ -79,11 +82,16 @@ async def send_topchat(message: types.Message):
|
|||||||
|
|
||||||
if top_:
|
if top_:
|
||||||
dickos = tops.get_tops( top_ = True, chat_id=message.chat.id )
|
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')
|
||||||
elif glob_:
|
elif glob_:
|
||||||
dickos = tops.get_tops( glob_ = True )
|
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')
|
||||||
else:
|
else:
|
||||||
print('вызывают хз что!')
|
print('вызывают хз что!')
|
||||||
|
|
||||||
|
|
||||||
if not dickos:
|
if not dickos:
|
||||||
await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌')
|
await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌')
|
||||||
else:
|
else:
|
||||||
@ -94,3 +102,15 @@ async def send_topchat(message: types.Message):
|
|||||||
else:
|
else:
|
||||||
await message.reply('Работает только в общих чатах!\n'\
|
await message.reply('Работает только в общих чатах!\n'\
|
||||||
'Вы можете посмотреть топ по миру /globaldick')
|
'Вы можете посмотреть топ по миру /globaldick')
|
||||||
|
|
||||||
|
|
||||||
|
async def clean_posts(chat_id, type=None):
|
||||||
|
psts = postcleaner.get_posts_to_be_removed(chat_id, type)
|
||||||
|
print(psts)
|
||||||
|
|
||||||
|
for p in psts:
|
||||||
|
postcleaner.del_post(chat_id=p['chat_id'], msg_id=p['msg_id'])
|
||||||
|
|
||||||
|
#### TODO проверить админит ли бот
|
||||||
|
await bot.delete_message( chat_id=p['chat_id'], message_id=p['msg_id'], )
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user