示例#1
0
def modifier(file_name):
    u"""Основная функция ДЗ"""
    input_file = open(file_name, 'rb')
    reader = csv.DictReader(input_file)
    field_names = reader.fieldnames
    data_list = []
    for row in reader:
        data_list.append(row)
        var_class = Person(
            data_list[-1]['surname'],
            data_list[-1]['name'],
            data_list[-1]['birthdate'],
            data_list[-1]['nickname'])

        data_list[-1]['fullname'] = var_class.get_fullname()
        data_list[-1]['age'] = var_class.get_age()
    field_names.append('fullname')
    field_names.append('age')
    input_file.close()
    output_file = open(file_name, 'wb')
    writer = csv.DictWriter(output_file, field_names)
    writer.writeheader()
    for str_data in data_list:
        writer.writerow(str_data)
    output_file.close()