activity_dates.py

FIELDS = [f'{i}-{j}' for i in ['start', 'end'] for j in ['planned', 'actual']]


def activity_dates(data):
    """
    Requested by FCDO. Single valued fields.
    Activity dates is one of four types: planned or actual, start or end.
    start planned (type 1), start actual (type 2),
    end planned (type 3), end actual (type 4)

    Additionally, add the common start and end dates.
    Shows the available between planned and actual dates.

    Relevant fields:
    'activity-date.type' and 'activity-date.iso-date'

    :param data: reference to the activity in the data
    """
    if 'activity-date' in data:
        if type(data['activity-date']) is dict:
            data['activity-date'] = [data['activity-date']]
        for date in data['activity-date']:
            data = extract_activity_dates(date, data)
    return data


def extract_activity_dates(date, data):
    # This approach was tested to be the fastest with 10.000 runs
    if 'type' not in date or 'iso-date' not in date:
        return data
    common_start = None
    common_end = None
    for i, field in enumerate(FIELDS, 1):
        if date['type'] == i:
            data[f'activity-date.{field}'] = date['iso-date']
            if i <= 2:
                common_start = date['iso-date']
            else:
                common_end = date['iso-date']
    if common_start:
        data['activity-date.common.start'] = common_start
    if common_end:
        data['activity-date.common.end'] = common_end
    return data