python操作輕量級資料庫sqlite較為簡單，直接上程式碼啦~

一. 函式版

　　舉例了建表、查詢、入庫等操作，其它操作大同小異。

import sqlite3 as db


def execute(sql: str, data=''):
    conn = db.connect('useful.db')  # 沒有的話會自動建立
    cursor = conn.cursor()
    res = cursor.execute(sql, data)
    yield res.fetchall()
    conn.commit()
    cursor.close()
    conn.close()


 
if __name__ == '__main__':
    create_table = "create table if not exists zhihu(" \
                   "hot text primary key unique, title text, excerpt text, detail_url text)"
    select_all = 'select * from zhihu'
    insert_any = '''insert or ignore into zhihu values ("faffadfadd","Fda","fda","fa")
 
'''

    for i in execute(insert_any):
        print(i)

　　函式版的雖然也可以，但是通過yield丟擲這裡多少有點不夠完美。。。

　　這裡可以用上下文解決。

二. 上下文版

　　資料庫的連線和關閉通過上下文解決，完美！

import sqlite3 as db


class DB:
    def __init__(self, sql: str, data='', db_path='useful.db'):
        self._sql = sql
        self._data = data
        self._conn 
 
= db.connect(db_path)  # 沒有的話會自動建立
        self._cursor = self._conn.cursor()

    def __enter__(self):
        res = self._cursor.execute(self._sql, self._data)
        self._conn.commit()
        return res.fetchall()

    def __exit__(self, exc_type, exc_value, exc_traceback):
        self._cursor.close()
        self._conn.close()

    def get(self):
        return self._cursor.fetchone()


if __name__ == '__main__':
    create_table = "create table if not exists zhihu(" \
                   "hot text primary key unique, title text, excerpt text, detail_url text)"
    select_all = 'select * from zhihu'
    insert_any = '''insert or ignore into zhihu values ("faffadfadd","Fda","fda","fa")'''

    with DB(select_all) as f:
        for i in f:
            print(i)

作者留言

　　原創不易，如果覺得本文對你有幫助，記得打賞噢O(∩_∩)O，您的支援是我創作路上最大的動力~