delete_callbacks.py

import logging
import re
from telegram import (
    InlineKeyboardButton,
    InlineKeyboardMarkup,
    Update,
    constants,
)
from telegram.ext import (
    ConversationHandler,
    ContextTypes,
)
from db_connection import uc
from bson.objectid import ObjectId
from dashboard_callbacks import format_cred_query

logging.basicConfig(
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
    level=logging.INFO,
)
# for monitoring activities
logger = logging.getLogger(__name__)


async def del_cred(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
    """delete retrieved credential"""
    query = update.callback_query
    context.user_data["del_cred_id"] = re.split("_", query.data, 1)[1]
    user = query.from_user
    logger.info(f"{user.first_name} is deleteing cred")

    for i in context.user_data["credentials"]:
        if str(i["_id"]) == context.user_data["del_cred_id"]:
            context.user_data["del_cred"] = i
            break

    await query.answer()
    keyboard = [
        [
            InlineKeyboardButton(
                "Yes, I am sure",
                callback_data=f"do_del_{context.user_data['del_cred_id']}",
            ),
            InlineKeyboardButton(
                "No, Stop",
                callback_data=f"return_del_{context.user_data['del_cred_id']}",
            ),
        ],
        [
            InlineKeyboardButton(
                "<< BACK",
                callback_data=f"return_del_{context.user_data['del_cred_id']}",
            ),
        ],
    ]
    reply_markup = InlineKeyboardMarkup(keyboard)
    await query.edit_message_text(
        f"You are about to delete <b>{context.user_data['del_cred']['name']}</b> credential",
        reply_markup=reply_markup,
        parse_mode=constants.ParseMode.HTML,
    )
    return ConversationHandler.END


async def do_del_cred(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
    """do deletion of retrieved credential"""
    query = update.callback_query
    context.user_data["del_cred_id"] = re.split("_", query.data, 2)[2]
    user = query.from_user
    logger.info(f"{user.first_name} is accepting to deleteing cred")

    for i in context.user_data["credentials"]:
        if str(i["_id"]) == context.user_data["del_cred_id"]:
            context.user_data["del_cred"] = i
            break

    await query.answer()
    uc.update_one(
        {"user_id": f"{context._user_id}"},
        {
            "$pull": {
                "credentials": {
                    "_id": ObjectId(context.user_data["del_cred_id"]),
                },
            },
        },
    )
    keyboard = [
        [
            InlineKeyboardButton(
                "Dashboard",
                callback_data="dash",
            ),
        ],
    ]
    reply_markup = InlineKeyboardMarkup(keyboard)
    await query.edit_message_text(
        f"Credential {context.user_data['del_cred']['name']} successfully deleted.",
        reply_markup=reply_markup,
    )

    context.user_data.clear()
    return ConversationHandler.END


async def return_del_cred(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
    """return almost deleted credential"""
    query = update.callback_query
    context.user_data["del_cred_id"] = re.split("_", query.data, 2)[2]
    user = query.from_user
    logger.info(f"{user.first_name} aborted deleteing cred")

    for i in context.user_data["credentials"]:
        if str(i["_id"]) == context.user_data["del_cred_id"]:
            context.user_data["del_cred"] = i
            break

    await query.answer()
    if context.user_data["del_cred"]["name"] in context.user_data["favourites"]:
        keyboard_fav = [
            [
                InlineKeyboardButton(
                    "EDIT",
                    callback_data=f"edit_{context.user_data['del_cred_id']}",
                ),
                InlineKeyboardButton(
                    "DELETE",
                    callback_data=f"del_{context.user_data['del_cred_id']}",
                ),
            ],
            [
                InlineKeyboardButton(
                    "REMOVE FROM FAV",
                    callback_data="defav_cred",
                ),
            ],
            [
                InlineKeyboardButton(
                    "DONE",
                    callback_data="done",
                ),
            ],
        ]
        reply_markup = InlineKeyboardMarkup(keyboard_fav)
        await query.edit_message_text(
            format_cred_query(context.user_data["del_cred"]),
            reply_markup=reply_markup,
            parse_mode=constants.ParseMode.HTML,
        )
        return ConversationHandler.END
    else:
        keyboard = [
            [
                InlineKeyboardButton(
                    "EDIT",
                    callback_data=f"edit_{context.user_data['del_cred_id']}",
                ),
                InlineKeyboardButton(
                    "DELETE",
                    callback_data=f"del_{context.user_data['del_cred_id']}",
                ),
            ],
            [
                InlineKeyboardButton(
                    "ADD TO FAV",
                    callback_data="fav_cred",
                ),
            ],
            [
                InlineKeyboardButton(
                    "DONE",
                    callback_data="done",
                ),
            ],
        ]
        reply_markup = InlineKeyboardMarkup(keyboard)
        await query.edit_message_text(
            format_cred_query(context.user_data["del_cred"]),
            reply_markup=reply_markup,
            parse_mode=constants.ParseMode.HTML,
        )
        return ConversationHandler.END