rename_activities.py

#!/usr/bin/python3
import argparse
import csv
import datetime as dt
import os
import sys
from pathlib import Path

from termcolor import cprint


def load_csv(csv_file: str) -> list[dict[str, str]]:
    data = []

    with open(csv_file) as csv_file:
        reader = csv.DictReader(csv_file)
        for row in reader:
            data.append(row)

    print(len(data), "rows found")
    if len(data) == 0:
        sys.exit("No data found")

    return data


def convert_date_string(input_string: str) -> str:
    # May 22, 2010, 7:39:29 PM
    # ->
    # 20100522-193929
    new_date = dt.datetime.strptime(input_string, "%b %d, %Y, %I:%M:%S %p")
    new_date = new_date.isoformat()
    new_date = new_date.replace("-", "")
    new_date = new_date.replace(":", "")
    new_date = new_date.replace("T", "-")
    return new_date


def output_file(activity: dict[str, str], ext: str | None = None) -> Path:
    # Return name like 20141207-152233-Ride.gpx
    new_date = convert_date_string(activity["Activity Date"])

    path = Path(activity["Filename"])
    if not ext:
        ext = "".join(path.suffixes)
    outfile = f"{new_date}-{activity['Activity Type']}{ext}"
    outfile = path.parent / outfile
    return outfile


def main() -> None:
    parser = argparse.ArgumentParser(
        description="Rename files, eg. 1836025202.gpx to 20180912-064451-Ride.gpx",
        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
    )
    parser.add_argument("-c", "--csv", default="activities.csv", help="CSV filename")
    parser.add_argument(
        "-n", "--dry-run", action="store_true", help="Don't rename files"
    )
    args = parser.parse_args()

    activities = load_csv(args.csv)

    for activity in activities:
        # We want to rename the file like 20141207-152233-Ride.gpx
        # print(activity)
        infile = activity["Filename"]
        outfile = output_file(activity)

        print(f"{infile}\t->\t{outfile}")
        if not args.dry_run:
            try:
                os.rename(activity["Filename"], outfile)
            except FileNotFoundError:
                cprint(f"File not found: {infile}", "yellow")


if __name__ == "__main__":
    main()