teleexit/message/edit.py

import re
from datetime import datetime

from telegram import Update, ParseMode
from telegram.ext import CallbackContext

from callbackQuery.create import address_re
from callbackQuery.data import data
from main import local, messages, database

rex = {
    "first_name": re.compile(r"^([a-zA-Z]| )+$"),
    "last_name": re.compile(r"^([a-zA-Z]| )+$"),
    "birth_date": re.compile(r"^[0-9]{1,2}/[0-9]{1,2}/[0-9]{4}$"),
    "birth_city": re.compile(r"^[a-zA-Z]+(?:[\s-][a-zA-Z]+)*$"),
    "address": address_re
}


def check_date(date) -> bool:
    try:
        datetime.strptime(date, "%d/%m/%Y")
    except:
        return False
    else:
        return True


def edit(update: Update, context: CallbackContext, data_edit: str):
    name = data_edit.replace("edit_", "")
    if not rex[name].fullmatch(update.message.text) or\
            (name == "birth_date" and not check_date(update.message.text)):
        context.bot.send_message(chat_id=update.effective_chat.id, text=f"Invalid value for `{local[name]}` \!",
                                 parse_mode=ParseMode.MARKDOWN_V2)
    else:
        del messages[update.effective_chat.id][update["_effective_user"]["id"]]
        database[update["_effective_user"]["id"]][name] = update.message.text
        data(update, context)