@kicherov_maxim

Как красиво использовать strptime с различными форматами?

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

Тест
@mark.parametrize(
    'arg, result',
    [
        (date(1996, 9, 1), date(1996, 9, 1)),
        (datetime(1996, 9, 1), date(1996, 9, 1)),
        ('01.09.2004', date(2004, 9, 1)),
        ('01/09/2004', date(2004, 9, 1)),
        ('2004-09-1', date(2004, 9, 1)),
    ],
)
def test_convert_date(arg, result):
    assert_that(convert_date(arg), is_(result))


Функция
def convert_date(value: date | datetime | str) -> date:
    if isinstance(value, datetime):
        return value.date()
    if isinstance(value, date):
        return value
    try:
        return datetime.strptime(value, '%d.%m.%Y').date()
    except ValueError:
        pass
    try:
        return datetime.strptime(value, '%d/%m/%Y').date()
    except ValueError:
        pass
    try:
        return datetime.strptime(value, '%Y-%m-%d').date()
    except ValueError as e:
        raise ValueError('Cant detect date format') from e
  • Вопрос задан
  • 79 просмотров
Решения вопроса 1
trapwalker
@trapwalker Куратор тега Python
Программист, энтузиаст
Если тянуть зависимости не хотите и во что бы то ни стало желаете реализовать самостоятельно, а набор форматов, которые нужно поддерживать, понятен, то можно селать так.
Делаете список форматов в порядке оценки невозрастания вероятности его применения.
По списку форматов идёте циклом и пробуете каждый. Попытку парсинга внутри цикла заворачиваете в try\except и при отсутствии ошибки произойдёт спешный return, а по факту завершения цикла райсите свою ошибку о том, что не удалось подобрать формат для парсинга даты.
Ответ написан
Пригласить эксперта
Ответы на вопрос 1
@bacon
возьми готовое решение https://github.com/scrapinghub/dateparser либо посмотри как они сделали
ЗЫ одна из проблем разбора, '09-01-2004' это 9 января или 1 сентября, спасибо Америке.
Ответ написан
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы