CSVFormat.py

# This can be used to convert a normal csv file into a beautiful tabular form in ASCII Art style.

import csv
import sys
from tabulate import tabulate

def main():
    if len(sys.argv) < 2:
        sys.exit("Too few command-line arguments")
    elif len(sys.argv) > 2:
        sys.exit("Too many command-line arguments")
    else:
        if sys.argv[1][-4:] != ".csv":
            sys.exit("Not a CSV file")
        else:
            print(tabulize(sys.argv[1]))

def tabulize(file):
    try:
        with open(file) as f:
            reader = csv.reader(f)
            table = tabulate(reader, headers="firstrow", tablefmt="grid")
            return table
    except FileNotFoundError:
        sys.exit("File does not exist")


if __name__ == "__main__":
    main()

#! Run using `python CSVFormat.py nameOfFile.csv` in terminal