Data analysis workflow module: time data structured output

def main(time_period) -> dict:
    import datetime
    
    # Get the current date
    current_date = datetime.date.today()
    
    if time_period == 'daily':
        # Calculate yesterday's date
        yesterday = current_date - datetime.timedelta(days=1)
        day_of_year = yesterday.timetuple().tm_yday
        start_of_period = yesterday.strftime('%Y-%m-%d')
        end_of_period = current_date.strftime('%Y-%m-%d')
        
        return {
            "start_of_period": start_of_period,
            "end_of_period": end_of_period,
            "series_no_of_period": day_of_year
        }

    elif time_period == 'weekly':
        # Get the current week number
        week_number = current_date.isocalendar()[1]
        # Calculate the start of the week (Monday)
        start_of_week = current_date - datetime.timedelta(days=current_date.weekday())
        # Calculate the end of the week (Sunday)
        end_of_week = start_of_week + datetime.timedelta(days=6)
        
        # Ensure end_of_week does not exceed today
        end_of_week = min(end_of_week, current_date)
        
        return {
            "start_of_period": start_of_week.strftime('%Y-%m-%d'),
            "end_of_period": end_of_week.strftime('%Y-%m-%d'),
            "series_no_of_period": week_number
        }

    elif time_period == 'monthly':
        month_number = current_date.month
        start_of_month = current_date.replace(day=1)
        end_of_month = (start_of_month + datetime.timedelta(days=32)).replace(day=1) - datetime.timedelta(days=1)
        
        # Ensure end_of_month does not exceed today
        end_of_month = min(end_of_month, current_date)
        
        return {
            "start_of_period": start_of_month.strftime('%Y-%m-%d'),
            "end_of_period": end_of_month.strftime('%Y-%m-%d'),
            "series_no_of_period": month_number
        }

    elif time_period == 'quarterly':
        quarter = (current_date.month - 1) // 3 + 1
        start_of_quarter = datetime.date(current_date.year, 3 * quarter - 2, 1)
        end_of_quarter = (start_of_quarter + datetime.timedelta(days=95)).replace(day=1) - datetime.timedelta(days=1)
        
        # Ensure end_of_quarter does not exceed today
        end_of_quarter = min(end_of_quarter, current_date)
        
        return {
            "start_of_period": start_of_quarter.strftime('%Y-%m-%d'),
            "end_of_period": end_of_quarter.strftime('%Y-%m-%d'),
            "series_no_of_period": quarter
        }

    else:
        return {
            "Error": f"Invalid time period: '{time_period}'. Choose 'daily', 'weekly', 'monthly', or 'quarterly'."
        }

# Example usage
# print(main('daily'))
# print(main('weekly'))
# print(main('monthly'))
# print(main('quarterly'))