load a csv file named foo.txt and, depending on the first specified argument to the application, either add a new entry or remove an existing one in python

main.py
import csv
import argparse


def add_entry(file_path, new_entry):
    with open(file_path, mode='a', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(new_entry)


def remove_entry(file_path, entry_to_remove):
    rows_to_keep = []
    with open(file_path, mode='r') as file:
        reader = csv.reader(file)
        for row in reader:
            if row != entry_to_remove:
                rows_to_keep.append(row)
    
    with open(file_path, mode='w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(rows_to_keep)
        

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Add or remove entry in a CSV file.')
    parser.add_argument('action', type=str, choices=['add', 'remove'], help='Add or remove an entry.')
    parser.add_argument('file_path', type=str, help='Path to the CSV file.')
    parser.add_argument('entry', type=str, nargs='+', help='Entry to add or remove as a list of values separated by commas.')
    args = parser.parse_args()

    if args.action == 'add':
        add_entry(args.file_path, args.entry)
    elif args.action == 'remove':
        remove_entry(args.file_path, args.entry)
1200 chars
35 lines