import this
0eb6eb855b
* initialise nym versin of good-bad isps by tor community * syntax edit * finalise csv2md * initialise isp-list page * add last update info * add python3 plugis to ci/cd runnners * remove argparse installation * python3 syntax edit * pip install attempt * pip install attempt * change python version in cmdrun command * fix python modules installation * correct command python version * workflows python env update * add probe guide, clean FnF pages * fix typos
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
#!/usr/bin/python3
|
|
|
|
"""CLI to display .csv files as markdown"""
|
|
|
|
import argparse
|
|
import pandas as pd
|
|
import sys
|
|
import csv
|
|
|
|
def create_table(args):
|
|
"""Imports csv and creates a table"""
|
|
file = args.file
|
|
csv = pd.read_csv(file)
|
|
if args.sort != None:
|
|
csv = csv.sort_values(csv.columns[args.sort])
|
|
if args.table:
|
|
table = csv.to_markdown(tablefmt="grid", index=args.index)
|
|
else:
|
|
table = csv.to_markdown(index=args.index)
|
|
return table
|
|
|
|
def display_file(args):
|
|
"""Display csv file as a table"""
|
|
table = create_table(args)
|
|
print(table)
|
|
|
|
def panic(msg):
|
|
"""Error message print"""
|
|
print(f"error: {msg}", file=sys.stderr)
|
|
sys.exit(-1)
|
|
|
|
def parser_main():
|
|
"""Main function initializing ArgumentParser, storing arguments and executing commands."""
|
|
# Top level parser
|
|
parser = argparse.ArgumentParser(
|
|
prog='CSV2MD',
|
|
description='''Displays .csv files in markdown''',
|
|
epilog='''Code is power!'''
|
|
)
|
|
|
|
# Parser arguments
|
|
parser.add_argument("-V","--version", action="version", version='%(prog)s 1.1.0')
|
|
parser.add_argument("file", help="path/to/file.csv")
|
|
parser.add_argument("-t","--table", default=False, action="store_true", help="output with a tabulate option for terminal reading - does not render in mdbook")
|
|
parser.add_argument("-i","--index", default=False, action="store_true", help="output with an index column")
|
|
parser.add_argument("-s","--sort", type=int, help="supply with column index to sort your output accordingly (ascending way)")
|
|
|
|
parser.set_defaults(func=display_file)
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
args.func(args)
|
|
except AttributeError as e:
|
|
msg = f"{e}.\nPlease run with --help or read the error message in case your .csv file is corrupted."
|
|
panic(msg)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser_main()
|