import sqlite3

conn = sqlite3.connect('studentsdb.db')

# 连接数据库

cursor = conn.cursor( )

# 创建数据表

def createDatabase():

    '''创建一个数据表'''

    sql = 'create table student(stuId int primary key,stuName text,stuAge text,stuGender text,stuClass text)'

    cursor.execute(sql)

    conn.commit()

def addInfo(sql = ''):

    '''添加数据'''

    if sql =='':

        # 如果是初始化,则默认会进行增加 6 条数据

        stuInfo = [(1001, '小华', '', '男', '二班'),

                    (1002, '小明', '', '女', '二班'),

                    (1003, '小李', '', '女', '一班'),

                    (1004, '小王', '', '男', '一班'),

                    (1005, '小刘', '', '女', '二班'),

                    (1006, '小张', '', '女', '一班')]

        cursor.executemany("insert into student values(?,?,?,?,?)",stuInfo)

        # 插入多条语句

        conn.commit()

def deleteInfo():

    '''删除数据'''

    cursor.execute("delete from student where stuId = 1005")

    # 将学号为 1005 的小刘同学删除

    conn.commit()

def modifyInfo():

    '''修改数据'''

    sql = "update student set stuAge = ? where stuId = ?"

    cursor.execute(sql,(20,1006))

    # 将小张的年龄修改为 20

    conn.commit()

def selectInfo():

    '''查询学生信息'''

    sql = 'select * from student'

    # 查询全部数据

    cursor.execute(sql)

    print(cursor.fetchall())

def main():

    # 创建一个数据表

    createDatabase()

    # 添加数据

    print("添加六条学生数据之后")

    addInfo()

    selectInfo()

    # 修改数据

    print("将小张的年龄修改为 20")

    modifyInfo()

    selectInfo()

    # 删除数据

    print("将学号为 1005 的小刘同学删除")

    deleteInfo()

    selectInfo()

    # cursor.execute('drop table student')

    # conn.commit()

main()