import sqlite3


def build_database(database_name, excel_name):
    # 连接数据库
    conn = sqlite3.connect(database_name)

    # 创建表格
    conn.execute(f'''CREATE TABLE IF NOT EXISTS {excel_name} (
                 id INTEGER PRIMARY KEY,
                 myarray TEXT);''')


# 将数据写入数据库
def save_database(database_name, excel_name, myarray):
    # 连接数据库
    conn = sqlite3.connect(database_name)
    # 将数组转换为字符串，以便保存到数据库中
    myarray_str = str(myarray)

    # 将数组保存到数据库中
    conn.execute(f"INSERT INTO {excel_name} (myarray) VALUES (?)", (myarray_str,))

    # 提交更改
    conn.commit()

    # 关闭连接
    conn.close()


# 读取数据库数据
def read_database(database_name, excel_name):
    # 连接数据库
    conn = sqlite3.connect(database_name)

    # 从数据库中读取保存的一维数组
    cursor = conn.execute(f"SELECT myarray from {excel_name}")
    myarray_str = cursor.fetchone()[0]
    print(len(myarray_str))
    myarray1 = eval(myarray_str)
    print(len(myarray1))

    # 关闭连接
    conn.close()