Python sqlite3操作筆記

創建數據庫

def create_tables(dbname):
    conn = sqlite3.connect(dbname) print "Opened database successfully"; c = conn.cursor() c.execute('''CREATE TABLE VULNDB  (Plugin_ID INT PRIMARY KEY NOT NULL, NAME TEXT NOT NULL, Risk TEXT NOT NULL, Description CHAR(1000), Solution CHAR(1000));''') print "Table created successfully"; conn.commit() conn.close()

查詢或刪除

def selectdb():
    conn = sqlite3.connect('vuln.db') conn.text_factory=str c = conn.cursor() cursor = c.execute("SELECT count(Plugin_ID) from VULNDB") for row in cursor: print row

c.execute("DELETE from VULNDB where Plugin_ID=34311;")

 

在sqlite3中插入中文字符

#!/usr/bin/python
# -*- coding:utf-8 -*- 

import sqlite3
conn = sqlite3.connect('test.db')
conn.text_factory=str
c = conn.cursor()

users = (2,'username','high','騰訊qq', '[e-mail]')
ins="INSERT INTO VULNDB(Plugin_ID,NAME,Risk,Description,Solution) VALUES (?,?,?,?,?)"
c.execute(ins,users)  
conn.commit()
print "Records created successfully";
c.close()
conn.close()

 

 CSV導入數據庫

def insertdata(v):
    conn = sqlite3.connect('vuln.db')
    conn.text_factory=str
    c = conn.cursor()
    ins="INSERT INTO VULNDB(Plugin_ID,NAME,Risk,Description,Solution) VALUES (?,?,?,?,?)"
    c.execute(ins,v)
    conn.commit()
    conn.close()
def write():    
    with open("loudong-20180913.csv", "r") as csvFile:
        reader2 = csv.reader(csvFile) # 讀取csv文件，返回的是迭代類型
        for item2 in reader2:
            #print item2
            insertdata(item2)        
    csvFile.close()

 

參考鏈接：

https://my.oschina.net/letiantian/blog/217770