def rename_headers()

in datasets/san_francisco_311/pipelines/_images/run_csv_transform_kub/csv_transform.py [0:0]

• 25 lines of code
• 1 McCabe index (conditional complexity)

def rename_headers(df: pd.DataFrame) -> pd.DataFrame:
    logging.info("Renaming headers")
    header_names = {
        "CaseID": "unique_key",
        "Opened": "created_date",
        "Closed": "closed_date",
        "Updated": "resolution_action_updated_date",
        "Status": "status",
        "Status Notes": "status_notes",
        "Responsible Agency": "agency_name",
        "Category": "category",
        "Request Type": "complaint_type",
        "Request Details": "descriptor",
        "Address": "incident_address",
        "Supervisor District": "supervisor_district",
        "Neighborhood": "neighborhood",
        "Point": "location",
        "Source": "source",
        "Media URL": "media_url",
        "Latitude": "latitude",
        "Longitude": "longitude",
        "Police District": "police_district",
    }
    df = df.rename(columns=header_names)

    return df