45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
|
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()
|