增删改查
import sqlite3
conn = None
#连接数据库
def connect():
global conn
conn = sqlite3.connect('test.db')
return
def getcursor():
return conn.cursor()
def disconnect():
conn.close()
return
#建表
def createTable(c):
c.execute('''CREATE TABLE COMPANY
(ID INT PRIMARY KEY NOT NULL,
NAME TEXT NOT NULL,
AGE INT NOT NULL,
ADDRESS CHAR(50),
SALARY REAL);''')
return
def addData(c):
c.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
VALUES (1, 'Paul', 32, 'California', 20000.00 )")
c.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
VALUES (2, 'Allen', 25, 'Texas', 15000.00 )")
c.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
VALUES (3, 'Teddy', 23, 'Norway', 20000.00 )")
c.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
VALUES (4, 'Mark', 25, 'Rich-Mond ', 65000.00 )")
conn.commit()
return
def deleteData(c):
c.execute("DELETE from COMPANY where ID=2;")
conn.commit()
return
def updateData(c):
c.execute("UPDATE COMPANY set SALARY = 25000.00 where ID=1")
conn.commit()
return
def getData(c):
cursor = c.execute('select id,name,address,salary from COMPANY')
print('---------------------------------')
for row in cursor:
print('ID:{} NAME:{} ADDRESS:{} SALARY:{}'.format(row[0],row[1],row[2],row[3]))
print('---------------------------------')
return
def main():
connect()
c = getcursor()
createTable(c)
getData(c)
addData(c)
getData(c)
deleteData(c)
getData(c)
updateData(c)
getData(c)
disconnect()
if __name__ == '__main__':
main()
[Running] python -u “c:\Users\fwk\Desktop\db.py”
ID:1 NAME:Paul ADDRESS:California SALARY:20000.0
ID:2 NAME:Allen ADDRESS:Texas SALARY:15000.0
ID:3 NAME:Teddy ADDRESS:Norway SALARY:20000.0
ID:4 NAME:Mark ADDRESS:Rich-Mond SALARY:65000.0
ID:1 NAME:Paul ADDRESS:California SALARY:20000.0
ID:3 NAME:Teddy ADDRESS:Norway SALARY:20000.0
ID:4 NAME:Mark ADDRESS:Rich-Mond SALARY:65000.0
ID:1 NAME:Paul ADDRESS:California SALARY:25000.0
ID:3 NAME:Teddy ADDRESS:Norway SALARY:20000.0
ID:4 NAME:Mark ADDRESS:Rich-Mond SALARY:65000.0
[Done] exited with code=0 in 0.079 seconds