import mysql.connector as mc

cnx = mc.connect(host="localhost", user="alen", password="alen", database="contact")
cur = cnx.cursor()

def create_record():
    name = input("Enter name: ")
    address = input("Enter address: ")
    mobile = input("Enter mobile: ")
    email = input("Enter email: ")

    cur.execute("INSERT INTO BOOK(name, address, book, email) VALUES (%s, %s, %s, %s)", (name, address, mobile, email))
    cnx.commit()

    print("Records Inserted Successfully")

def search(name):
    cur.execute("SELECT * FROM BOOK WHERE NAME = %s", (name,))

    record = cur.fetchone()

    if record == None:
        print("No such record exists")

    else:
        print("Name: ",record[0])
        print("Address: ",record[1])
        print("Mobile: ", record[2])
        print("Email: ", record[3])

def display_all():
    cur.execute("SELECT * FROM BOOK")

    print("{:<20}".format("Name"), "{:<20}".format("Address"), "{:<20}".format("MOBILE NO"), "{:<20}".format("E-MAIL"))

    for k in cur:
        print("{:<20}".format(k[0]), "{:<20}".format(k[1]), "{:<20}".format(k[2]), "{:<20}".format(k[3]))

def delete_record(name):
    cur.execute("DELETE FROM BOOK WHERE NAME=%s", (name,))
    cnx.commit()

    if cur.rowcount == 0:
        print("No records found")

    else:
        print("Record deleted successfully")

def modify_record(name):
    cur.execute("SELECT * FROM BOOK WHERE NAME=%s", (name,))
    record = cur.fetchone()

    if record == None:
        print("No such record exists")

    else:
        while True:
            print("1. Name")
            print("2. Address")
            print("3. Mobile")
            print("4. BACK")

            ch = int(input("Enter your selection: "))

            if ch == 1:
                nname = input("Enter new name: ")
                cur.execute("UPDATE BOOK SET NAME = %s WHERE NAME = %s", (nname, name))
                cnx.commit()
                print(cur.rowcount, " record(s) updated successfully")

            elif ch == 2:
                naddress = input("Enter new Address: ")
                cur.execute("UPDATE BOOK SET ADDRESS = %s WHERE NAME = %s", (naddress, name))

            elif ch == 3:
                nmobile = input("Enter new mobile: ")
                cur.execute("UPDATE BOOK SET MOBILE=%s WHERE NAME=%s", (nmobile, name))
                cnx.commit()
                print(cur.rowcount, "record(s) updated successfully")

            elif ch == 4:
                break

            else:
                print("Invalid option, please try again later")

while True:
    print("        MAIN MENU")
    print("1. Add a new record")
    print("2. Search Record")
    print("3. Display all records")
    print("4. Delete Record")
    print("5. Modify Record")
    print("6. EXIT")

    ch = int(input("Enter your choice: "))

    if ch == 1:
        create_record()

    elif ch == 2:
        name = input("Enter name: ")
        search(name)

    elif ch == 3:
        display_all()

    elif ch == 4:
        name = int(input("Enter name: "))
        delete_record(name)

    elif ch == 5:
        name = input("Enter name: ")
        modify_record(name)

    elif ch == 6:
        break

    else:
        print("Invalid option, Please try again later")