notebook.py

from os import close
import sys
import getopt


def add(name, quantity, price):
    with open('data', 'a') as f:
        f.write(f"{name}-{quantity}-{price}\n")


def delete(name):
    
    content = open('data', 'r')
    items = content.readlines()
    content.close()

    f = open('data', 'w')
    for item in items:
        if not(name in item.split('-')[0]):
            f.write(item)
    f.close()


def print_list():
    with open('data', 'r') as f:
        print("Name", "Quantity", "Price")
        print(f.read().replace('-', '\t'))

def main(argv):
    #try to parse input argument
    try:
        opts, args = getopt.getopt(argv, "ga:d:", ["add=", "del="])
    except getopt.GetoptError:
        print("[-a][--a] to enter new value and [-g] to print list")


    for opt ,arg in opts:
        if opt in ["-a", "--add"]:
            input_data = (arg.split('-'))
            add(input_data[0], input_data[1], input_data[2])
        elif opt in ["-d", "--del"]:
            delete(arg)
        elif opt == "-g":
            print_list()
    


if __name__ == "__main__":
    main(sys.argv[1:])