save_database_info.py

# -*- coding: utf-8 -*-
import numpy as np
import pandas as pd
import re
from re import sub
import smtplib
import matplotlib.pyplot as plt
from itertools import product
from tqdm import tqdm_notebook, tqdm, trange
import time
import seaborn as sns
from matplotlib.pyplot import style
from rich import print
import warnings
warnings.filterwarnings('ignore')
sns.set()
# style.use('seaborn')

import csv

def save_database_info(filepath, database_name, date):
    # 读取CSV文件
    with open('/Users/yunshi/Downloads/360Data/Data Center/Working-On Task/演讲与培训/2023ChatGPT/Coding/code_interpreter/test/database_name.csv', 'r', encoding='utf-8') as file:
        # 创建CSV读取器
        reader = csv.reader(file)

        # 将内容存储到列表中
        rows = []
        for row in reader:
            rows.append(row)

    # 添加新行
    # new_row = ['New Data 1', 'New Data 2'] # 新行的数据
    new_row = [database_name, date] # 新行的数据
    rows.append(new_row)

    # 写入CSV文件
    with open('/Users/yunshi/Downloads/360Data/Data Center/Working-On Task/演讲与培训/2023ChatGPT/Coding/code_interpreter/test/database_name.csv', 'w', newline='', encoding='utf-8') as file:
        # 创建CSV写入器
        writer = csv.writer(file)
        # 写入所有行
        writer.writerows(rows)
    
    # close the file to save the data. 
    file.close()

    return None