Как организовать сравнение даты в переменной с форматом эталона?

Question

[Vladimir Iusupov]

Пытаюсь сделать блок, в котором будет приходить сообщение с датой, далее сообщение в переменную и сравнение с форматом эталоном, далее если совпадает, то сообщение ок, если нет, то ошибка и скрипт повторяется. К сожалению не нашел решения, которое бы подошло. Решил написать сам, но не работает вторая часть.

import telebot
import config
from datetime import datetime
bot = telebot.TeleBot(config.TOKEN) # Токен бота из файла кофигурации 

@bot.message_handler(commands=['start'])
def date_start(message):
    bot.send_message(message.chat.id, 'Введите дату формата ДД-ММ-ГГГГ')
    bot.register_next_step_handler(message, date_step_1)


def date_step_1(message):
    res = bool(datetime.strptime(message.text, '%d-%m-%Y'))
    if res == True:
        bot.send_message(message.chat.id, 'Спасибо, дата введена')
    elif res == False:
        bot.send_message(message.chat.id, 'Не верный формат даты, необходимо ДД-ММ-ГГГГ')


bot.infinity_polling()

Comments

[Влад Григорьев]

Вопрос зачем?
Кажется вы пытаетесь решить не ту проблему.

Answer 1

[Vladimir Iusupov]

Посоветовали вот такую библиотеку использовать

import telebot
import config
from datetime import datetime
from dateutil.parser import parse

bot = telebot.TeleBot(config.TOKEN)

@bot.message_handler(commands=['start'])
def date_start(message):
    bot.send_message(message.chat.id, 'Введите дату формата ДД-ММ-ГГГГ')
    bot.register_next_step_handler(message, date_step_1)

def date_step_1(message):
    try:
        date = parse(message.text)
        bot.send_message(message.chat.id, 'Спасибо, дата введена')
    except ValueError:
        bot.send_message(message.chat.id, 'Не верный формат даты, необходимо ДД-ММ-ГГГГ')

bot.infinity_polling()

Answer 2

[Vindicar]

Документацию читать пробовал?

ValueError is raised if the date_string and format can’t be parsed by time.strptime() or if it returns a value which isn’t a time tuple

Если входная строка или строка формата не поддаются разбору вызовом time.strptime(), или если эта функция возвращает не кортеж с описанием даты-времени, функция выбрасывает исключение ValueError.

Учись обрабатывать исключения блоком try-except.
Впрочем, если судить вот по этому:

if res == True:
    ...
elif res == False:
    ...

Стоит вообще поучить язык, а потом уже браться за ботов. Это технически сработает, но выдаёт непонимание того, как работают логические переменные.
Правильнее будет

if res:
    ...
else:
    ...

Также задачу соответствия текста шаблону часто удобнее решать регулярными выражениями. Например:

import datetime
import re
# цифра, за ней может быть ещё одна, за ними "-", "/" или ".", 
# потом опять цифра, за ней может быть ещё одна, за ними "-", "/" или ".", 
# потом 2 или 4 цифры
regexp = re.compile(r'(\d\d?)[-/.](\d\d?)[-/.](\d{2}|\d{4})')
text = input('Введите дату в формате DD/MM/YYYY: ')
match = regexp.search(text)
if match is None:
    print('Вы ввели не дату, или ввели её в неправильном формате')
else:
    day, month, year = int(regexp.group(1)), int(regexp.group(2)), int(regexp.group(3))
    # тут нужна ещё проверка, так как пользователь может ввести 99/99/9999
    try:
        userdate = datetime.date(year, month, day)
    except ValueError:
        print('Это некорректная дата.')
    else:
        print('Дата корректна.')

Comments

[Александр]

я делал вот так

def parse_datetime_string(datetime_string: str) -> datetime | None:
    """
    Парсит строку с датой и временем и возвращает объект datetime
    """
    datetime_string = datetime_string.replace("/", "-").replace("\\", "").replace(".", "-")
    try:
        return datetime.strptime(f"{datetime_string}", '%d-%m-%y %H:%M')
    except ValueError:
        return None

[Vindicar]

Александр, ну сработает.
Но в чём заключается чёрная магия f"{datetime_string}", если datetime_string - это уже строка?

[Vladimir Iusupov]

Vindicar, к сожалению выдает ошибку, после ввода даты.

Traceback (most recent call last):
  File "/Users/vladimir/Library/Mobile Documents/com~apple~CloudDocs/Documents/Bot on python/1.py", line 12, in <module>
    day, month, year = int(regexp.groups(1)), int(regexp.groups(2)), int(regexp.groups(3))
TypeError: 'int' object is not callable

[Vladimir Iusupov]

Александр, тоже выдает ошибку

TypeError: unsupported operand type(s) for |: 'module' and 'NoneType'

[Александр]

Vindicar, чёрная магия f"{datetime_string}"
а что-то я уже и не помню. что-то хотел там сделать, но потом так и осталось. некогда рефачить

[Александр]

YaVladimir, можете вообще убрать -> datetime | None
это работает в питоне от версии 3.9 и выше
либо версию питона повыстьте, либо уберите, либо типизацию соответсвующую вашей версии питона сделайте

[Vladimir Iusupov]

Александр, версия 3.10.5

[Александр]

YaVladimir, аа, понял, что за ошибка. у вас не from datetime import datetime а просто import datetime

[Vladimir Iusupov]

Александр, спасибо, сработало, а как теперь поставить переменную? немного не понимаю

[Александр]

YaVladimir, что значит поставить переменную?

[Vladimir Iusupov]

Александр, я так понял, переменная тут datetime_string, из нее мы и берем значения. Но если я ставлю input до функции parse_datetime_string, ничего не происходит.

[Александр]

Vladimir Iusupov,
я не знаю как работают шаги в телеботе. попробую пояснить

# возможно здесь не хватает декоратора
def date_step_1(message):
    date = parse_datetime_string(message.text)
    if date is not None:
        bot.send_message(message.chat.id, 'Спасибо, дата введена')
        # здесь должен быть переход на следующий шаг
    else:
        bot.send_message(message.chat.id, 'Не верный формат даты, необходимо ДД-ММ-ГГГГ')
        # остаемся на этом же шаге

[Александр]

Vladimir Iusupov, только еще вам нужно переписать мою функцию просто для даты. у меня же там дата и время.
и соответствено поменять ее название на parse_date_string

[Vladimir Iusupov]

Александр, спасибо, я так понял, что я в начале ставлю Вашу функцию, потом ее использую у себя в телеботе, верно? Я был уверен в том, что мне нужно все сделать в рамках одной функции.

[Александр]

мою функцию можете хоть вынести в отдельный модуль и импортировать в нужное место.
еще раз скажу, что не разбираюсь в телеботе, работаю с aiogram. Там нет функций похожих на register_next_step_handler и я не очень понимаю, как здесь должно работать.
Свои предположения я написал выше комментариями к коду. почитайте комментарии