python db读写实践

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'  # sql查询语句
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',
                       # fail:如果表存在，啥也不做，
                       # replace:如果表存在，删了表，再建立一个新表，把数据插入，
                       # append:如果表存在，把数据插入，如果表不存在创建一个表！
                       index=False,
                       index_label=False)
    conn.close()

name_tmp = input("请输入用户名")
passwd_tmp = input("请输入密码")
writeDB(name_tmp, passwd_tmp)
print("写入成功")

# 再用ReadDB检查一下有没有增加
sql = 'SELECT * FROM user'  # sql查询语句
print(ReadDB(path))