import argparse
from rich.console import Console
from rich.markdown import Markdown
from rich.table import Column, Table


def parse_arguments():
    parser = argparse.ArgumentParser(
        "Displays a nicely formatted cheatsheet of")
    parser.add_argument("-a", "--append",
                        help="Append a shortcut to your collection using Markdown.")
    args = parser.parse_args()
    return args


def display_file():
    with open("cheatsheet.md") as cheats:
        markdown = Markdown(cheats.read())
    console = Console()
    console.print(markdown)


def append_to_file(data):
    # This is not implemented.
    x = 0


def main():
    args = parse_arguments()
    a = args.append
    # print(args)
    if a is None:
        display_file()
    else:
        print("Appending to file...")
        append_to_file(args.append)
    x = 4


if __name__ == "__main__":
    main()