import csv


def add_to_file():
    name = str(input('Enter name : '))
    salary = str(input('Enter salary : '))
    file = open('Salaries.csv', 'a')
    file.write(str(name + ', ' + salary + '\n'))
    file.close()


def view_file():
    file = open('Salaries.csv', 'r')
    count_row = 0
    for row in file:
        print('Row  ' + str(count_row) + ' : ' + str(row))
        count_row += 1


def delete_record():
    file = open('Salaries.csv')
    file_read = list(csv.reader(file))
    count_row = 0
    list_records = []
    for row in file_read:
        print('Row  ' + str(count_row) + ' : ' + str(row))
        list_records.append(row)
        count_row += 1
    file.close()
    ask = int(input('Enter row number to delete : '))
    list_records.pop(ask)
    file_write = open('Salaries.csv', 'w')
    i = 0
    for row in range(len(list_records)):
        file_write.write(list_records[i][0] + ', ' + list_records[i][1] + '\n')
        i += 1
    file_write.close()


def main():
    state = True
    count = 0
    while state:
        if count != 3:
            print('1) Add to file')
            print('2) View all records')
            print('3) Delete a records')
            print('4) Quit')
            operation_ask = int(input('Enter the number of your selection : '))
            if operation_ask == 1:
                add_to_file()
            elif operation_ask == 2:
                view_file()
            elif operation_ask == 3:
                delete_record()
            elif operation_ask == 4:
                print('Good bye')
                state = False
            else:
                print('Invalid input, please try again.')
        else:
            state = False


main()