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, max_id=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) if max_id: posts_ret = [ p for p in posts_ret if p['msg_id'] != max_id] else: 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)