how to take date as input from user in python code example

Example: python input date

date = input("Enter date in YYYY-MM-DD format")
year, month, day = map(int, date.split('-'))
date = datetime.date(year, month, day)