Тип конверсии кликов

Question

Я использую click определить CLI, который принимает datetimes и список параметров через запятую.

import click

def valid_date(s):
    try:
        return dt.strptime(s, "%Y-%m-%d")
    except ValueError:
        msg = "Not a valid date: '{0}'.".format(s)
        raise Exception(msg)
    except TypeError:
        return None

split_parameter = lambda _, __, s: s.split(",")
check_date = lambda _, __, s: valid_date(s)


@click.command()
@click.argument('symbols', callback=split_parameter)
@click.option('--start_date', callback=check_date)
@click.option('--end_date', callback=check_date)
@click.option('--file_name')
def f(symbols, start_date, end_date, file_name):
     return None

Это работает, но логика вокруг callbackнемного неловко И разделение списка и преобразование в datetime прерывает красивую простую структуру клика. Есть ли питонский способ сделать это?

Answer 1

Вы можете инкапсулировать уродливые lambda _, __, x часть в декораторе:

import click
from datetime import datetime as dt

def click_callback(f):
    return lambda _, __, x: f(x)

def check_date(s):
    try:
        return dt.strptime(s, "%Y-%m-%d")
    except ValueError:
        msg = "Not a valid date: '{0}'.".format(s)
        raise Exception(msg)
    except TypeError:
        return None

@click.command()
@click.argument('symbols', callback=click_callback(lambda s: s.split(',')))
@click.option('--start_date', callback=click_callback(check_date))
@click.option('--end_date', callback=click_callback(check_date))
@click.option('--file_name')
def f(symbols, start_date, end_date, file_name):
    print(symbols, start_date, end_date, file_name)

if __name__ == '__main__':
    f()

Или вот так (показана только соответствующая часть):

def click_callback(f):
    return {'callback': lambda _, __, x: f(x)}

@click.command()           #  vvvvvvvvvvvvvvvv
@click.argument('symbols',    **click_callback(lambda s: s.split(',')))
@click.option('--start_date', **click_callback(check_date))
@click.option('--end_date',   **click_callback(check_date))
@click.option('--file_name')
def f(symbols, start_date, end_date, file_name):
    print(symbols, start_date, end_date, file_name)

Answer 2

Я хотел, чтобы строка даты была в формате '05-Jul-2018', у меня сработало следующее

def check_date(ctx, param, value):
    try:
        _ = dt.strptime(value, "%Y-%m-%d")
        return value
    except ValueError:
        msg = "Not a valid date: '{0}'.".format(value)
        raise Exception(msg)

def getDate():
    return '09-Jul-2018' #you can have any code here

@click.command()
@click.option('--start_date', callback=check_date, default=lambda :getDate())
def f(symbols, start_date):
    print(symbols, start_date)

if __name__ == '__main__':
    f()

Вы можете найти больше помощи здесь