def prepare_data(
        dataframe,
        add_datetime_features=True
    ):

    print('Building features...')
    
    if add_datetime_features:
        dataframe = datetime_features(dataframe)

    return dataframe

def datetime_features(dataframe, date='date', suffix=''):

    dataframe[f'{suffix}_month'] = dataframe[date].dt.month
    dataframe[f'{suffix}_year'] = dataframe[date].dt.year
    dataframe[f'{suffix}_quarter'] = dataframe[date].dt.quarter
    dataframe[f'{suffix}_weekofyear'] = dataframe[date].dt.isocalendar().week

    return dataframe