1. 读DB
import sqlite3
import pandas as pd
def ReadDB(path: str, sql):
conn = sqlite3.connect(path)
df_sql = pd.read_sql(sql, conn)
conn.close()
return df_sql
path = "test.db"
sql = 'SELECT * FROM user'
print(ReadDB(path, sql))
2. 写DB
def writeDB(name_tmp,passwd_tmp):
a = []
data = {
'用户名': name_tmp,
'密码': passwd_tmp
}
a.append(data)
conn = sqlite3.connect(path)
df = pd.DataFrame(a, columns=['用户名', '密码'])
df_sql = df.to_sql(name='user',
con=conn,
if_exists='append',
index=False,
index_label=False)
conn.close()
name_tmp = input("请输入用户名")
passwd_tmp = input("请输入密码")
writeDB(name_tmp, passwd_tmp)
print("写入成功")
sql = 'SELECT * FROM user'
print(ReadDB(path))