check_acronyms.py

#!/usr/bin/env python3
"""
Script to check for various formatting issues with acronyms file
Does a few things, including:
1) Moving any all-lower-case definitions to Title case
2) Moving any all-upper-case strings to Title case
3) Turning smart quotes (e.g. “”‘’) to regular quotes
"""

import argparse
import csv
import sys


def fix_all_lower(row: list) -> list:
    if row[1].islower():
        row[1] = row[1].title()
    return row


def fix_all_upper(row: list) -> list:
    if row[1].isupper():
        row[1] = row[1].title()
    return row


def fix_smart_quotes(row: list) -> list:
    row = [elem.replace("‘", "'") for elem in row]
    row = [elem.replace("’", "'") for elem in row]
    row = [elem.replace("“", "\"") for elem in row]
    row = [elem.replace("”", "\"") for elem in row]
    return row


def process_row(row: list) -> list:
    row = fix_all_lower(row)
    row = fix_all_upper(row)
    row = fix_smart_quotes(row)
    return row


def main(csv_filename: str):

    csv.register_dialect('my_format', lineterminator='\n', strict=True)
    with open(csv_filename, newline='') as csv_file:
        csv_reader = csv.reader(csv_file)
        csv_writer = csv.writer(sys.stdout, quoting=csv.QUOTE_MINIMAL,
            dialect='my_format')
        line_count = 0
        for row in csv_reader:
            if line_count == 0:
                pass
            else:
                row = process_row(row)
            line_count += 1
            csv_writer.writerow(row)


if __name__ == '__main__':
    parser = argparse.ArgumentParser(
        description="Fixes and outputs to stdout various issues with " +
        "acronyms file.")
    parser.add_argument('-f', '--file', required=False, action='store',
                        default='acronyms.csv',
                        help="File to use. Default: acronyms.csv")
    args = parser.parse_args()
    main(args.file)