66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
|
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()
|