python读写csv文件

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
@File : csv_util.py
@Creation Time : 2023/7/19 17:23
Description : 读取csv文件
"""
import csv


class CsvUtil(object):

    def read_file(self, filename, key):
        """
        读取csv文件
        :param filename: 文件路径
        :param key: 需要读取的字段
        :return: 以列表返回改行数据
        """
        with open(filename, "r", newline="") as file:
            reader = csv.reader(file)
            for row in reader:
                if key in row:
                    return row[1:]

    def write_file(self, filename, key, value):
        """
        写入数据，写入数据如果不存在，会追加到文件尾部
        :param filename: 文件名
        :param key: 需要修改的字段
        :param value: 需要修改的数据
        :return:
        """
        flag = False
        index1 = 0
        list1 = [key]
        for i in value:
            list1.append(i)

        with open(filename, "r", newline="") as file:
            reader = csv.reader(file)
            data = list(reader)

        for row in data:
            if len(row) > 0 and row[0] == key:
                index1 = data.index(row)
                flag = True
                break
        if flag:
            data.pop(index1)
            data.insert(index1, list1)
        else:
            data.append(list1)
        with open(filename, "w", newline="") as file:
            writer = csv.writer(file)
            writer.writerows(data)


if __name__ == '__main__':
    from config.settings import get_files_path
    import os

    value = [1, 2, 3, 4, 5]
    filename = get_files_path() + os.sep + "test_rely.csv"
    res = CsvUtil()
    res.write_file(filename, "token", ["as111da","asd"])