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

Question

[kicherov_maxim]

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

Тест

@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

Comments

[lz961]

Классно, третье десятилетие 21 века, а люди до сих пор сочиняют подпрограммы со множественными точками входа.

Answer 1

[Сергей П]

Если тянуть зависимости не хотите и во что бы то ни стало желаете реализовать самостоятельно, а набор форматов, которые нужно поддерживать, понятен, то можно селать так.
Делаете список форматов в порядке оценки невозрастания вероятности его применения.
По списку форматов идёте циклом и пробуете каждый. Попытку парсинга внутри цикла заворачиваете в try\except и при отсутствии ошибки произойдёт спешный return, а по факту завершения цикла райсите свою ошибку о том, что не удалось подобрать формат для парсинга даты.

Comments

[kicherov_maxim]

Вот так получилось, вроде лучше, спасибо за совет.

def convert_date(value: date | datetime | str) -> date:
    date_directives = ['%d.%m.%Y', '%d/%m/%Y', '%Y-%m-%d']
    if isinstance(value, datetime):
        return value.date()
    if isinstance(value, date):
        return value
    for directive in date_directives:
        try:
            return datetime.strptime(value, directive).date()
        except ValueError:
            pass
    raise ValueError('Cant detect date format')

[Сергей П]

kicherov_maxim, ага. У меня так же. Ещё форматы засовываю дефолт-параметром в эту же функцию, а перечень на уровне модуля лежат в виде константы, но за счет параметра можно и подменить, а можно добавить плюсиком в список в аргументе.
Ещё момент. Вы смешали в одной функции и парсинг и отбрасывание времени. Парсьте универсально,а отбрасывайте потом.