# -*- encoding: utf-8 -*- 
# @Time: 2022/10/27 11:15
# @Author: 、一叶孤城
# @File: testSqlite
# @IDE: PyCharm
# @Motto: ABC(Always Be Coding)

import sqlite3

# Sqlite的用法
def connect_database():
    # 1，连接数据库
    conn = sqlite3.connect('test.db') # 如果不存在就会创建
    print('创建成功！')

def create_table():
    # 2，在数据库中创建表
    conn = sqlite3.connect('test.db') # 如果不存在就创建
    print('连接成功！')

    # 连接成功后就是获取游标
    cursor = conn.cursor() # 获取游标
    # 接下来就是sql语句，这里做一个建表示例
    sql = '''
        CREATE TABLE company (
            id int primary key not null,
            name text not null,
            age int not null,
            address char(50),
            salary real
        );
    '''
    # 执行sql语句
    cursor.execute(sql)

    # 执行完sql之后就是提交数据库更改
    conn.commit()

    # 关闭数据库
    conn.close()

def insert_to_table():
    # 3，像数据库添加内容
    # 首先还是先连接数据库
    conn = sqlite3.connect('test.db') # 连接数据库
    print('连接成功！')

    # 获取游标
    cursor = conn.cursor()

    # sql语句，要添加多条就写多条
    sql = '''
        INSERT INTO company (id,name,age,address,salary) 
            values (1, '张三', 28, '北京市王府井', 88000);
    '''
    sql1 = '''
        INSERT INTO company(id,name,age,address,salary)
            values (2, '李四',30,'厦门市',12000);
    '''

    # 利用游标执行sql语句
    cursor.execute(sql)
    cursor.execute(sql1)

    # 提交数据库更改
    conn.commit()
    print('添加成功！')
    # 关闭数据库
    conn.close()

def query_data_from_db():
    # 4，对数据库数据记性查询
    # 首先还是连接数据库
    conn = sqlite3.connect('test.db')
    print('连接成功！')

    # 获取游标
    cursor = conn.cursor()
    # sql查询语句
    query_sql = '''
        SELECT * FROM company;
    '''
    # 也可以这样写
    query_sql = '''
        SELECT id, name, age, address,salary FROM company;
    '''

    # 执行sql语句并返回一个结果
    query_results = cursor.execute(query_sql)
    # print(type(results))
    # 遍历结果
    for row in query_results:
        print(row) # 得到的每行结果都是一个元组类型
        print("查询序列号 = ", row[0])
        print("名字 = ",row[1])
        print("年龄 = ", row[2])
        print("地址 = " ,row[3])
        print("薪资 = ", row[4], '\n') # 每一查询一次换行
        # print("\n")

    # 因为是查询可以不用对数据库进行commit，直接关闭数据库
    # conn.commit()  # 查询的时候这行代码不需要
    print('数据库查询完毕，即将关闭数据库！')
    conn.close()

query_data_from_db()