📅  Mandatory Date Range Picker

Submitted by Zachary Blackwood

Summary

Just like st.date_input, but enforces that it always and only returns a start and end date, even if the user has only selected one of the dates. Until the user selects both dates, the app will not run.

Functions

date_range_picker

Working with date_input with a date range is frustrating becuase if you're assuming you will get a start and end date out of it, your code can break (not to mention your type hints), because if a user clicks on just one date, the app will go ahead and run with a single output. This widget enforces a start and end date being selected, and will stop the app if only one is chosen.

Parameters:

Name	Type	Description	Default
title	str	Title of the date widget	required
default_start	Optional[date]	Default start date. Defaults to None.	None
default_end	Optional[date]	Default end date. Defaults to None.	None
min_date	Optional[date]	Minimum date. Defaults to None.	None
max_date	Optional[date]	Maximum date. Defaults to None.	None
error_message	str	Error message when only one date is chosen. Defaults to "Please select start and end date".	'Please select start and end date'
key	Optional[str]	Widget key. Defaults to None.	None

Returns:

Type	Description
Tuple[date, date]	Tuple[date, date]: Start and end date chosen in the widget

Source code in src/streamlit_extras/mandatory_date_range/__init__.py

@extra
def date_range_picker(
    title: str,
    default_start: Optional[date] = None,
    default_end: Optional[date] = None,
    min_date: Optional[date] = None,
    max_date: Optional[date] = None,
    error_message: str = "Please select start and end date",
    key: Optional[str] = None,
) -> Tuple[date, date]:
    """
    Working with date_input with a date range is frustrating becuase if you're
    assuming you will get a start and end date out of it, your code can break (not
    to mention your type hints), because if a user clicks on just one date, the app
    will go ahead and run with a single output. This widget enforces a start and
    end date being selected, and will stop the app if only one is chosen.

    Args:
        title (str): Title of the date widget
        default_start (Optional[date], optional): Default start date. Defaults to None.
        default_end (Optional[date], optional): Default end date. Defaults to None.
        min_date (Optional[date], optional): Minimum date. Defaults to None.
        max_date (Optional[date], optional): Maximum date. Defaults to None.
        error_message (str, optional): Error message when only one date is chosen.
            Defaults to "Please select start and end date".
        key (Optional[str], optional): Widget key. Defaults to None.

    Returns:
        Tuple[date, date]: Start and end date chosen in the widget
    """

    if default_start is None:
        default_start = date.today() - timedelta(days=30)
    if default_end is None:
        default_end = date.today()

    val = st.date_input(
        title,
        value=[default_start, default_end],
        min_value=min_date,
        max_value=max_date,
        key=key,
    )
    try:
        start_date, end_date = cast(Tuple[date, date], val)
    except ValueError:
        st.error(error_message)
        st.stop()

    return start_date, end_date

Import:

from streamlit_extras.mandatory_date_range import date_range_picker # (1)!

1. You should add this to the top of your .py file

Examples

example

def example():
    st.write(
        """
        This is an example of a date range picker that *always* returns a start and
        end date, even if the user has only selected one of the dates. Until the
        user selects both dates, the app will not run.
        """
    )
    result = date_range_picker("Select a date range")
    st.write("Result:", result)

Output (beta)

🎈 Powered by stlite