首页 > 解决方案 > 如何使其成为导出员工信息的功能?

问题描述

这是一个从员工管理系统中导入员工信息的功能。我想做一个类似的功能,可以导出员工信息。如何制作?

def import_employees():
    with open('Employee_database.csv', 'r') as csvfile: #Opens the student answer file, and closes once finished
        employee_reader = csv.reader(csvfile)
   
        first_row = True
        for row in employee_reader:
            #Skip the first row with column names
            if first_row:
                first_row = False
                continue
            try:
                fullName = row[0] + ' ' + row[1] #Creates full name for dictionary key
                employeeList[fullName] = row
            except IndexError:
                pass
    print('The following employees were imported:\n')
   
    for employee in employeeList.keys():
        print(employee)
    print(' ')
    input('<Press Enter to Continue>')


try:
    import_employees()
except:
    with open('Employee_database.csv', 'w') as employeeCSV:
                employee_writer = csv.writer(employeeCSV)
                csvrow1 = ['FirstName','LastName','SSN','Phone','Email','Salary']
                employee_writer.writerow(csvrow1)
       

标签: pythonpython-3.xfunction

解决方案


推荐阅读