Untitled

 avatar
unknown
plain_text
a year ago
1.7 kB
4
Indexable
import csv

def ReadRecord(_id):
    with open('employee.csv', 'r') as f:
        reader = csv.reader(f)
        for _ in reader:
            print(_)
def WriteRecord():
    with open('employee.csv', 'a', newline='') as f:
        n = int(input('Enter No. Of Records To Enter: '))
        writer = csv.writer(f)
        for _ in range(1):
            empid = int(input('Enter Employee ID: '))
            name = input('Enter Employee Name: ')
            dept = input('Enter Employee Department: ')
            salary = int(input('Enter Employee Salary: '))
            record = [empid, name, dept, salary]
        for _ in record:
            writer.writerow(record)
def DeleteRecord(_id):
    data = []
    with open('employee.csv', 'r') as f:
        reader = csv.reader(f)
        for _ in reader:
            if _[0] == _id:
                pass
            else:
                data.append(i)
    with open('employee.csv', 'w') as f:
        writer = csv.writer(f)
        for _ in data:
            writer.writerow(_)
def main():
    while True:
        print('1. Write Employee Records')
        print('2. Read Employee Record')
        print('3. Delete Employee Record')
        print('4. Exit')
        x = int(input('Enter Your Choice: '))
        if x == 1:
            WriteRecord()
        elif x == 2:
            n = int(input('Enter The Employee ID To Read Record: '))
            ReadRecord(n)
        elif x == 3:
            n = int(input('Enter The Employee ID To Delete Record: '))
            DeleteRecord(n)
        elif x == 4:
            break
        else:
            print('Invalid Option')
main()
Editor is loading...
Leave a Comment