python中数据库的代码片

sqlite数据库操作

PYTHON

# 查询操作
def search(sql: str) -> list:
    try:
        conn = sqlite3.connect(SQLITE_DATABASE_URL)
        cursor = conn.cursor()
        cursor.execute(sql)
        # 返回的是list，也就是多行数据，每行数据是一个元组，只能用下标获取数据
        # returnList[0]获取第一行数据，returnList[0][0]获取第一行数据的第一列数据
        return cursor.fetchall()
    finally:
        cursor.close()
        conn.close()

# 增删改操作
def execute(sql: str):
    try:
        conn = sqlite3.connect(SQLITE_DATABASE_URL)
        cursor = conn.cursor()
        cursor.execute(sql)
        conn.commit()
    finally:
        cursor.close()
        conn.close()