python 连接 sqlite 小示例

13 阅读1分钟
import sqlite3

create_table = '''
        CREATE TABLE student (
            id INTEGER PRIMARY KEY,
            name TEXT NOT NULL,
            gender TEXT,
            age INTEGER
        )
    '''
insert_data = 'insert into student(name,gender,age) values("Mon","Male",19),("Tue","Female",18),("Wed","Male",17)'


def create(cur):
    cur.execute(create_table)
    cur.execute(insert_data)


def insert(cur, name, sex, age):
    sql = f'insert into student(name,gender,age) values("{name}","{sex}",{age})'
    print(sql)
    cur.execute(sql)


if __name__ == '__main__':
    # 若文件存在则进行连接,若不存在则会自动创建后进行连接
    conn = sqlite3.connect('mydatabase.db')
    try:
        cursor = conn.cursor()
        # create(cursor)
        insert(cursor, 'Thue', 'F', 17)
        insert(cursor, 'Fri', 'M', 18)
        conn.commit()
    finally:
        conn.close()